mail2telegram/defs/models.py
2024-02-22 22:26:21 +08:00

37 lines
781 B
Python

from pydantic import BaseModel
from typing import TYPE_CHECKING
if TYPE_CHECKING:
from pyrogram import Client
from pyrogram.types import Message
TEMP = """#mail
✉️ %s (%s)
To: <code>%s</code>
%s"""
class Mail(BaseModel):
id: int
from_: str
subject: str
to: str
@property
def from_name(self) -> str:
li = self.from_.split(" <")
return " <".join(li[:-1]).strip()
@property
def from_at(self) -> str:
li = self.from_.split(" <")
return li[-1].strip()[:-1]
@property
def text(self) -> str:
return TEMP % (self.from_name, self.from_at, self.to, self.subject)
async def send(self, bot: "Client", chat_id: int) -> "Message":
return await bot.send_message(chat_id, self.text)