Move Telegram loop
This commit is contained in:
parent
5c156e92f1
commit
467d40d026
|
@ -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:
|
||||
|
|
|
@ -1,3 +0,0 @@
|
|||
from abstract_source_platform.telegram import Telegram
|
||||
|
||||
platforms = [Telegram]
|
|
@ -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"""
|
|
@ -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
|
|
@ -1 +1,2 @@
|
|||
from .matrix import Matrix
|
||||
from .telegram import Telegram
|
||||
|
|
15
mirrortea/messengers/telegram.py
Normal file
15
mirrortea/messengers/telegram.py
Normal 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)
|
Reference in a new issue