def/MirrorTea
def/MirrorTea
Archived
1
1
Fork 0
Code Issues Pull requests Releases Activity

Add an application class

Browse source
This commit is contained in:
Alex Kotov 2023-01-08 23:28:13 +04:00
parent 3c40b08b02
commit 2f5694cffb
No known key found for this signature in database
GPG key ID: 553C0EBBEB5D5F08
1 changed files with 21 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
mirrortea/__main__.py
View file

@ -10,23 +10,28 @@ MATRIX_FULL_USER_ID = os.environ['MATRIX_FULL_USER_ID']
MATRIX_PASSWORD = os.environ['MATRIX_PASSWORD'] MATRIX_PASSWORD = os.environ['MATRIX_PASSWORD']
TELEGRAM_BOT_TOKEN = os.environ['TELEGRAM_BOT_TOKEN'] TELEGRAM_BOT_TOKEN = os.environ['TELEGRAM_BOT_TOKEN']
async def main(): def main():
asyncio.run(Application().run())
class Application:
def __init__(self):
self.matrix_loop = MatrixLoop(
MATRIX_HOMESERVER_URL,
MATRIX_FULL_USER_ID,
MATRIX_PASSWORD,
)
self.telegram_loop = TelegramLoop(TELEGRAM_BOT_TOKEN)
async def run(self):
try: try:
matrix_loop = None await self.matrix_loop.prepare()
matrix_loop = MatrixLoop(MATRIX_HOMESERVER_URL, MATRIX_FULL_USER_ID,
MATRIX_PASSWORD)
await matrix_loop.prepare()
telegram_loop = TelegramLoop(TELEGRAM_BOT_TOKEN)
await asyncio.gather( await asyncio.gather(
matrix_loop.run(), self.matrix_loop.run(),
telegram_loop.run(), self.telegram_loop.run(),
) )
finally: finally:
if matrix_loop: if self.matrix_loop:
await matrix_loop.finish() await self.matrix_loop.finish()
class MatrixLoop: class MatrixLoop:
def __init__(self, homeserver_url, full_user_id, password): def __init__(self, homeserver_url, full_user_id, password):
@ -59,4 +64,4 @@ class TelegramLoop:
print(msg, file=sys.stderr) print(msg, file=sys.stderr)
if __name__ == '__main__': if __name__ == '__main__':
asyncio.run(main()) main()