LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-02-18 12:30:59 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix AlchemySession session table updating

Browse Source
This commit is contained in:
Tulir Asokan 2018-03-02 21:58:16 +02:00
parent 47cdcda9e2
commit 290afd85fc
1 changed files with 15 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
telethon/sessions/sqlalchemy.py
View File

@ -117,9 +117,9 @@ class AlchemySession(MemorySession):
container.Version, container.Session, container.Entity, container.Version, container.Session, container.Entity,
container.SentFile) container.SentFile)
self.session_id = session_id self.session_id = session_id
self.load_session() self._load_session()
def load_session(self): def _load_session(self):
sessions = self._db_query(self.Session).all() sessions = self._db_query(self.Session).all()
session = sessions[0] if sessions else None session = sessions[0] if sessions else None
if session: if session:
@ -134,6 +134,19 @@ class AlchemySession(MemorySession):
def set_dc(self, dc_id, server_address, port): def set_dc(self, dc_id, server_address, port):
super().set_dc(dc_id, server_address, port) super().set_dc(dc_id, server_address, port)
self._update_session_table()
sessions = self._db_query(self.Session).all()
session = sessions[0] if sessions else None
if session and session.auth_key:
self._auth_key = AuthKey(data=session.auth_key)
else:
self._auth_key = None
@MemorySession.auth_key.setter
def auth_key(self, value):
self._auth_key = value
self._update_session_table()
def _update_session_table(self): def _update_session_table(self):
self.Session.query.filter( self.Session.query.filter(