Move Telegram loop

This commit is contained in:
Alex Kotov 2023-01-14 21:23:59 +04:00
parent 5c156e92f1
commit 467d40d026
No known key found for this signature in database
GPG Key ID: 553C0EBBEB5D5F08
6 changed files with 19 additions and 40 deletions

View File

@ -1,7 +1,6 @@
import asyncio
from messengers import Matrix as MatrixLoop
from abstract_source_platform.telegram import Telegram
from messengers import Matrix as MatrixLoop, Telegram as TelegramLoop
from config_dataclass import Config
@ -18,14 +17,14 @@ class Application:
def __init__(self, config):
self.config = config
self.matrix_loop = MatrixLoop(self)
self.telegram = Telegram(self)
self.telegram_loop = TelegramLoop(self)
async def run(self):
try:
await self.matrix_loop.prepare()
await asyncio.gather(
self.matrix_loop.run(),
self.telegram.run(),
self.telegram_loop.run(),
)
finally:
if self.matrix_loop:

View File

@ -1,3 +0,0 @@
from abstract_source_platform.telegram import Telegram
platforms = [Telegram]

View File

@ -1,9 +0,0 @@
from abc import ABC, abstractmethod
from models.user import User
class AbstractSourcePlatform(ABC):
@abstractmethod
def get_user_information(self) -> User:
"""Init tables in database"""

View File

@ -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):
await self.dispatcher.start_polling()
async def on_message(self, msg):
print(msg, file=sys.stderr)
async def get_user_information(self) -> User:
pass

View File

@ -1 +1,2 @@
from .matrix import Matrix
from .telegram import Telegram

View File

@ -0,0 +1,15 @@
import aiogram
import sys
class Telegram:
def __init__(self, app):
self.app = app
self.bot = aiogram.Bot(token=app.config.telegram_bot_token)
self.dispatcher = aiogram.Dispatcher(bot=self.bot)
self.dispatcher.register_message_handler(self.on_message)
async def run(self):
await self.dispatcher.start_polling()
async def on_message(self, msg):
print(msg, file=sys.stderr)