Python学习笔记8：银行管理系统
自己写的练手小程序，练习面向对象的概念，代码中都有注释，刚学的同学也可以很容易看懂（需要自己用pickle模块新建一个database.txt文件，用来存储用户数据信息）。
# atm.py
# 各种操作类
import random
from card import Card
from user import User
import time
class ATM(object):
# 初始化
def __init__(self, allUsersInfo):
# 将所有用户信息传递过来，以便后续操作
self.allUsersInfo = allUsersInfo # 卡号 ：用户
# 确认密码
def checkPwd(self, realPwd):
myList = [3, 2, 1, 0]
for i in myList:
# 如果密码输入错误超过三次
if i == 0:
return False
rePass = input('请再次输入密码：')
if realPwd == rePass:
return True
elif (i - 1) != 0:
print('输入错误，你还有%d次输入的机会！' % (i - 1))
# 随机生成开户卡号
def randomId(self):
while True:
# 存储卡号
str = ''
# 随机生成6位卡号
for i in range(6):
ch = chr(random.randrange(ord('0'), ord('9') + 1))
str += ch
# 判断卡号是否重复
if not self.allUsersInfo.get(str):
return str
# 开卡操作
def createUser(self):
try:
name = input('请输入姓名：') # 用户姓名
idCard = input('请输入身份证号码：') # 用户身份证号
phone = input('请输入电话号码：') # 用户的电话号码
preMoney = int(input('请输入预存金额：')) # 用户的预存金额
# 预存金额少于0则开户失败
if preMoney < 0:
print('预存金额有误，开户失败！')
return -1
# 设置密码
onePwd = input('请输入密码：') # 第一次密码
twoPwd = self.checkPwd(onePwd)
if not twoPwd:
print('确认密码错误，开户失败！')
return -1
# 以上全部完成，则开户所需的信息准备齐全，开户成功
# 系统生成开户卡号
# 生成卡号
cardStr = self.randomId()
# 创建一个卡的实例，用于存储卡的信息
card = Card(cardStr, onePwd, preMoney)
# 创建一个用户实例，用于存储用户的信息
user = User(name, idCard, phone, card)
# 按 卡号 ： 用户信息 键值对的形式存入字典中
self.allUsersInfo[cardStr] = user
# 提示用户，已经开户成功
time.sleep(1)
print('开户成功，请牢记卡号({})...'.format(cardStr))
except Exception as e:
print('抱歉，系统出现故障，暂时无法操作！')
return
# 判断卡号是否存在
def isExistence(self, cardStrInp):
# 卡号不存在，直接退回到选项界面
if self.allUsersInfo.get(cardStrInp):
return True
# 卡号不存在的情况下
else:
print('卡号不存在！')
return False
# 做查询等操作时，输入密码
def secretOption(self, cardStrInp):
myList = [3, 2, 1, 0]
for i in myList:
# 如果密码输入错误超过三次
if i == 0:
return False
passInp = input('请输入密码：')
if passInp == self.allUsersInfo[cardStrInp].cardInfo.cardPwd:
return True
elif (i - 1) != 0:
print('输入错误，你还有%d次输入的机会！' % (i - 1))
# 查询用户信息
def searchUserInfo(self):
cardStrInp = input('请输入卡号：') # 用户想要查询的卡号
# 判断卡号的状态
isExist = self.isExistence(cardStrInp)
# 卡号存在
if isExist:
# 是否被锁定
if not self.allUsersInfo[cardStrInp].cardInfo.isLock:
# 输入密码并判断
# 密码正确则输出余额
if self.secretOption(cardStrInp):
print('账号：%s 余额：%d' % (cardStrInp, self.allUsersInfo[cardStrInp].cardInfo.cardMoney))
return 0
# 密码错误则锁定卡号
else:
print('密码输入错误超过3次，卡号已被锁定，请先解锁！')
self.allUsersInfo[cardStrInp].cardInfo.isLock = True
return -1
else:
print('该卡号已被锁定，请先解锁！')
return -1
# 取钱
def getMoney(self):
cardStrInp = input('请输入卡号：')
isExist = self.isExistence(cardStrInp)
# print(isExist)
# 如果卡号存在
if isExist:
# 如果没被锁定
if not self.allUsersInfo[cardStrInp].cardInfo.isLock:
# 如果密码正确
if self.secretOption(cardStrInp):
getMoney = int(input('请输入取款金额：'))
# 如果取款金额超出余额
if getMoney > self.allUsersInfo[cardStrInp].cardInfo.cardMoney:
print('余额不足，取款失败！')
return -1
# 余额足够的情况，取款成功
else:
self.allUsersInfo[cardStrInp].cardInfo.cardMoney -= getMoney
print('取款成功，卡上余额：%d' % self.allUsersInfo[cardStrInp].cardInfo.cardMoney)
return 0
else:
print('密码输入错误超过3次，卡号已被锁定，请先解锁！')
self.allUsersInfo[cardStrInp].cardInfo.isLock = True
return -1
else:
print('该卡号已被锁定，请先解锁！')
return -1
# 存钱
def saveMoney(self):
cardStrInp = input('请输入卡号：')
isExist = self.isExistence(cardStrInp)
if isExist:
# 如果没被锁定
if not self.allUsersInfo[cardStrInp].cardInfo.isLock:
# 如果密码正确
if self.secretOption(cardStrInp):
saveMoney = int(input('请输入存款金额：'))
if saveMoney <= 0:
print('无效的存款金额！')
return -1
else:
# 存款成功
self.allUsersInfo[cardStrInp].cardInfo.cardMoney += saveMoney
print('存款成功，卡内余额为%d！' % self.allUsersInfo[cardStrInp].cardInfo.cardMoney)
else:
print('密码输入错误超过3次，卡号已被锁定，请先解锁！')
self.allUsersInfo[cardStrInp].cardInfo.isLock = True
return -1
else:
print('该卡号已被锁定，请先解锁！')
return -1
# 转账
def transMoney(self):
cardInp = input('请输入卡号：')
isExist = self.isExistence(cardInp)
# 卡号存在
if isExist:
# 没被锁定
if not self.allUsersInfo[cardInp].cardInfo.isLock:
# 密码正确
if self.secretOption(cardInp):
transCardInp = input('请输入要转账的卡号：')
isExistTrans = self.isExistence(transCardInp)
# 要转账的卡号存在
if isExistTrans:
# 要转账的卡号没被锁定
if not self.allUsersInfo[transCardInp].cardInfo.isLock:
# 要转账的卡号不是自己的卡号
if transCardInp != cardInp:
transMoneyInp = int(input('请输入要转账的金额：'))
# 判断转账金额是否大于卡内余额
if transMoneyInp > self.allUsersInfo[cardInp].cardInfo.cardMoney:
print('余额不足，转账失败！')
return -1
else:
self.allUsersInfo[cardInp].cardInfo.cardMoney -= transMoneyInp
self.allUsersInfo[transCardInp].cardInfo.cardMoney += transMoneyInp
print('转账成功，卡内余额为%d!' % self.allUsersInfo[cardInp].cardInfo.cardMoney)
else:
print('无法转入自己的账户！')
return -1
else:
print('对方卡号已被锁定，无法转账，请先解锁！')
return -1
else:
print('密码输入错误超过3次，卡号已被锁定，请先解锁！')
self.allUsersInfo[cardInp].cardInfo.isLock = True
return -1
else:
print('该卡号已被锁定，请先解锁！')
return -1
# 改密
def changePwd(self):
cardStrInp = input('请输入卡号：')
isExist = self.isExistence(cardStrInp)
if isExist:
# 如果没被锁定
if not self.allUsersInfo[cardStrInp].cardInfo.isLock:
# 如果密码正确
if self.secretOption(cardStrInp):
# 输入修改后的密码
changePwd = input('请输入新密码：')
# 不允许新密码和旧密码相同
if changePwd != self.allUsersInfo[cardStrInp].cardInfo.cardPwd:
myList = [3, 2, 1, 0]
for i in myList:
# 如果密码输入错误超过三次
if i == 0:
print('确认密码错误次数超过3次，修改密码失败！')
return -1
passInp = input('请确认新密码：')
# 如果新密码确认成功，则修改密码
if passInp == changePwd:
print('密码修改成功！')
self.allUsersInfo[cardStrInp].cardInfo.cardPwd = changePwd
return 0
elif (i - 1) != 0:
print('输入错误，你还有%d次输入的机会！' % (i - 1))
# 新密码和旧密码相同，不允许修改，提示并退出
else:
print('新密码不允许和旧密码相同！')
return -1
# 密码不正确
else:
print('密码输入错误超过3次，卡号已被锁定，请先解锁！')
self.allUsersInfo[cardStrInp].cardInfo.isLock = True
return -1
else:
print('该卡号已被锁定，请先解锁！')
return -1
# 解锁
def unlockUser(self):
cardStrInp = input('请输入卡号：')
isExist = self.isExistence(cardStrInp)
# 存在
if isExist:
# 被锁定
if self.allUsersInfo[cardStrInp].cardInfo.isLock:
# 如果密码正确
if self.secretOption(cardStrInp):
# 输入身份证号信息做确认
idCardInp = input('请输入身份证号：')
# 身份证号信息匹配
if idCardInp == self.allUsersInfo[cardStrInp].idCard:
# 解锁成功
print('解锁成功！')
self.allUsersInfo[cardStrInp].cardInfo.isLock = False
return 0
else:
print('身份证号信息不匹配，解锁失败！')
return -1
# 密码不正确
else:
print('密码输出错误，解锁失败！')
return -1
else:
print('该卡号未被锁定，无需解锁！')
return -1# card.py
# 卡类
class Card(object):
def __init__(self, cardStr, cardPwd, cardMoney):
self.cardStr = cardStr # 卡号
self.cardPwd = cardPwd # 卡密
self.cardMoney = cardMoney # 卡中余额
self.isLock = False # 卡号是否被锁# main.py
# 主函数
from view import View
import pickle
from atm import ATM
import time
def main():
# 定义管理员对象
admin = View()
# 管理员登录界面
admin.printAdminView()
# 如果管理员账号、密码输入错误则退出
if admin.adminOption():
return -1
# 读取存储在文件中的所有用户的信息
file = open('database.txt', 'rb')
allUsersInfo = pickle.load(file) # 将所有用户信息存储到 allUsersInfo中去
file.close() # 关闭文件
# 打印所有用户信息，以便验证
print('*' * 10)
print('usersInfo:', allUsersInfo)
# 创建atm实例，并将所有用户信息传递过去，进行初始化
atm = ATM(allUsersInfo)
# 循环进行操作选项
while True:
# 调用功能显示方法，显示功能选项
admin.printSysFunctionView()
# 提示用户输入功能选项进行操作
option= input('请输入操作选项（数字）:')
# 对用户的操作选项进行判断
if option == "1": # 开户操作
atm.createUser()
elif option == "2": # 查询余额操作
atm.searchUserInfo()
elif option == "3": # 取钱操作
atm.getMoney()
elif option == "4": # 存钱操作
atm.saveMoney()
elif option == "5": # 转账操作
atm.transMoney()
elif option == "6": # 改密码操作
atm.changePwd()
elif option == "7": # 解锁操作
atm.unlockUser()
# 以下操作暂时未写
# elif option == "q": # 退出操作
# # atm.quitSys()
# print('退出')
# elif option == "9": # 补卡操作
# print("补卡")
#
# elif option == "0": # 销户操作
# print("销户")
elif option == "8": # 退出系统
return 0
# 如果用户输入的不是上述操作
else:
print('无效的操作选项！')
return -1
# 将新信息写入文件
# 将当前系统中的用户信息保存到文件中
f = open('database.txt', "wb")
pickle.dump(allUsersInfo, f)
f.close()
if __name__ == '__main__':
main()# user.py
# 用户信息类
class User(object):
def __init__(self, name, idCard, phone, cardInfo):
self.name = name # 用户姓名
self.idCard = idCard # 用户身份证号
self.phone = phone # 用户的手机号码
self.cardInfo = cardInfo # 用户的卡的信息（卡号、密码、余额）# view.py
# 视图类
import time
class View(object):
admin = '1'
pwd = '1'
# 管理员登录界面
def printAdminView(self):
print("*****************************************************************")
print("* *")
print("* *")
print("* 欢迎登陆丹丹的银行 *")
print("* *")
print("* *")
print("*****************************************************************")
def printSysFunctionView(self):
print("*****************************************************************")
print("* 开户(1) 查询(2) *")
print("* 取款(3) 存款(4) *")
print("* 转账(5) 改密(6) *")
print("* 解锁(7) 退出(8) *")
# print("* 补卡(9) 销户(0) *")
# print("* 退出(t) *")
print("*****************************************************************")
def adminOption(self):
inputAdmin = input('请输入管理员账号：')
if inputAdmin != self.admin:
print('账号错误！')
return -1
inputPwd = input('请输入管理员密码：')
if inputPwd != self.pwd:
print('密码错误！')
return -1
# 管理员账号和密码全部正确
print('操作成功，请稍后...')
time.sleep(2)
return 0
# view = View()
# view.printAdminView()
# view.adminOption()
# view.printSysFunctionView()