mirror of
https://github.com/tropicoo/yt-dlp-bot.git
synced 2024-09-20 06:46:08 +08:00
Refactor
This commit is contained in:
parent
1677d43cc8
commit
fa48a9d53b
|
@ -29,7 +29,13 @@ class TelegramCallback:
|
|||
async def on_message(self, client: VideoBot, message: Message) -> None:
|
||||
"""Receive video URL and send to the download worker."""
|
||||
self._log.debug('Received Telegram Message: %s', message)
|
||||
urls = message.text.splitlines()
|
||||
text = message.text
|
||||
if not text:
|
||||
self._log.warning('Received empty text: %s', message)
|
||||
await self._send_on_empty_message(message)
|
||||
return
|
||||
|
||||
urls = text.splitlines()
|
||||
user = client.allowed_users[get_user_id(message)]
|
||||
if user.use_url_regex_match:
|
||||
urls = self._url_parser.filter_urls(
|
||||
|
@ -61,6 +67,14 @@ class TelegramCallback:
|
|||
reply_to_message_id=message.id,
|
||||
)
|
||||
|
||||
@staticmethod
|
||||
async def _send_on_empty_message(message: Message) -> None:
|
||||
await message.reply(
|
||||
text='❓ Did you send anything?',
|
||||
parse_mode=ParseMode.HTML,
|
||||
reply_to_message_id=message.id,
|
||||
)
|
||||
|
||||
def _format_acknowledge_text(self, url_count: int) -> str:
|
||||
is_multiple = url_count > 1
|
||||
return self._MSG_SEND_OK.format(
|
||||
|
|
|
@ -18,7 +18,7 @@ from yt_shared.utils.file import file_size
|
|||
|
||||
|
||||
class InbMediaPayload(RealBaseModel):
|
||||
"""RabbitMQ incoming media payload from Telegram Bot or API service."""
|
||||
"""RabbitMQ inbound media payload from Telegram Bot or API service."""
|
||||
|
||||
id: uuid.UUID | None = None
|
||||
from_chat_id: StrictInt | None
|
||||
|
|
Loading…
Reference in a new issue