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

Support interactively signing in as a bot

Browse Source
This commit is contained in:
Lonami Exo 2018-07-10 10:21:15 +02:00
parent ac5f8da50c
commit a50d013ee6
1 changed files with 14 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
telethon/client/auth.py
View File

@ -16,7 +16,7 @@ class AuthMethods(MessageParseMethods, UserMethods):
def start(
self,
phone=lambda: input('Please enter your phone: '),
phone=lambda: input('Please enter your phone (or bot token): '),
password=lambda: getpass.getpass('Please enter your password: '),
*,
bot_token=None, force_sms=False, code_callback=None,
@ -45,7 +45,8 @@ class AuthMethods(MessageParseMethods, UserMethods):
Args:
phone (`str` | `int` | `callable`):
The phone (or callable without arguments to get it)
to which the code will be sent.
to which the code will be sent. If a bot-token-like
string is given, it will be used as such instead.
password (`callable`, optional):
The password for 2 Factor Authentication (2FA).
@ -119,14 +120,21 @@ class AuthMethods(MessageParseMethods, UserMethods):
if await self.is_user_authorized():
return self
if not bot_token:
# Turn the callable into a valid phone number (or bot token)
while callable(phone):
value = phone()
if ':' in value:
# Bot tokens have 'user_id:access_hash' format
bot_token = value
break
phone = utils.parse_phone(value) or phone
if bot_token:
await self.sign_in(bot_token=bot_token)
return self
# Turn the callable into a valid phone number
while callable(phone):
phone = utils.parse_phone(phone()) or phone
me = None
attempts = 0
two_step_detected = False