def create_account():
username = view.create_username()
password = view.create_password()
password_hash = hash_password(password)
new_account = Account(username=username,
password_hash=password_hash,
balance=0)
new_account.generate_api_key()
new_account.save()
view.login_menu()
login()
def login_menu():
while True:
try:
user_input = view.login_menu()
if int(user_input) == 3:
view.program_end()
sys.exit()
elif int(user_input) == 2:
username = view.username_inpt()
pwd = util.hash_pass(view.password_inpt())
user = Account.login(username, pwd)
if user == None:
view.invalid_info()
return login_menu()
return user
elif int(user_input) == 1:
user = Account()
user.username = view.username_inpt()
user.set_password(util.hash_pass(view.set_password_inpt()))
user.balance = view.deposit_inpt()
user.save()
view.acc_created(user.username)
return user
except ValueError:
view.choose_valid()
def login_loop(user_account):
while True:
choice = view.login_menu()
if choice == '1':
buy(user_account)
elif choice == '2':
sell(user_account)
elif choice == '3':
trades(user_account)
elif choice == '4':
withdraw(user_account)
elif choice == '5':
deposit(user_account)
elif choice == '6':
balance(user_account)
elif choice == '7':
get_positions(user_account)
elif choice == '8':
get_trades(user_account)
elif choice == '9':
look_up()
elif choice == '10':
get_api_key(user_account)
elif choice == '11':
logout_exit()