def main(): allUserInfo = {} try: f = open("peoplelist.txt","rb") if len(f.read())>0: f.seek(0,0) allUserInfo = pickle.load(f) except Exception as msg: print(msg) f = open("peoplelist.txt","wb") print(msg) finally: f.close() adminview = View() adminview.printSysUI() atm = Atm(allUserInfo) #选择角色 print("请选择角色:") print("1:管理员") print("2:普通用户") roleNum = input() #角色序号 if roleNum == '1': if not adminview.adminlogin(): print("管理员登录失败") return time.sleep(1) print("管理员登录成功!请稍候...") elif roleNum == '2': #验证卡号是否存在 while True: inputCard = input("请输入您的卡号:") isExist = atm.isExistCard(inputCard) if isExist: break #验证密码是否正确 if not atm.isExistPwd(inputCard): #文件中写入所有用户信息 f = open("peoplelist.txt","wb") pickle.dump(atm.allUserInfo,f) f.close() return while True: adminview.printUI(roleNum) inputNum = input("请输入操作选项(数字):") if inputNum == '1' and roleNum == '1': # print(atm.allUserInfo) atm.createUser() print(atm.allUserInfo) # allUserInfo.clear() elif inputNum == '2' and roleNum == '2': # atm.outMoney(inputCard) if not atm.outMoney(inputCard): print("取款失败") elif inputNum == '3' and roleNum == '2': atm.inMoney(inputCard) elif inputNum == '4' and roleNum == '2': atm.transMoney(inputCard) elif inputNum == '5' and roleNum == '2': atm.findMoney(inputCard) elif inputNum == '6' and roleNum == '2': if not atm.updatePwd(inputCard): print("修改密码失败") return print("修改密码成功") elif inputNum == '7' and roleNum == '1': if atm.unstatus(): print("解锁成功!") f = open("peoplelist.txt","wb") pickle.dump(atm.allUserInfo,f) f.close() return elif inputNum == '0': exit() else: print("您操作有误,请重新操作!") #文件中写入所有用户信息 f = open("peoplelist.txt","wb") pickle.dump(atm.allUserInfo,f) f.close()
def setUp(self): self.atm = Atm() self.atm.enter_pin(777) self.atm2 = Atm()
def setUp(self): self.term = Atm()
def setUp(self): self.term = Atm() self.term.enter_pin(777)
def main(): atm = Atm() atm.run()