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

Revert "Add workaround for SQLiteSession needing save after init"

Browse Source 
This reverts commit 8190a92aae.
This commit is contained in:
Lonami Exo 2022-08-30 12:20:20 +02:00
parent d5c864597c
commit 44e3651adf
2 changed files with 2 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
telethon/client/telegrambaseclient.py
View File

@ -523,12 +523,6 @@ class TelegramBaseClient(abc.ABC):
except OSError: except OSError:
print('Failed to connect') print('Failed to connect')
""" """
# Workaround specific to SQLiteSession, which sometimes need to persist info after init.
# Since .save() is now async we can't do that in init. Instead we do it in the first connect.
if isinstance(self.session, SQLiteSession) and not self.session._init_saved:
await self.session.save()
self.session._init_saved = True
if not await self._sender.connect(self._connection( if not await self._sender.connect(self._connection(
self.session.server_address, self.session.server_address,
self.session.port, self.session.port,

5
telethon/sessions/sqlite.py
View File

@ -37,7 +37,6 @@ class SQLiteSession(MemorySession):
super().__init__() super().__init__()
self.filename = ':memory:' self.filename = ':memory:'
self.save_entities = True self.save_entities = True
self._init_saved = True
if session_id: if session_id:
self.filename = session_id self.filename = session_id
@ -56,7 +55,7 @@ class SQLiteSession(MemorySession):
self._upgrade_database(old=version) self._upgrade_database(old=version)
c.execute("delete from version") c.execute("delete from version")
c.execute("insert into version values (?)", (CURRENT_VERSION,)) c.execute("insert into version values (?)", (CURRENT_VERSION,))
self._init_saved = False self.save()
# These values will be saved # These values will be saved
c.execute('select * from sessions') c.execute('select * from sessions')
@ -110,7 +109,7 @@ class SQLiteSession(MemorySession):
c.execute("insert into version values (?)", (CURRENT_VERSION,)) c.execute("insert into version values (?)", (CURRENT_VERSION,))
self._update_session_table() self._update_session_table()
c.close() c.close()
self._init_saved = False self.save()
def clone(self, to_instance=None): def clone(self, to_instance=None):
cloned = super().clone(to_instance) cloned = super().clone(to_instance)