class Stock(): def __init__ (self): self.view = View() self.model = Model() def get_company_info (self): self.company = self.view.get_company() response = self.model.search_company(self.company) dictt = response[0] self.view.company_info(dictt['Name'],dictt['Exchange'],dictt['Symbol']) def get_stock_quote (self): self.stock = self.view.get_stock() self.stock_info = self.model.stock_info(self.stock) self.view.stock_quote(self.stock_info) user1.choose_option() def buy_stocks (self): symbol = self.view.get_symbol() num = self.view.get_num_shares() if user1.info_list[0][4] == None: print ("Sorry! You are currently out of funds.") print ("") else: self.model.buy_stock(symbol, num, user1.info_list[0][0]) user1.choose_option() def sell_stocks(self): symbol = self.view.get_symbol() num = self.view.get_num_shares_to_sell() self.model.sell_stock(symbol,num,user1.info_list[0][0]) user1.choose_option()
def __init__ (self): self.view = View() self.model = Model()
def __init__ (self): self.view = View() self.model = Model() self.stock = Stock()
class User(): def __init__ (self): self.view = View() self.model = Model() self.stock = Stock() def choice_login (self): self.view.welcome() choice = self.view.choice_login() if choice == "1": return self.signup() if choice == "2": return self.login() if choice == "3": return self.login_admin() else: self.choice_login() def login_admin(self): return_adminlst = self.view.admin_login() result = self.model.check_login(return_adminlst[0], return_adminlst[1]) if result != []: all_accounts = self.model.get_top_accounts() self.view.display_all_accounts(all_accounts) else: print("You do not have admin rights") self.view.choice_login() def signup (self): name = self.view.get_name() username = self.view.get_username() password = self.view.get_password() balance = 10000.00 "send the values to the db" self.model.create_user(name, username, password, balance) self.view.restart() sys.exit() def login (self): # pu.db self.username = self.view.login_username() self.password = self.view.login_password() self.info_list = self.model.check_login (self.username, self.password) if self.info_list is not None: print ("") print ("Welcome back to the stock trading game!") balance = self.model.get_balance(self.username, self.password) print ("Your current balance is: ", balance, "dollars.") self.choose_option() else: self.view.try_again() sys.exit() def choose_option (self): choice = self.view.choose_option() if choice == "1": self.stock.get_company_info() self.stock.get_stock_quote() elif choice == "2": self.stock.buy_stocks() elif choice == "3": self.stock.sell_stocks() elif choice == "4": self.get_portfolio() elif choice == "5": print ("Goodbye.") sys.exit() else: print ("Please enter a valid choice. (1, 2, 3 or 4) ") self.choose_option() def get_portfolio (self): portfolio = self.model.get_portfolio(self.username, self.password) self.view.print_portfolio(portfolio) self.choose_option()