spawnrys/flanabot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add timeout_for_media

Browse Source
This commit is contained in:
AlberLC
2022-09-21 05:10:44 +02:00
parent 25a1a1b601
commit 68c0558f65
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
flanabot/bots/flana_tele_bot.py
View File

@@ -8,6 +8,7 @@ from typing import Callable, Sequence
import flanautils import flanautils
import telethon.tl.functions import telethon.tl.functions
from flanautils import Media, OrderedSet
from multibot import TelegramBot, find_message, user_client from multibot import TelegramBot, find_message, user_client
from flanabot.bots.flana_bot import FlanaBot from flanabot.bots.flana_bot import FlanaBot
@@ -58,6 +59,9 @@ class FlanaTeleBot(TelegramBot, FlanaBot):
return [contact.user_id for contact in contacts_data.contacts] return [contact.user_id for contact in contacts_data.contacts]
async def _search_medias(self, message: Message, audio_only=False, timeout_for_media: int | float = None) -> OrderedSet[Media]:
return await super()._search_medias(message, audio_only, timeout_for_media=45)
@user_client @user_client
async def _update_whitelist(self): async def _update_whitelist(self):
self.whitelist_ids = [self.owner_id, self.id] + await self._get_contacts_ids() self.whitelist_ids = [self.owner_id, self.id] + await self._get_contacts_ids()