#link to other classes by importing
from person import *
from account import *
#import needed libaries
import pickle
import tkinter as tk
from tkinter import messagebox
class BankSystem(object):
def __init__(self):
self.customerList = []
self.adminList = []
self.loadBankData() #used to received bank data when testing
self.save() #saves loadBankData into a file
self.load() #loads bank data from a file
def save(self):
#put customer and admin list into another list
data1 = (self.adminList, self.customerList)
#makes a file and writes bank data into it
output = open('bankData.pkl', 'wb')
pickle.dump(data1, output)
output.close()
def load(self):
#loads bank data from another file, then closes that file
pkl_file = open('bankData.pkl', 'rb')
data = pickle.load(pkl_file)
pkl_file.close()
#set the admin and customer list
self.adminList = data[0]
self.customerList = data[1]
def loadBankData(self):
#create customers and add them to the cusomterList
#create specific accounts for customers
#used to reset data when testing
customer_1 = Customer("amin", "1234", ["14", " Street2", "tabriz" , "hesabjari"])
account_no = 90455673
account_1 = hesabjari(account_no, "hesabjari", "1234", 5000)
customer_1.openAccount(account_1)
self.customerList.append(customer_1)
customer_2 = Customer("Davood", "password", ["60", "Street1", "tehran" , "hesabboland"])
account_no += 1
account_2 = hesabjari(account_no, "hesabjari", "2601", 3200)
account_no += 1
account_3 = hesabboland(account_no, "hesabboland", "2601", 600)
customer_2.openAccount(account_2)
customer_2.openAccount(account_3)
self.customerList.append(customer_2)
customer_3 = Customer("Niloo", "moonlight", ["5", "Street3", "tabriz", "hesabkotah"])
account_no += 1
account_4 = hesabkotah(account_no, "hesabkotah", "1010", 18000)
customer_3.openAccount(account_4)
self.customerList.append(customer_3)
customer_4 = Customer("Ali", "150A",["44", "nishan tashi", "istanbul", "hesabjari"])
account_no+= 1
account_5 = hesabjari(account_no, "hesabjari", "6666", 50)
customer_4.openAccount(account_5)
self.customerList.append(customer_4)
#create admins and add them to admins_list
admin_1 = Admin("ahora", "1441", True, ["12", "maqsodiye", "tehran" , ""])
self.adminList.append(admin_1)
def customerLogin(self, name, password):
#check the data inputed
foundCustomer = self.searchCustomersByName(name)
if foundCustomer == None:
return("The customer has not been found! \n")
else:
#make sure the correct password is entered
if (foundCustomer.checkPassword(password) == True):
self.runCustomerOptions(foundCustomer)
else:
return("you have input a wrong password \n")
def searchCustomersByName(self, customerName):
#finds a customer from the customers_list and compares wuth inputted data
foundCustomer = None
for a in self.customerList:
name = a.getName()
if name == customerName:
foundCustomer = a
break
if foundCustomer == None:
print("The customer %s does not exist! Try again...\n" %customerName)
#returns the found_customer to be used customer_login
return foundCustomer
def mainMenu(self):
#print the options you have
print()
print()
print ("~~~~~~~~~~~~~~~~~~~~~~~~~~~~~")
print ("Welcome to the EN Simple Bank System")
print ("~~~~~~~~~~~~~~~~~~~~~~~~~~~~~")
print ("1) Admin login")
print ("2) Customer login")
print ("3) Quit")
print (" ")
while True:
try:
option = int(input("\nChoose your option: "))
break
except ValueError:
print("Option not valid. Try again. \n")
return option
def runMainOption(self):
#options when code first runs
loop = 1
while loop == 1:
choice = self.mainMenu()
if choice == 1:
name = input("\nPlease input admin name: ")
password = input("\nPlease input admin password: ")
#calls admin_login to check the data inputted
msg = self.adminLogin(name, password)
print(msg)
elif choice == 2:
name = input("\nPlease input customer name: ")
password = input("\nPlease input customer password: ")
#calls customer_login to check data inputted
msg = self.customerLogin(name, password)
print(msg)
elif choice == 3:
#uses a GUI for a message box to confirm if the user wants to quit
root = tk.Tk()
if messagebox.askyesno("Quit", "Are you sure you want to quit?"):
root.destroy() #gets rid of blank GUI box the tkinter makes
self.save() #saves all changes
loop = 0 #quits
else:
print("Option not valid \n")
print ("Thank you for stopping by the bank! \n")
def customerMenu(self, customerName):
#print the options you have when logged in as a customer
print (" ")
print ("Welcome %s : Your transaction options are:" %customerName)
print ("~~~~~~~~~~~~~~~~~~~~~~~~~~~~~")
print ("1) Account operations")
print ("2) Profile settings")
print ("3) Sign out")
print (" ")
while True:
try:
option = int(input("\nChoose your option: "))
break
except ValueError:
print("Option not valid. Try again.")
return option
def runCustomerOptions(self, customer):
loop = 1
while loop == 1:
choice = self.customerMenu(customer.getName())
if choice == 1:
accName = input("\nPlease input name of account: ")
account = customer.searchAccountByName(accName)
if account == None:
print ("Account could not be found \n")
else:
account.runAccountOptions(customer, self)
elif choice == 2:
customer.runProfileOptions()
elif choice == 3:
loop = 0
print ("Exit account operations \n")
def adminLogin(self, name, password):
#checks admin login same as for customers
foundAdmin = self.searchAdminByName(name)
if foundAdmin== None:
return("The admin has not been found! \n")
else:
if (foundAdmin.checkPassword(password) == True):
self.runAdminOptions(foundAdmin)
else:
return("you have input a wrong password \n")
def searchAdminByName(self, adminName):
#searchs for admins using admin list
foundAdmin = None
for a in self.adminList:
name = a.getName()
if name == adminName:
foundAdmin = a
break
if foundAdmin == None:
print("The admin %s does not exist! Try again...\n" %adminName)
return foundAdmin
def interest(self):
#allows admins to manually apply interest to all accounts
for customer in
