コード例 #1
0
	def __init__(self):
		self.mymenu=Views()
		self.myclientcsv=CSV('clientlist')
		self.mybankercsv=CSV('bankerlist')
コード例 #2
0
class Controller:
	def __init__(self):
		self.mymenu=Views()
		self.myclientcsv=CSV('clientlist')
		self.mybankercsv=CSV('bankerlist')

	def checkownnameandpass(self):
		self.mymenu.firstmenu()
		if (self.mymenu.neworexuser=='n'):
			return self.createnewuser()
		elif (self.mymenu.neworexuser=='e'):

			countfound=0
			countwrongpass=0
			if (self.mymenu.userrole=='b'):
				self.mybankercsv.openreader()
				with self.mybankercsv.file:
					self.mybankercsv.dictreader()
					for row in self.mybankercsv.reader:
						print (row)
						if (self.mymenu.ownname==row['Username'] and self.mymenu.ownpassword==row['Password']):
							countfound+=1
						elif (self.mymenu.ownname==row['Username'] and self.mymenu.ownpassword!=row['Password']):
							countwrongpass+=1
					if (countfound==0 and countwrongpass==0 ):
						print ("you are not an existing user in our bank, please create a new banker account first!")
						return self.createnewuser()

					elif (countfound==0 and countwrongpass >0 ):
						print ("you entered the wrong banker account password, please reenter your banker account username and password!")
						return self.checkownnameandpass()
					elif (countfound>0 and countwrongpass==0):
						print ("you are now logged into your banker account!")
						return self.actions()
			elif (self.mymenu.userrole=='c'):
				self.myclientcsv.openreader()
				with self.myclientcsv.file:
					self.myclientcsv.dictreader()
					for row in self.myclientcsv.reader:
						if (self.mymenu.ownname==row['Username'] and self.mymenu.ownpassword==row['Password']):
							countfound+=1
						elif (self.mymenu.ownname==row['Username'] and self.mymenu.ownpassword!=row['Password']):
							countwrongpass+=1
					if (countfound==0 and countwrongpass==0 ):
						print ("you are not an existing user in our bank, please have a banker create an account for youn first!")
						return self.checkownnameandpass()

					elif (countfound==0 and countwrongpass >0 ):
						print ("you entered the wrong client account password, please reenter your client account username and password!")
						return self.checkownnameandpass()
					elif (countfound>0 and countwrongpass==0):
						print ("you are now logged into your client account!")
						return self.actions()

		


	def checkclientname(self):
		if (self.mymenu.action=='1c' or self.mymenu.action=='2c'  or self.mymenu.action=='4c' or self.mymenu.action=='1b'):
			return True
		elif (self.mymenu.action=='3c'):
			return self.checkothername()
		else:
			countfound=0
			self.myclientcsv.openreader()
			with self.myclientcsv.file:
				self.myclientcsv.dictreader()
				for row in self.myclientcsv.reader:
					if (self.mymenu.clientname==row['Username'] ):
						countfound+=1
			
				if (countfound==0):
					print ("the client username is not an existing client in our bank!")
					return False

				elif (countfound>0):
					print ("your client is found in our database!")
					if (self.mymenu.action=='4b'):
						return self.checkothername()
					elif (self.mymenu.action!='4b' and self.mymenu.action!='3c' and self.mymenu.action!='1c' and self.mymenu.action!='2c'):
						return True


		
	def checkothername(self):
		countfound=0
		self.myclientcsv.openreader()
		with self.myclientcsv.file:
			self.myclientcsv.dictreader()
			for row in self.myclientcsv.reader:
				if (self.mymenu.otherclientname==row['Username'] ):
					countfound+=1
			
			if (countfound==0):
				print ("the client username is not an existing client in our bank!")
				return False

			elif (countfound>0):
				print ("your client is found in our database!")
				return True

		
	def createnewuser(self):

		if (self.mymenu.newuserchoice=='b'):
			countfound=0
			self.mybankercsv.openreader()
			with self.mybankercsv.file:
				self.mybankercsv.dictreader()
				for row in self.mybankercsv.reader:
					
					if (self.mymenu.newusername==row['Username'] ):
						countfound+=1
				if (countfound>0):
					print ("this banker name already exists in our system, try to log in as an existing user!")
					return self.checkownnameandpass()
				elif (countfound==0):
					bankerlist.append(Banker(self.mymenu.newusername, self.mymenu.newuserpassword))
					global bankernum
					bankernamedic[self.mymenu.newusername]=bankerlist[bankernum]
					bankernum+=1
					self.mybankercsv.openwriter()
					with self.mybankercsv.file:
						self.mybankercsv.dictwriter()
						self.mybankercsv.writer.writerow({'Username':self.mymenu.newusername, 'Password':self.mymenu.newuserpassword})
						self.mybankercsv.close()
					mydatabase.connect()
					mydatabase.cursor()
					mydatabase.curs.execute("INSERT INTO Bankers(Username, Password, Time_Account_Created)"" VALUES(?,?,?)", (bankernamedic[self.mymenu.newusername].username, bankernamedic[self.mymenu.newusername].password, bankernamedic[self.mymenu.newusername].time_created))
					mydatabase.commit()
					mydatabase.close_connection()

					print ("your new banker account has been successfully created, please log in as an existing user now. your username is "+self.mymenu.newusername)	
					return self.checkownnameandpass()


	def actions(self):
		self.mymenu.whattodo()
		mycheckclient=self.checkclientname()
		while(mycheckclient==False):
			self.mymenu.getinfo()

		if (self.mymenu.action=='1b'):
			
			bankernamedic[self.mymenu.ownname].createaccount(self.mymenu.accounttype, self.mymenu.clientbalance, self.mymenu.clientname, self.mymenu.clientpassword, self.mymenu.newcreateaccountnum)
			mydatabase.connect()
			mydatabase.cursor()
			mydatabase.curs.execute("INSERT INTO Clients(Username, Password, Time_Account_Created)"" VALUES(?,?,?)", (self.mymenu.clientname, self.mymenu.clientpassword, usernamedic[self.mymenu.clientname].time_created))
			mydatabase.curs.execute("INSERT INTO Client_Accounts(Accountnumber, Balance, Account_Type, Client_Username)"" VALUES(?,?,?,?)", (self.mymenu.newcreateaccountnum, self.mymenu.clientbalance, self.mymenu.accounttype, self.mymenu.clientname))
			mydatabase.commit()
			mydatabase.close_connection()
			mylogout=self.mymenu.logout()
			if (mylogout==False):
				return self.actions()
			elif (mylogout==True):
				return self.checkownnameandpass()
		elif (self.mymenu.action=='2b'):
			
			clientnewbal=bankernamedic[self.mymenu.ownname].deposit(float(self.mymenu.clientamount),self.mymenu.clientname ,int(self.mymenu.clientaccountnum) )
			
			mydatabase.connect()
			mydatabase.cursor()
			mydatabase.curs.execute("UPDATE Client_Accounts SET Balance= ? WHERE Accountnumber=?",(clientnewbal, int(self.mymenu.clientaccountnum)))
			mydatabase.commit()
			mydatabase.close_connection()
			mylogout=self.mymenu.logout()
			if (mylogout==False):
				return self.actions()
			elif (mylogout==True):
				return self.checkownnameandpass()

		elif (self.mymenu.action=='3b'):
		
			clientnewbal=bankernamedic[self.mymenu.ownname].withdraw(float(self.mymenu.clientamount),self.mymenu.clientname ,int(self.mymenu.clientaccountnum))
			mydatabase.connect()
			mydatabase.cursor()
			mydatabase.curs.execute("UPDATE Client_Accounts SET Balance= ? WHERE Accountnumber=?",(clientnewbal, int(self.mymenu.clientaccountnum)))
			mydatabase.commit()
			mydatabase.close_connection()

			mylogout=self.mymenu.logout()
			if (mylogout==False):
				return self.actions()
			elif (mylogout==True):
				return self.checkownnameandpass()
		elif (self.mymenu.action=='4b'):
			iuser=bankernamedic[self.mymenu.ownname]
			clientnewbal=iuser.transfer(float(self.mymenu.clientamount), self.mymenu.clientname, self.mymenu.otherclientname, int(self.mymenu.clientaccountnum), int(self.mymenu.clientaccountnum2))
			otherusernewbal=usernamedic[self.mymenu.otherclientname].accountsdic[int(self.mymenu.clientaccountnum2)].balance
			mydatabase.connect()
			mydatabase.cursor()
			mydatabase.curs.execute("UPDATE Client_Accounts SET Balance= ? WHERE Accountnumber=?",(clientnewbal, int(self.mymenu.clientaccountnum)))
			mydatabase.curs.execute("UPDATE Client_Accounts SET Balance= ? WHERE Accountnumber=?",(otherusernewbal, int(self.mymenu.clientaccountnum2)))
			mydatabase.commit()
			mydatabase.close_connection()

			mylogout=self.mymenu.logout()
			if (mylogout==False):
				return self.actions()
			elif (mylogout==True):
				return self.checkownnameandpass()
		elif (self.mymenu.action=='5b'):
			iuser=bankernamedic[self.mymenu.ownname]
			iuser.displayuseraccounts(self.mymenu.clientname)
			mylogout=self.mymenu.logout()
			if (mylogout==False):
				return self.actions()
			elif (mylogout==True):
				return self.checkownnameandpass()
		elif (self.mymenu.action=='1c'):
			iuser=usernamedic[self.mymenu.ownname]
			mynewbal=iuser.deposit(float(self.mymenu.amount), int(self.mymenu.accountnum))
			mydatabase.connect()
			mydatabase.cursor()
			mydatabase.curs.execute("UPDATE Client_Accounts SET Balance= ? WHERE Accountnumber=?",(mynewbal, int(self.mymenu.accountnum)))
			mydatabase.commit()
			mydatabase.close_connection()

			mylogout=self.mymenu.logout()
			if (mylogout==False):
				return self.actions()
			elif (mylogout==True):
				return self.checkownnameandpass()
		elif(self.mymenu.action=='2c'):
		 	iuser=usernamedic[self.mymenu.ownname]
		 	mynewbal=iuser.withdraw(float(self.mymenu.amount), int(self.mymenu.accountnum))
		 	mydatabase.connect()
		 	mydatabase.cursor()
		 	mydatabase.curs.execute("UPDATE Client_Accounts SET Balance=? WHERE Accountnumber=?",(mynewbal, int(self.mymenu.accountnum)))
		 	mydatabase.commit()
		 	mydatabase.close_connection()
		 	mylogout=self.mymenu.logout()
		 	if (mylogout==False):
		 		return self.actions()
		 	elif (mylogout==True):
		 		return self.checkownnameandpass()
			
		elif(self.mymenu.action=='3c'):
			iuser=usernamedic[self.mymenu.ownname]
			mynewbal=iuser.transfer( float(self.mymenu.amount),self.mymenu.otherclientname , int(self.mymenu.accountnum),int(self.mymenu.accountnum2))
			otherusernewbal=usernamedic[self.mymenu.otherclientname].accountsdic[int(self.mymenu.accountnum2)].balance
			mydatabase.connect()
			mydatabase.cursor()
			mydatabase.curs.execute("UPDATE Client_Accounts SET Balance= ? WHERE Accountnumber=?",(mynewbal, int(self.mymenu.accountnum)))
			mydatabase.curs.execute("UPDATE Client_Accounts SET Balance= ? WHERE Accountnumber=?",(otherusernewbal, int(self.mymenu.accountnum2)))
			mydatabase.commit()
			mydatabase.close_connection()

			mylogout=self.mymenu.logout()
			if (mylogout==False):
				return self.actions()
			elif (mylogout==True):
				return self.checkownnameandpass()

		elif (self.mymenu.action=='4c'):
			iuser=usernamedic[self.mymenu.ownname]
			iuser.displayaccounts()
			mylogout=self.mymenu.logout()
			if (mylogout==False):
				return self.actions()
			elif (mylogout==True):
				return self.checkownnameandpass()