Revert "Add workaround for SQLiteSession needing save after init"

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

View File

@ -523,12 +523,6 @@ class TelegramBaseClient(abc.ABC):
except OSError:
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(
self.session.server_address,
self.session.port,

View File

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