yasirarism/pyrofork
1
0
Fork 0
mirror of https://github.com/Mayuri-Chan/pyrofork.git synced 2026-01-08 16:04:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Handle FirstnameInvalid error

Browse source
This commit is contained in:
Dan 2018-01-24 21:53:27 +01:00
parent e6fdc6a4e9
commit 604fc7af71
1 changed files with 10 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
pyrogram/client/client.py
View file

@ -37,7 +37,7 @@ from pyrogram.api.errors import (
PhoneNumberUnoccupied, PhoneCodeInvalid, PhoneCodeHashEmpty, PhoneNumberUnoccupied, PhoneCodeInvalid, PhoneCodeHashEmpty,
PhoneCodeExpired, PhoneCodeEmpty, SessionPasswordNeeded, PhoneCodeExpired, PhoneCodeEmpty, SessionPasswordNeeded,
PasswordHashInvalid, FloodWait, PeerIdInvalid, FilePartMissing, PasswordHashInvalid, FloodWait, PeerIdInvalid, FilePartMissing,
ChatAdminRequired ChatAdminRequired, FirstnameInvalid
) )
from pyrogram.api.types import ( from pyrogram.api.types import (
User, Chat, Channel, User, Chat, Channel,
@ -215,6 +215,7 @@ class Client:
phone_number_invalid_raises = self.phone_number is not None phone_number_invalid_raises = self.phone_number is not None
phone_code_invalid_raises = self.phone_code is not None phone_code_invalid_raises = self.phone_code is not None
password_hash_invalid_raises = self.password is not None password_hash_invalid_raises = self.password is not None
first_name_invalid_raises = self.first_name is not None
while True: while True:
if self.phone_number is None: if self.phone_number is None:
@ -298,8 +299,8 @@ class Client:
except PhoneNumberUnoccupied: except PhoneNumberUnoccupied:
pass pass
self.first_name = self.first_name or input("First name: ") self.first_name = self.first_name if self.first_name is not None else input("First name: ")
self.last_name = self.last_name or input("Last name: ") self.last_name = self.last_name if self.last_name is not None else input("Last name: ")
r = self.send( r = self.send(
functions.auth.SignUp( functions.auth.SignUp(
@ -316,6 +317,12 @@ class Client:
else: else:
print(e.MESSAGE) print(e.MESSAGE)
self.phone_code = None self.phone_code = None
except FirstnameInvalid as e:
if first_name_invalid_raises:
raise
else:
print(e.MESSAGE)
self.first_name = None
except SessionPasswordNeeded as e: except SessionPasswordNeeded as e:
print(e.MESSAGE) print(e.MESSAGE)
r = self.send(functions.account.GetPassword()) r = self.send(functions.account.GetPassword())