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

Associate phone code hash with phone (so phone can change)

Browse Source
This commit is contained in:
Lonami Exo 2018-01-08 14:18:36 +01:00
parent c12af5e412
commit 01820c9943
1 changed files with 12 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
telethon/telegram_client.py
View File

@ -142,8 +142,9 @@ class TelegramClient(TelegramBareClient):
**kwargs **kwargs
) )
# Some fields to easy signing in # Some fields to easy signing in. Let {phone: hash} be
self._phone_code_hash = None # a dictionary because the user may change their mind.
self._phone_code_hash = {}
self._phone = None self._phone = None
# endregion # endregion
@ -167,18 +168,19 @@ class TelegramClient(TelegramBareClient):
Information about the result of the request. Information about the result of the request.
""" """
phone = utils.parse_phone(phone) or self._phone phone = utils.parse_phone(phone) or self._phone
phone_hash = self._phone_code_hash.get(phone)
if not self._phone_code_hash: if not phone_hash:
result = self(SendCodeRequest(phone, self.api_id, self.api_hash)) result = self(SendCodeRequest(phone, self.api_id, self.api_hash))
self._phone_code_hash = result.phone_code_hash self._phone_code_hash[phone] = phone_hash = result.phone_code_hash
else: else:
force_sms = True force_sms = True
self._phone = phone self._phone = phone
if force_sms: if force_sms:
result = self(ResendCodeRequest(phone, self._phone_code_hash)) result = self(ResendCodeRequest(phone, phone_hash))
self._phone_code_hash = result.phone_code_hash self._phone_code_hash[phone] = result.phone_code_hash
return result return result
@ -218,7 +220,9 @@ class TelegramClient(TelegramBareClient):
return self.send_code_request(phone) return self.send_code_request(phone)
elif code: elif code:
phone = utils.parse_phone(phone) or self._phone phone = utils.parse_phone(phone) or self._phone
phone_code_hash = phone_code_hash or self._phone_code_hash phone_code_hash = \
phone_code_hash or self._phone_code_hash.get(phone, None)
if not phone: if not phone:
raise ValueError( raise ValueError(
'Please make sure to call send_code_request first.' 'Please make sure to call send_code_request first.'
@ -274,7 +278,7 @@ class TelegramClient(TelegramBareClient):
""" """
result = self(SignUpRequest( result = self(SignUpRequest(
phone_number=self._phone, phone_number=self._phone,
phone_code_hash=self._phone_code_hash, phone_code_hash=self._phone_code_hash.get(self._phone, ''),
phone_code=code, phone_code=code,
first_name=first_name, first_name=first_name,
last_name=last_name last_name=last_name