def welcome_homepage():
while True:
selection = view.welcome_screen()
if selection not in ["1", "2", "3"]:
view.improper_selection()
continue
if selection == "1":
username, balance, password, confirm_password = view.get_username(
), view.add_balance(), view.get_password(), view.confirm_password(
)
if password != confirm_password:
view.improper_password()
continue
if not balance.isdigit() or int(balance) < 0:
view.improper_balance()
continue
account = Account(username=username, balance=balance)
hashed_pw = Account.set_password(account, password)
account.set_api_key()
account.save()
logged_in_homepage(account)
return
elif selection == "2":
username, password = view.get_username(), view.get_password()
logged_in_account = Account.login(username=username,
password=password)
if logged_in_account:
logged_in_homepage(logged_in_account)
return
else:
print("Invalid credentials supplied")
continue
elif selection == "3":
view.goodbye()
return
