From ad2434ab6766e6b025df5881dc07643cf178c01a Mon Sep 17 00:00:00 2001 From: AlberLC Date: Thu, 17 Apr 2025 23:44:16 +0200 Subject: [PATCH] Update type hints --- flanabot/bots/btc_offers_bot.py | 3 ++- flanabot/bots/flana_tele_bot.py | 4 ++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/flanabot/bots/btc_offers_bot.py b/flanabot/bots/btc_offers_bot.py index d0b5603..058f979 100644 --- a/flanabot/bots/btc_offers_bot.py +++ b/flanabot/bots/btc_offers_bot.py @@ -8,6 +8,7 @@ import json import os from abc import ABC from collections.abc import Callable +from typing import Any import aiohttp import flanautils @@ -23,7 +24,7 @@ from flanabot.models import Chat, Message # ---------------------------------------------------- # def preprocess_btc_offers(func: Callable) -> Callable: @functools.wraps(func) - async def wrapper(self: BtcOffersBot, message: Message): + async def wrapper(self: BtcOffersBot, message: Message) -> Any: if message.chat.is_group and not self.is_bot_mentioned(message): return diff --git a/flanabot/bots/flana_tele_bot.py b/flanabot/bots/flana_tele_bot.py index 7b4a146..6ec1ad2 100644 --- a/flanabot/bots/flana_tele_bot.py +++ b/flanabot/bots/flana_tele_bot.py @@ -4,7 +4,7 @@ __all__ = ['whitelisted', 'FlanaTeleBot'] import functools import os -from typing import Callable +from typing import Any, Callable import telethon.tl.functions from flanautils import Media, OrderedSet @@ -21,7 +21,7 @@ from flanabot.models import Message def whitelisted(func: Callable) -> Callable: @functools.wraps(func) @find_message - async def wrapper(self: FlanaTeleBot, message: Message, *args, **kwargs): + async def wrapper(self: FlanaTeleBot, message: Message, *args, **kwargs) -> Any: if message.author.id not in self.whitelist_ids: return