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

Call GetState on login to init MessageBox

Browse Source
This commit is contained in:
Lonami Exo
2022-05-16 10:04:49 +02:00
parent 3c7f53802f
commit 898eb5b82f
2 changed files with 12 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
telethon/client/auth.py
View File

@@ -7,6 +7,7 @@ import warnings
from .. import utils, helpers, errors, password as pwd_mod
from ..tl import types, functions, custom
from .._updates import SessionState
if typing.TYPE_CHECKING:
from .telegramclient import TelegramClient
@@ -371,7 +372,7 @@ class AuthMethods:
self._tos = result.terms_of_service
raise errors.PhoneNumberUnoccupiedError(request=request)
return self._on_login(result.user)
return await self._on_login(result.user)
async def sign_up(
self: 'TelegramClient',
@@ -466,9 +467,9 @@ class AuthMethods:
await self(
functions.help.AcceptTermsOfServiceRequest(self._tos.id))
return self._on_login(result.user)
return await self._on_login(result.user)
def _on_login(self, user):
async def _on_login(self, user):
"""
Callback called whenever the login or sign up process completes.
@@ -478,6 +479,9 @@ class AuthMethods:
self._self_input_peer = utils.get_input_peer(user, allow_self=False)
self._authorized = True
state = await self(functions.updates.GetStateRequest())
self._message_box.load(SessionState(0, 0, 0, state.pts, state.qts, int(state.date.timestamp()), state.seq, 0), [])
return user
async def send_code_request(