hoelee/Telethon
1
0
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-06-25 06:26:40 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Avoid unnecessary "or 0" in the session classes

Browse Source
This commit is contained in:
Lonami Exo 2018-03-03 14:59:30 +01:00
parent 6060b3430c
commit 0e0e7f1c9e
3 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
telethon/sessions/memory.py
View File

@ -29,7 +29,7 @@ class MemorySession(Session):
def __init__(self): def __init__(self):
super().__init__() super().__init__()
self._dc_id = None self._dc_id = 0
self._server_address = None self._server_address = None
self._port = None self._port = None
self._auth_key = None self._auth_key = None
@ -38,7 +38,7 @@ class MemorySession(Session):
self._entities = set() self._entities = set()
def set_dc(self, dc_id, server_address, port): def set_dc(self, dc_id, server_address, port):
self._dc_id = dc_id self._dc_id = dc_id or 0
self._server_address = server_address self._server_address = server_address
self._port = port self._port = port

3
telethon/sessions/sqlalchemy.py
View File

@ -129,8 +129,7 @@ class AlchemySession(MemorySession):
self._auth_key = AuthKey(data=session.auth_key) self._auth_key = AuthKey(data=session.auth_key)
def clone(self, to_instance=None): def clone(self, to_instance=None):
cloned = to_instance or self.__class__(self.container, self.session_id) return super().clone(MemorySession())
return super().clone(cloned)
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)

2
telethon/sessions/sqlite.py
View File

@ -198,7 +198,7 @@ class SQLiteSession(MemorySession):
# multiple DCs. Probably done differently. # multiple DCs. Probably done differently.
c.execute('delete from sessions') c.execute('delete from sessions')
c.execute('insert or replace into sessions values (?,?,?,?)', ( c.execute('insert or replace into sessions values (?,?,?,?)', (
self._dc_id or 0, self._dc_id,
self._server_address, self._server_address,
self._port, self._port,
self._auth_key.key if self._auth_key else b'' self._auth_key.key if self._auth_key else b''