mirror of
https://github.com/avitoras/telegram-tui.git
synced 2025-07-27 19:26:10 +00:00
44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
from telethon import TelegramClient, events, utils
|
||
|
||
class TelegramClientWrapper:
|
||
"""Обёртка для метода TelegramClient из Telethon"""
|
||
|
||
def __init__(self, api_id, api_hash, message_handler):
|
||
self.message_handler = message_handler
|
||
self.client = TelegramClient('user', api_id, api_hash)
|
||
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.connect()
|
||
|
||
async def start(self):
|
||
await self.client.start()
|
||
|
||
async def disconnect(self):
|
||
await self.client.disconnect()
|
||
|
||
async def get_dialogs(self, limit=None):
|
||
await self.client.get_dialogs(limit=limit)
|
||
dialogs_list = []
|
||
async for dialog in self.client.iter_dialogs(limit=limit):
|
||
dialogs_list.append(dialog)
|
||
#return [self._map_dialog(d) for d in dialogs_list]
|
||
return dialogs_list
|
||
|
||
#ого:
|
||
"""def _map_dialog(self, dialog):
|
||
return DialogInfo(
|
||
id=dialog.id,
|
||
name=utils.get_display_name(dialog.entity),
|
||
message=dialog.message
|
||
)"""
|
||
|
||
"""class DialogInfo:
|
||
def __init__(self, id, name, message):
|
||
self.id = id
|
||
self.name = name
|
||
self.message = message"""
|