Move messengers to separate module #3
|
@ -1,7 +1,6 @@
|
||||||
import asyncio
|
import asyncio
|
||||||
|
|
||||||
from messengers import Matrix as MatrixLoop
|
from messengers import Matrix as MatrixLoop, Telegram as TelegramLoop
|
||||||
from abstract_source_platform.telegram import Telegram
|
|
||||||
|
|
||||||
from config_dataclass import Config
|
from config_dataclass import Config
|
||||||
|
|
||||||
|
@ -18,14 +17,14 @@ class Application:
|
||||||
def __init__(self, config):
|
def __init__(self, config):
|
||||||
self.config = config
|
self.config = config
|
||||||
self.matrix_loop = MatrixLoop(self)
|
self.matrix_loop = MatrixLoop(self)
|
||||||
self.telegram = Telegram(self)
|
self.telegram_loop = TelegramLoop(self)
|
||||||
|
|
||||||
async def run(self):
|
async def run(self):
|
||||||
try:
|
try:
|
||||||
await self.matrix_loop.prepare()
|
await self.matrix_loop.prepare()
|
||||||
await asyncio.gather(
|
await asyncio.gather(
|
||||||
self.matrix_loop.run(),
|
self.matrix_loop.run(),
|
||||||
self.telegram.run(),
|
self.telegram_loop.run(),
|
||||||
)
|
)
|
||||||
finally:
|
finally:
|
||||||
if self.matrix_loop:
|
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 .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