LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-02-21 14:05:17 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix non-obvious error when not connected

Browse Source
This commit is contained in:
Lonami Exo 2022-02-24 11:10:35 +01:00
parent d5cdda28c5
commit a76c9e9f4c
1 changed files with 15 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
telethon/_client/telegrambaseclient.py
View File

@ -132,7 +132,7 @@ def init(
self._session = session self._session = session
# In-memory copy of the session's state to avoid a roundtrip as it contains commonly-accessed values. # In-memory copy of the session's state to avoid a roundtrip as it contains commonly-accessed values.
self._session_state = None self._session_state = _default_session_state()
# Nice-to-have. # Nice-to-have.
self._request_retries = request_retries self._request_retries = request_retries
@ -219,13 +219,8 @@ def set_flood_sleep_threshold(self, value):
self._flood_sleep_threshold = min(value or 0, 24 * 60 * 60) self._flood_sleep_threshold = min(value or 0, 24 * 60 * 60)
async def connect(self: 'TelegramClient') -> None: def _default_session_state():
all_dcs = {dc.id: dc for dc in await self._session.get_all_dc()} return SessionState(
self._session_state = await self._session.get_state()
if self._session_state is None:
try_fetch_user = False
self._session_state = SessionState(
user_id=0, user_id=0,
dc_id=DEFAULT_DC_ID, dc_id=DEFAULT_DC_ID,
bot=False, bot=False,
@ -235,6 +230,15 @@ async def connect(self: 'TelegramClient') -> None:
seq=0, seq=0,
takeout_id=None, takeout_id=None,
) )
async def connect(self: 'TelegramClient') -> None:
all_dcs = {dc.id: dc for dc in await self._session.get_all_dc()}
self._session_state = await self._session.get_state()
if self._session_state is None:
try_fetch_user = False
self._session_state = _default_session_state()
else: else:
try_fetch_user = self._session_state.user_id == 0 try_fetch_user = self._session_state.user_id == 0
if self._catch_up: if self._catch_up: