Use Telethon's new AlchemySession for session storage

mautrix_telegram/__main__.py

@@ -22,6 +22,7 @@ import asyncio
 import sqlalchemy as sql
 from sqlalchemy import orm
 
+from telethon.sessions import AlchemySessionContainer
 from mautrix_appservice import AppService
 
 from .base import Base
@@ -77,14 +78,17 @@ db_session = orm.scoping.scoped_session(db_factory)
 Base.metadata.bind = db_engine
 Base.metadata.create_all()
 
+telethon_session_container = AlchemySessionContainer(engine=db_engine, session=db_session,
+                                                     table_base=Base, table_prefix="telethon_",
+                                                     manage_tables=False)
+
 loop = asyncio.get_event_loop()
 
 appserv = AppService(config["homeserver.address"], config["homeserver.domain"],
                      config["appservice.as_token"], config["appservice.hs_token"],
                      config["appservice.bot_username"], log="mau.as", loop=loop)
 
-
-context = Context(appserv, db_session, config, loop, None, None)
+context = Context(appserv, db_session, config, loop, None, None, telethon_session_container)
 
 if config["appservice.public.enabled"]:
     public = PublicBridgeWebsite(loop)

mautrix_telegram/abstract_user.py

@@ -30,6 +30,7 @@ MAX_DELETIONS = 10
 
 
 class AbstractUser:
+    session_container = None
     loop = None
     log = None
     db = None
@@ -46,13 +47,15 @@ class AbstractUser:
         self.log.debug(f"Initializing client for {self.name}")
         device = f"{platform.system()} {platform.release()}"
         sysversion = MautrixTelegramClient.__version__
-        self.client = MautrixTelegramClient(self.name,
-                                            config["telegram.api_id"],
-                                            config["telegram.api_hash"],
-                                            loop=self.loop,
-                                            app_version=__version__,
-                                            system_version=sysversion,
-                                            device_model=device)
+        self.session = self.session_container.new_session(self.name)
+        self.session.app_version = __version__
+        self.session.system_version = sysversion
+        self.session.device_model = device
+
+        self.client = MautrixTelegramClient(session=self.session,
+                                            api_id=config["telegram.api_id"],
+                                            api_hash=config["telegram.api_hash"],
+                                            loop=self.loop)
         await self.client.add_event_handler(self._update_catch)
 
     async def update(self, update):
@@ -290,4 +293,5 @@ class AbstractUser:
 def init(context):
     global config, MAX_DELETIONS
     AbstractUser.az, AbstractUser.db, config, AbstractUser.loop, _ = context
+    AbstractUser.session_container = context.telethon_session_container
     MAX_DELETIONS = config.get("bridge.max_telegram_delete", 10)

mautrix_telegram/context.py

@@ -17,13 +17,14 @@
 
 
 class Context:
-    def __init__(self, az, db, config, loop, bot, mx):
+    def __init__(self, az, db, config, loop, bot, mx, telethon_session_container):
         self.az = az
         self.db = db
         self.config = config
         self.loop = loop
         self.bot = bot
         self.mx = mx
+        self.telethon_session_container = telethon_session_container
 
     def __iter__(self):
         yield self.az