ilia/nanobot
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

refactor: clean up telegram media sending logic

Browse Source
This commit is contained in:
Re-bin 2026-02-17 10:37:55 +00:00
parent c81cc07032
commit 5ad9c837df
1 changed files with 24 additions and 39 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

63
nanobot/channels/telegram.py
View File

@ -198,17 +198,17 @@ class TelegramChannel(BaseChannel):
await self._app.shutdown() await self._app.shutdown()
self._app = None self._app = None
def _get_media_type(self, path: str) -> str: @staticmethod
def _get_media_type(path: str) -> str:
"""Guess media type from file extension.""" """Guess media type from file extension."""
path = path.lower() ext = path.rsplit(".", 1)[-1].lower() if "." in path else ""
if path.endswith(('.jpg', '.jpeg', '.png', '.gif', '.webp')): if ext in ("jpg", "jpeg", "png", "gif", "webp"):
return "photo" return "photo"
elif path.endswith('.ogg'): if ext == "ogg":
return "voice" return "voice"
elif path.endswith(('.mp3', '.m4a', '.wav', '.aac')): if ext in ("mp3", "m4a", "wav", "aac"):
return "audio" return "audio"
else: return "document"
return "document"
async def send(self, msg: OutboundMessage) -> None: async def send(self, msg: OutboundMessage) -> None:
"""Send a message through Telegram.""" """Send a message through Telegram."""
@ -224,39 +224,24 @@ class TelegramChannel(BaseChannel):
logger.error(f"Invalid chat_id: {msg.chat_id}") logger.error(f"Invalid chat_id: {msg.chat_id}")
return return
# Handle media files # Send media files
if msg.media: for media_path in (msg.media or []):
for media_path in msg.media: try:
try: media_type = self._get_media_type(media_path)
media_type = self._get_media_type(media_path) sender = {
"photo": self._app.bot.send_photo,
# Determine caption (only for first media or if explicitly set, "voice": self._app.bot.send_voice,
# but here we keep it simple: content is sent separately if media is present "audio": self._app.bot.send_audio,
# to avoid length issues, unless we want to attach it to the first media) }.get(media_type, self._app.bot.send_document)
# For simplicity: send media first, then text if present. param = "photo" if media_type == "photo" else media_type if media_type in ("voice", "audio") else "document"
# Or: if single media, attach text as caption. with open(media_path, 'rb') as f:
await sender(chat_id=chat_id, **{param: f})
# Let's attach content as caption to the last media if single, except Exception as e:
# otherwise send text separately. filename = media_path.rsplit("/", 1)[-1]
logger.error(f"Failed to send media {media_path}: {e}")
with open(media_path, 'rb') as f: await self._app.bot.send_message(chat_id=chat_id, text=f"[Failed to send: {filename}]")
if media_type == "photo":
await self._app.bot.send_photo(chat_id=chat_id, photo=f)
elif media_type == "voice":
await self._app.bot.send_voice(chat_id=chat_id, voice=f)
elif media_type == "audio":
await self._app.bot.send_audio(chat_id=chat_id, audio=f)
else:
await self._app.bot.send_document(chat_id=chat_id, document=f)
except Exception as e:
logger.error(f"Failed to send media {media_path}: {e}")
await self._app.bot.send_message(
chat_id=chat_id,
text=f"[Failed to send file: {media_path}]"
)
# Send text content if present # Send text content
if msg.content and msg.content != "[empty message]": if msg.content and msg.content != "[empty message]":
for chunk in _split_message(msg.content): for chunk in _split_message(msg.content):
try: try: