74 lines
2.3 KiB
Python
74 lines
2.3 KiB
Python
from aiogram.types import Message
|
|
|
|
from app.models.domain import TgIncomingMessage
|
|
|
|
|
|
def format_tg_author(message: Message) -> tuple[str, str | None]:
|
|
user = message.from_user
|
|
if user is None:
|
|
return "Unknown", None
|
|
parts = [user.first_name or "", user.last_name or ""]
|
|
name = " ".join(p for p in parts if p).strip() or (user.username or "User")
|
|
return name, user.username
|
|
|
|
|
|
def format_tg_to_max_text(author_name: str, username: str | None, text: str) -> str:
|
|
handle = f" (@{username})" if username else ""
|
|
return f"{author_name}{handle}:\n{text}"
|
|
|
|
|
|
def extract_tg_media(message: Message) -> list[dict]:
|
|
items: list[dict] = []
|
|
if message.photo:
|
|
photo = message.photo[-1]
|
|
items.append({"kind": "photo", "file_id": photo.file_id})
|
|
elif message.video:
|
|
items.append({"kind": "video", "file_id": message.video.file_id})
|
|
elif message.document:
|
|
items.append(
|
|
{
|
|
"kind": "document",
|
|
"file_id": message.document.file_id,
|
|
"file_name": message.document.file_name,
|
|
"mime_type": message.document.mime_type,
|
|
}
|
|
)
|
|
elif message.audio:
|
|
items.append({"kind": "audio", "file_id": message.audio.file_id})
|
|
elif message.voice:
|
|
items.append({"kind": "voice", "file_id": message.voice.file_id})
|
|
elif message.sticker or message.animation or message.video_note:
|
|
items.append({"kind": "unsupported", "file_id": None})
|
|
return items
|
|
|
|
|
|
def build_tg_incoming(message: Message, forum_channel_id: int) -> TgIncomingMessage | None:
|
|
if message.chat.id != forum_channel_id:
|
|
return None
|
|
if message.is_topic_message is False and message.message_thread_id is None:
|
|
return None
|
|
|
|
author_name, username = format_tg_author(message)
|
|
text = message.text or message.caption or ""
|
|
media = extract_tg_media(message)
|
|
|
|
if not text and not media:
|
|
return None
|
|
|
|
if media and not text and any(m.get("kind") == "unsupported" for m in media):
|
|
text = "[Telegram files]"
|
|
|
|
return TgIncomingMessage(
|
|
tg_chat_id=message.chat.id,
|
|
tg_message_id=message.message_id,
|
|
tg_thread_id=message.message_thread_id,
|
|
text=text,
|
|
author_name=author_name,
|
|
author_username=username,
|
|
is_bot=bool(message.from_user and message.from_user.is_bot),
|
|
reply_to_tg_message_id=message.reply_to_message.message_id
|
|
if message.reply_to_message
|
|
else None,
|
|
media=media,
|
|
)
|