hoelee/Telethon
1
0
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-08-08 21:10:29 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update and persist session state on successful login

Browse Source
This commit is contained in:
Lonami Exo
2021-09-19 18:15:09 +02:00
parent 3f13357d0f
commit cc3d4145d8
4 changed files with 20 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
telethon/_client/auth.py
View File

@@ -253,7 +253,7 @@ async def sign_in(
self._tos = result.terms_of_service
raise errors.PhoneNumberUnoccupiedError(request=request)
return _on_login(self, result.user)
return await _update_session_state(self, result.user)
async def sign_up(
self: 'TelegramClient',
@@ -306,17 +306,28 @@ async def sign_up(
await self(
_tl.fn.help.AcceptTermsOfService(self._tos.id))
return _on_login(self, result.user)
return await _update_session_state(self, result.user)
def _on_login(self, user):
async def _update_session_state(self, user, save=True):
"""
Callback called whenever the login or sign up process completes.
Returns the input user parameter.
"""
self._bot = bool(user.bot)
self._self_input_peer = utils.get_input_peer(user, allow_self=False)
self._authorized = True
self._session_state.user_id = user.id
self._session_state.bot = user.bot
state = await self(_tl.fn.updates.GetState())
self._session_state.pts = state.pts
self._session_state.qts = state.qts
self._session_state.date = int(state.date.timestamp())
self._session_state.seq = state.seq
await self.session.set_state(self._session_state)
if save:
await self.session.save()
return user
async def send_code_request(