yasirarism/pyrofork
1
0
Fork 0
mirror of https://github.com/Mayuri-Chan/pyrofork.git synced 2026-01-07 15:34:52 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Replace input() with ainput() in Client

Browse source
This commit is contained in:
Dan 2018-07-02 14:10:48 +02:00
parent 26bb97af46
commit af5c5d20cf
1 changed files with 7 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
pyrogram/client/client.py
View file

@ -48,6 +48,7 @@ from pyrogram.crypto import AES
from pyrogram.session import Auth, Session from pyrogram.session import Auth, Session
from .dispatcher import Dispatcher from .dispatcher import Dispatcher
from .ext import BaseClient, Syncer, utils from .ext import BaseClient, Syncer, utils
from .ext.utils import ainput
from .handlers import DisconnectHandler from .handlers import DisconnectHandler
from .methods import Methods from .methods import Methods
@ -413,15 +414,15 @@ class Client(Methods, BaseClient):
while True: while True:
if self.phone_number is None: if self.phone_number is None:
self.phone_number = input("Enter phone number: ") self.phone_number = await ainput("Enter phone number: ")
while True: while True:
confirm = input("Is \"{}\" correct? (y/n): ".format(self.phone_number)) confirm = await ainput("Is \"{}\" correct? (y/n): ".format(self.phone_number))
if confirm in ("y", "1"): if confirm in ("y", "1"):
break break
elif confirm in ("n", "2"): elif confirm in ("n", "2"):
self.phone_number = input("Enter phone number: ") self.phone_number = await ainput("Enter phone number: ")
self.phone_number = self.phone_number.strip("+") self.phone_number = self.phone_number.strip("+")
@ -488,7 +489,7 @@ class Client(Methods, BaseClient):
while True: while True:
self.phone_code = ( self.phone_code = (
input("Enter phone code: ") if self.phone_code is None await ainput("Enter phone code: ") if self.phone_code is None
else self.phone_code if type(self.phone_code) is str else self.phone_code if type(self.phone_code) is str
else str(self.phone_code(self.phone_number)) else str(self.phone_code(self.phone_number))
) )
@ -514,8 +515,8 @@ class Client(Methods, BaseClient):
except PhoneNumberUnoccupied: except PhoneNumberUnoccupied:
pass pass
self.first_name = self.first_name if self.first_name is not None else input("First name: ") self.first_name = self.first_name if self.first_name is not None else await ainput("First name: ")
self.last_name = self.last_name if self.last_name is not None else input("Last name: ") self.last_name = self.last_name if self.last_name is not None else await ainput("Last name: ")
r = await self.send( r = await self.send(
functions.auth.SignUp( functions.auth.SignUp(