diff --git a/telegram/client.py b/telegram/client.py index ad759ce..4a3ab25 100644 --- a/telegram/client.py +++ b/telegram/client.py @@ -2,12 +2,15 @@ from telethon import TelegramClient, events, utils class TelegramClientWrapper: def __init__(self, api_id, api_hash, message_handler): + self.message_handler = message_handler self.client = TelegramClient('user', api_id, api_hash) - #self.client.add_event_handler(message_handler, events.NewMessage()) - self.client.on(events.NewMessage())(message_handler) + self.client.on(events.NewMessage())(self.local_message_handler) #ни то ни то не работает, костя спаси + async def local_message_handler(self, event): + await self.message_handler() + async def connect(self): await self.client.start()