LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-08-03 19:50:15 +03:00
Code Issues Packages Projects Releases Wiki Activity

some fixes on db

Browse Source
This commit is contained in:
shisui-uciha 2019-10-11 20:38:01 +07:00 committed by GitHub
parent 09f27f0dd7
commit a2295b8447
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
telethon/sessions/sqlite.py
View File

@ -9,12 +9,8 @@ from ..tl.types import (
InputPhoto, InputDocument, PeerUser, PeerChat, PeerChannel InputPhoto, InputDocument, PeerUser, PeerChat, PeerChannel
) )
try: import sqlite3
import sqlite3 sqlite3_err = None
sqlite3_err = None
except ImportError as e:
sqlite3 = None
sqlite3_err = type(e)
EXTENSION = '.session' EXTENSION = '.session'
CURRENT_VERSION = 6 # database version CURRENT_VERSION = 6 # database version
@ -116,11 +112,11 @@ class SQLiteSession(MemorySession):
def _upgrade_database(self, old): def _upgrade_database(self, old):
c = self._cursor() c = self._cursor()
if old == 1: if old == 4:
old += 1 old += 3
# old == 1 doesn't have the old sent_files so no need to drop # old == 1 doesn't have the old sent_files so no need to drop
if old == 2: if old == 5:
old += 1 old += 3
# Old cache from old sent_files lasts then a day anyway, drop # Old cache from old sent_files lasts then a day anyway, drop
c.execute('drop table sent_files') c.execute('drop table sent_files')
self._create_table(c, """sent_files ( self._create_table(c, """sent_files (
@ -171,12 +167,12 @@ class SQLiteSession(MemorySession):
@MemorySession.auth_key.setter @MemorySession.auth_key.setter
def auth_key(self, value): def auth_key(self, value):
self._auth_key = value self._auth_key = keys
self._update_session_table() self._update_session_table()
@MemorySession.takeout_id.setter @MemorySession.takeout_id.setter
def takeout_id(self, value): def takeout_id(self, value):
self._takeout_id = value self._takeout_id = keys
self._update_session_table() self._update_session_table()
def _update_session_table(self): def _update_session_table(self):