LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2024-11-26 11:23:46 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fetch and persist each auth_key per DC

Browse Source
This commit is contained in:
Lonami Exo 2017-12-29 19:41:12 +01:00
parent 0570c55120
commit d2121c76cb
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
telethon/tl/session.py
View File

@ -168,6 +168,17 @@ class Session:
self._port = port self._port = port
self._update_session_table() self._update_session_table()
# Fetch the auth_key corresponding to this data center
c = self._conn.cursor()
c.execute('select auth_key from sessions')
tuple_ = c.fetchone()
if tuple_:
from ..crypto import AuthKey
self._auth_key = AuthKey(data=tuple_[0])
else:
self._auth_key = None
c.close()
@property @property
def server_address(self): def server_address(self):
return self._server_address return self._server_address
@ -188,8 +199,7 @@ class Session:
def _update_session_table(self): def _update_session_table(self):
with self._db_lock: with self._db_lock:
c = self._conn.cursor() c = self._conn.cursor()
c.execute('delete from sessions') c.execute('insert or replace into sessions values (?,?,?,?)', (
c.execute('insert into sessions values (?,?,?,?)', (
self._dc_id, self._dc_id,
self._server_address, self._server_address,
self._port, self._port,