From 9d7df8202b3a68c330007b8cd1fc0c5715dd0eac Mon Sep 17 00:00:00 2001 From: def Date: Tue, 10 Jan 2023 20:08:27 +0400 Subject: [PATCH] brought everything back to basics --- mirrortea/__main__.py | 57 +++++++++++++++++-- .../abstract_source_platform/telegram.py | 24 -------- mirrortea/matrix.py | 35 ------------ 3 files changed, 52 insertions(+), 64 deletions(-) delete mode 100644 mirrortea/abstract_source_platform/telegram.py delete mode 100644 mirrortea/matrix.py diff --git a/mirrortea/__main__.py b/mirrortea/__main__.py index 2dc2aba..c49c7d3 100644 --- a/mirrortea/__main__.py +++ b/mirrortea/__main__.py @@ -1,7 +1,7 @@ -import asyncio +import sys -from matrix import MatrixLoop -from abstract_source_platform.telegram import Telegram +import asyncio as telegram +import nio as matrix from config_dataclass import Config @@ -11,14 +11,61 @@ CONFIG_FILE_NAME = "config.yaml" def main(): config = Config.from_yaml_config(CONFIG_FILE_NAME) - asyncio.run(Application(config).run()) + telegram.run(Application(config).run()) + + +class TelegramLopp: + def __init__(self, app): + self.app = app + self.bot = telegram.Bot(token=app.config.telegram_bot_token) + self.dispatcher = telegram.Dispatcher(bot=self.bot) + self.dispatcher.register_message_handler(self.on_message) + + async def run(self) -> None: + await self.dispatcher.start_polling() + + async def on_message(self, msg): + print(msg, file=sys.stderr) + + +class MatrixLoop: + def __init__(self, app): + self.app = app + self.client = matrix.AsyncClient( + app.config.matrix_homeserver_url, + app.config.matrix_full_bot_id, + ) + self.client.add_event_callback(self.on_message, matrix.RoomMessage) + + async def prepare(self): + await self.client.login(self.app.config.matrix_bot_password) + + async def finish(self): + await self.client.close() + + async def run(self): + await self.client.sync_forever(timeout=30000) + + async def on_message(self, room, event): + print(room, event, file=sys.stderr) + + def upgrade_room(self, room, telegram_nickname): + event_dict = matrix.event_builders.event_builder.EventBuilder( + name=telegram_nickname + ).as_dict() + client.room_send( + room_id=room, + message_type=event_dict["type"], + content=event_dict["content"], + ) # предположу что оно так работает + # https://matrix-nio.readthedocs.io/en/latest/nio.html#module-nio.event_builders.state_events class Application: def __init__(self, config): self.config = config self.matrix_loop = MatrixLoop(self) - self.telegram = Telegram(self) + self.telegram = TelegramLopp(self) async def run(self): try: diff --git a/mirrortea/abstract_source_platform/telegram.py b/mirrortea/abstract_source_platform/telegram.py deleted file mode 100644 index 5c4e383..0000000 --- a/mirrortea/abstract_source_platform/telegram.py +++ /dev/null @@ -1,24 +0,0 @@ -import sys -import aiogram as telegram - -from abstract_source_platform.abstact_source_platform import ( - AbstractSourcePlatform, -) -from models.user import User - - -class Telegram(AbstractSourcePlatform): - def __init__(self, app): - self.app = app - self.bot = telegram.Bot(token=app.config.telegram_bot_token) - self.dispatcher = telegram.Dispatcher(bot=self.bot) - self.dispatcher.register_message_handler(self.on_message) - - async def run(self) -> None: - await self.dispatcher.start_polling() - - async def on_message(self, msg): - print(msg, file=sys.stderr) - - async def get_user_information(self) -> User: - pass diff --git a/mirrortea/matrix.py b/mirrortea/matrix.py deleted file mode 100644 index 061b4a3..0000000 --- a/mirrortea/matrix.py +++ /dev/null @@ -1,35 +0,0 @@ -import nio as matrix -import sys - - -class MatrixLoop: - def __init__(self, app): - self.app = app - self.client = matrix.AsyncClient( - app.config.matrix_homeserver_url, - app.config.matrix_full_bot_id, - ) - self.client.add_event_callback(self.on_message, matrix.RoomMessage) - - async def prepare(self): - await self.client.login(self.app.config.matrix_bot_password) - - async def finish(self): - await self.client.close() - - async def run(self): - await self.client.sync_forever(timeout=30000) - - async def on_message(self, room, event): - print(room, event, file=sys.stderr) - - def upgrade_room(self, room, telegram_nickname): - event_dict = matrix.event_builders.event_builder.EventBuilder( - name=telegram_nickname - ).as_dict() - client.room_send( - room_id=room, - message_type=event_dict["type"], - content=event_dict["content"], - ) # предположу что оно так работает - # https://matrix-nio.readthedocs.io/en/latest/nio.html#module-nio.event_builders.state_events