LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-02-18 04:20:57 +03:00
Code Issues 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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@ -6,6 +6,7 @@
# User session # User session
*.session *.session
sessions/
/usermedia/ /usermedia/
# Builds and testing # Builds and testing

21
telethon/_client/auth.py
View File

@ -253,7 +253,7 @@ async def sign_in(
self._tos = result.terms_of_service self._tos = result.terms_of_service
raise errors.PhoneNumberUnoccupiedError(request=request) raise errors.PhoneNumberUnoccupiedError(request=request)
return _on_login(self, result.user) return await _update_session_state(self, result.user)
async def sign_up( async def sign_up(
self: 'TelegramClient', self: 'TelegramClient',
@ -306,17 +306,28 @@ async def sign_up(
await self( await self(
_tl.fn.help.AcceptTermsOfService(self._tos.id)) _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. Callback called whenever the login or sign up process completes.
Returns the input user parameter. 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._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 return user
async def send_code_request( async def send_code_request(

11
telethon/_client/telegrambaseclient.py
View File

@ -385,16 +385,7 @@ async def connect(self: 'TelegramClient') -> None:
# logged in). Attempt to fetch the user now. If it works, also get the update state. # logged in). Attempt to fetch the user now. If it works, also get the update state.
me = await self.get_me() me = await self.get_me()
if me: if me:
self._session_state.user_id = me.id await self._update_session_state(me, save=False)
self._session_state.bot = me.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)
await self.session.save() await self.session.save()

4
telethon/_client/telegramclient.py
View File

@ -3598,8 +3598,8 @@ class TelegramClient:
async def _handle_auto_reconnect(self: 'TelegramClient'): async def _handle_auto_reconnect(self: 'TelegramClient'):
return await updates._handle_auto_reconnect(**locals()) return await updates._handle_auto_reconnect(**locals())
def _self_id(self: 'TelegramClient') -> typing.Optional[int]: async def _update_session_state(self, user, save=True):
return users._self_id(**locals()) return await auth._update_session_state(**locals())
# endregion Private # endregion Private