def logged_in_homepage(account): while True: selection = view.logged_in_screen(account.username, account.balance) if selection not in ["1", "2", "3", "4", "5", "6", "7", "8"]: view.improper_selection() time.sleep(3) if selection == "1": view.account_positions(account.username) Account.get_positions(account) time.sleep(3) elif selection == "2": deposit = view.deposit_funds() if not deposit.isdigit() or int(deposit) < 1: view.improper_balance() time.sleep(3) else: account.balance = int(account.balance) + int(deposit) account.save() continue elif selection == "3": ticker_request = view.request_ticker_symbol() ticker_response = get_price(ticker_request) if type(ticker_response) == list: view.return_ticker_symbol_price(ticker_response) time.sleep(3) else: view.improper_ticker() time.sleep(3) elif selection == "4": ticker = view.request_ticker_symbol() ticker_price = get_price(ticker) if ticker_price == False: view.improper_ticker() time.sleep(3) continue purchase_amount = view.get_shares(ticker_price) if not purchase_amount.isdigit(): view.improper_money() time.sleep(3) continue total_cost = int(purchase_amount) * int(ticker_price[1]) if total_cost < int(account.balance): Account.buy(account, ticker, purchase_amount, ticker_price[1], total_cost) elif int(total_cost) > int(account.balance): view.not_enough_money() time.sleep(3) elif selection == "5": Account.get_positions(account) time.sleep(3) ticker_to_sell = view.sell_shares() has_stock = get_price(ticker_to_sell) position = account.get_position_for(ticker_to_sell) amount = view.sell_shares_amount() if has_stock and position.shares >= int(amount): Account.sell(account, ticker_to_sell, amount) elif not has_stock: view.improper_ticker() time.sleep(3) else: view.not_enough_shares() elif selection == "6": selection = view.select_trade_option(account.username) if len(selection) != 1 or selection.lower() not in ["a", "b", "c"]: view.improper_selection() time.sleep(3) elif selection.lower() == "a": account_trades = Account.get_trades(account) if not account_trades: view.no_trades(account.username) time.sleep(3) for trade in account_trades: view.show_trades(account.username, trade) time.sleep(3) elif selection.lower() == "b": ticker_symbol = view.request_ticker_symbol() account_trades_by_ticker = Account.trades_for(account, ticker_symbol) if get_price(ticker_symbol) == False: view.improper_ticker() time.sleep(3) elif not account_trades_by_ticker: view.no_trades(account.username) time.sleep(3) else: for trade in account_trades_by_ticker: view.show_trades(account.username, trade) time.sleep(3) elif selection.lower() == "c": continue elif selection == "7": view.goodbye() welcome_homepage() return elif selection == "8": view.goodbye() return
def logged_in_homepage(account): while True: print(account.username, account.balance) selection = view.logged_in_screen(account.username, account.balance) if selection not in ['1', '2', '3', '4', '5', '6', '7', '8', '9']: view.improper_selection() time.sleep(3) if selection == '1': Account.get_positions(account) elif selection == '2': deposit = view.deposit_funds() if not deposit.isdigit() or int(deposit) < 1: view.improper_balance() time.sleep(3) else: account.balance = int(account.balance) + int(deposit) account.save() continue elif selection == '3': ticker_request = view.request_ticker_symbol() ticker_response = get_price(ticker_request) if type(ticker_response) == list: view.return_ticker_symbol_price(ticker_response) time.sleep(2) else: view.improper_ticker() time.sleep(3) elif selection == '4': ticker = view.request_ticker_symbol() current_price = get_price(ticker) share_amount = view.get_shares(current_price) if share_amount: try: Account.buy(account, ticker, share_amount, current_price) except ValueError: view.insufficient_funds() else: view.improper_ticker() time.sleep(3) # elif selection == '5': # Account.get_positions(account) # # time.sleep(3) # ticker_to_sell = view.sell_shares() # has_stock = get_price(ticker_to_sell) # print(has_stock) # amount = view.sell_shares_amount() # current_price = get_price(ticker_to_sell) # if has_stock: # Account.sell(account, ticker_to_sell, amount, current_price) # else: # view.improper_ticker() # time.sleep(3) elif selection == '5': Account.get_positions(account) ticker = view.request_ticker_symbol() shares = int(view.sell_shares()) try: Account.sell(account, ticker, shares) except ValueError: view.wups() time.sleep(3) # view.clearscreen() elif selection == '6': selection = view.select_trade_option(account.username) if len(selection) != 1 or selection.lower() not in ['a', 'b', 'c']: view.improper_selection() time.sleep(3) elif selection.lower() == 'a': account_trades = Account.get_trades(account) for trade in account_trades: view.show_trades(account.username, trade) time.sleep(1) elif selection.lower() == 'b': ticker_symbol = view.request_ticker_symbol() account_trades_by_ticker = Account.trades_for( account, ticker_symbol) if account_trades_by_ticker: for trade in account_trades_by_ticker: view.show_trades(account.username, trade) time.sleep(3) else: view.improper_ticker() time.sleep(3) elif selection.lower() == 'c': continue elif selection == '7': view.goodbye() welcome_homepage() return elif selection == '8': view.goodbye() return elif selection == '9': top_traded( 'https://api.iextrading.com/1.0/stock/market/batch?symbols=aapl,fb&types=quote,news,chart&range=1m&last=5' ) time.sleep(3)
def logged_in_homepage(account): while True: selection = view.logged_in_screen(account.username, account.balance) if selection not in ["1", "2", "3", "4", "5", "6", "7", "8", "9"]: view.improper_selection() time.sleep(3) if selection == "1": Account.get_positions(account) elif selection == "2": deposit = view.deposit_funds() if not deposit.isdigit() or int(deposit) < 1: view.improper_balance() time.sleep(3) else: account.balance = int(account.balance) + int(deposit) account.save() continue elif selection == "3": ticker_request = view.request_ticker_symbol() ticker_response = get_price(ticker_request) if type(ticker_response) == list: view.return_ticker_symbol_price(ticker_response) time.sleep(3) else: view.improper_ticker() time.sleep(3) elif selection == "4": ticker = view.request_ticker_symbol() current_price = get_price(ticker) share_amount = view.get_shares(current_price) if share_amount: try: Account.buy(account, ticker, share_amount, current_price) except ValueError: view.insufficient_funds() else: view.improper_ticker() time.sleep(3) # elif selection == "5": # Account.get_positions(account) # # time.sleep(3) # ticker_to_sell = view.sell_shares() # has_stock = get_price(ticker_to_sell) # print(has_stock) # amount = view.sell_shares_amount() # current_price = get_price(ticker_to_sell) # if has_stock: # Account.sell(account, ticker_to_sell, amount, current_price) # else: # view.improper_ticker() # time.sleep(3) elif selection == "5": Account.get_positions(account) ticker = view.request_ticker_symbol() shares = int(view.sell_shares()) try: Account.sell(account, ticker, shares) except ValueError: view.wups() time.sleep(3) # view.clearscreen() elif selection == "6": selection = view.select_trade_option(account.username) if len(selection) != 1 or selection.lower() not in ["a", "b", "c"]: view.improper_selection() time.sleep(3) elif selection.lower() == "a": account_trades = Account.get_trades(account) for trade in account_trades: view.show_trades(account.username, trade) time.sleep(1) elif selection.lower() == "b": ticker_symbol = view.request_ticker_symbol() account_trades_by_ticker = Account.trades_for( account, ticker_symbol) if account_trades_by_ticker: for trade in account_trades_by_ticker: view.show_trades(account.username, trade) time.sleep(3) else: view.improper_ticker() time.sleep(3) elif selection.lower() == "c": continue elif selection == "7": view.goodbye() welcome_homepage() return elif selection == "8": view.goodbye() return elif selection == "9": top_traded( "https://api.iextrading.com/1.0/stock/market/batch?symbols=aapl,fb&types=quote,news,chart&range=1m&last=5" ) time.sleep(3)