def test_select_customer(self):
bank = Bank()
account = Account(100, 1000000)
customer = Customer('john', 'smith', '123456789', account)
bank.add_customer(customer)
bank.select_customer()
def test_add_customer(self):
bank = Bank()
account = Account(100, 1000000)
customer = Customer('john', 'smith', '123456789', account)
bank.add_customer(customer)
self.assertEquals(bank.customers[0], customer)
self.assertEquals(bank.customers[0].first_name, 'john')
self.assertEquals(bank.customers[0].last_name, 'smith')
self.assertEquals(bank.customers[0].ssn, '123456789')
self.assertEquals(bank.customers[0].account.balance, 100)
def __init__(self):
"""Constructor for the menu class
Args:
None
"""
self.bank = Bank()
if os.path.exists(os.path.abspath('data/bank.pkl')):
self.bank = self.load_state()
self.exit = False
class Menu(object):
"""Menu class represents the banking application's menu
Attributes:
bank (Bank): The bank
exit (boolean): An indicator for an exit condition of when to leave the app
"""
def __init__(self):
"""Constructor for the menu class
Args:
None
"""
self.bank = Bank()
if os.path.exists(os.path.abspath('data/bank.pkl')):
self.bank = self.load_state()
self.exit = False
@staticmethod
def print_header():
"""Function to print the menu's header
Args:
None
"""
print("****************************************************************************")
print("*********** *********************")
print("*********** Welcome to Jason's Awesome Banking App!! *********************")
print("*********** *********************")
print("****************************************************************************\n")
@staticmethod
def print_menu():
"""Function to print the menu's options
Args:
None
"""
print("\nPlease make a selection ")
print("1. Create a new account")
print("2. Make a deposit")
print("3. Make a withdrawal")
print("4. Show account details")
print("0. Exit")
print("")
def create_account(self):
"""Function to create an account in the bank
Args:
None
"""
account_type = DataValidation.check_valid_account_type("Would you like a checking or savings account? ")
# exit create_account if user types '0' without proceeding to first_name prompt
if not account_type:
return None
first_name = DataValidation.check_valid_name("What is your first name? ")
# exit create_account if user types '0' without proceeding to last_name prompt
if not first_name:
return None
last_name = DataValidation.check_valid_name("What is your last name? ")
# exit create_account if user types '0' without proceeding to ssn prompt
if not last_name:
return None
ssn = DataValidation.check_valid_ssn("What is your social security number? ")
# exit create_account if user types '0' without proceeding to initial deposit prompt
if not ssn:
return None
initial_deposit = DataValidation.get_valid_initial_deposit(account_type, "What is your initial deposit? ")
# exit create_account if user types '0' without proceeding to instanciate account
if not initial_deposit:
return None
# create account and add customer to bank
account = self.bank.add_account(initial_deposit, account_type)
customer = Customer(first_name, last_name, ssn, account)
self.bank.add_customer(customer)
def make_deposit(self):
"""Function to make a deposit into a customer's account
Args:
None
"""
customer = self.bank.select_customer()
# exit make_deposit if user types '0' without proceeding to deposit prompt
if not customer:
return None
# select customer's account, display current balances, and prompt for deposit amount
account = customer.account
account.print_balance()
amount = DataValidation.check_positive_float("How much would you like to deposit? ")
# exit make_deposit if user types '0' without proceeding to deposit prompt
if not amount:
return None
# deposit amount into account
account.deposit(amount)
return None
def make_withdrawal(self):
"""Function to withdraw money from an account in the bank
Args:
None
"""
customer = self.bank.select_customer()
# exit make_withdrawal if user types '0' without proceeding to withdrawal prompt
if not customer:
return None
# give customer ability to exit if they do not want to incur fee
proceed = DataValidation.check_valid_yes_no("Withdrawals incur a $5.00 fee, would you like to proceed? (Y/N) ")
# exit make_deposit if user types '0' without proceeding to deposit prompt
if not proceed:
return None
# select customer's account, display current balances, and prompt for withdrawal amount
account = customer.account
account.print_balance()
amount = DataValidation.check_positive_float("How much would you like to withdraw? ")
# exit make_deposit if user types '0' without proceeding to withdrawal prompt
if not amount:
return None
# withdraw amount from account
account.withdraw(amount)
return None
def show_details(self):
"""Function to create an account in the bank
Args:
None
"""
customer = self.bank.select_customer()
# exit make_withdrawal if user types '0' without proceeding to withdrawal prompt
if not customer:
return None
# print the customer object
print(customer)
raw_input("Press enter key to continue...")
return None
def exit_menu(self):
"""Function to exit the menu and application
Args:
None
"""
print("Thank you for visiting, see you next time!")
self.exit = True
def perform_action(self):
"""Function to perform an action from a menu's selection
Args:
None
"""
if self.choice == 1:
return self.create_account()
elif self.choice == 2:
return self.make_deposit()
elif self.choice == 3:
return self.make_withdrawal()
elif self.choice == 4:
return self.show_details()
elif self.choice == 0:
return self.exit_menu()
def get_choice(self):
"""Function to get a valid choice from the menu
Args:
None
"""
choice = raw_input('Enter your selection please: ')
if not choice.isdigit():
print("Please input a valid selection.")
return self.get_choice()
choice = int(choice)
if choice < 0 or choice > 4:
print("Please select a valid choice between 0 and 4")
return self.get_choice()
return choice
def load_state(self):
"""Function to load the current state of the bank
Args:
None
"""
return pickle.load(open(os.path.abspath('data/bank.pkl'), 'r'))
def save_state(self):
"""Function to save the current state of the bank
Args:
None
"""
pickle.dump(self.bank, open(os.path.abspath('data/bank.pkl'), 'w'))
def run_menu(self):
"""Function to run the menu
Args:
None
"""
self.print_header()
while not self.exit:
self.print_menu()
self.choice = self.get_choice()
self.perform_action()
self.save_state()
def run(self):
self.run_menu()
