mirror of
https://github.com/tropicoo/yt-dlp-bot.git
synced 2025-03-03 18:23:34 +08:00
27 lines
936 B
Python
27 lines
936 B
Python
import logging
|
|
|
|
from pyrogram import Client
|
|
from pyrogram.enums import ParseMode
|
|
from pyrogram.types import Message
|
|
|
|
from core.service import URLService
|
|
from core.utils import bold
|
|
from yt_shared.emoji import SUCCESS_EMOJI
|
|
|
|
|
|
class TelegramCallback:
|
|
_MSG_SEND_OK = f'{SUCCESS_EMOJI} {bold("URL sent for download")}'
|
|
_MSG_SEND_FAIL = f'🛑 {bold("Failed to send URL for download")}'
|
|
|
|
def __init__(self) -> None:
|
|
self._log = logging.getLogger(self.__class__.__name__)
|
|
self._url_service = URLService()
|
|
|
|
async def on_message(self, client: Client, message: Message) -> None:
|
|
"""Receive video URL and send to the download worker."""
|
|
is_sent = await self._url_service.process_url(message.text, message.id)
|
|
await message.reply(
|
|
self._MSG_SEND_OK if is_sent else self._MSG_SEND_FAIL,
|
|
parse_mode=ParseMode.HTML,
|
|
reply_to_message_id=message.id,
|
|
)
|