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

Update _on_scraping bot_mentioned

Browse Source
This commit is contained in:
AlberLC
2022-12-17 04:39:34 +01:00
parent ef5acc3f26
commit 333f494a20
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
flanabot/bots/scraper_bot.py
View File

@@ -9,7 +9,7 @@ from typing import Iterable
import flanautils import flanautils
from flanaapis import instagram, reddit, tiktok, twitter, yt_dlp_wrapper from flanaapis import instagram, reddit, tiktok, twitter, yt_dlp_wrapper
from flanautils import Media, MediaType, OrderedSet, return_if_first_empty from flanautils import Media, MediaType, OrderedSet, return_if_first_empty
from multibot import MultiBot, RegisteredCallback, SendError, bot_mentioned, constants as multibot_constants, reply from multibot import MultiBot, RegisteredCallback, SendError, constants as multibot_constants, reply
from flanabot import constants from flanabot import constants
from flanabot.models import Action, BotAction, Message from flanabot.models import Action, BotAction, Message
@@ -163,9 +163,10 @@ class ScraperBot(MultiBot, ABC):
async def _on_recover_message(self, message: Message): async def _on_recover_message(self, message: Message):
pass pass
@bot_mentioned
async def _on_scraping(self, message: Message, audio_only=False) -> OrderedSet[Media]: async def _on_scraping(self, message: Message, audio_only=False) -> OrderedSet[Media]:
sended_media_messages = OrderedSet() sended_media_messages = OrderedSet()
if not message.chat.config['auto_scraping'] and not self.is_bot_mentioned(message):
return sended_media_messages
if message.replied_message: if message.replied_message:
sended_media_messages += await self._scrape_and_send(message.replied_message, audio_only) sended_media_messages += await self._scrape_and_send(message.replied_message, audio_only)