import os
import pickle
# 用户的类
class User:
def __init__(self, uid, name, card):
self.uid = uid # 身份证
self.name = name # 用户名
self.card = card # 银行卡
def __str__(self):
return '姓名:{},身份证:{},卡号:{}'.format(self.name, self.uid, self.card.cid)
# 将用户对象保存到文件中
@staticmethod
def save_user(userinfo):
pathname = os.path.join(os.getcwd(), 'userinfo.db')
with open(pathname, 'wb') as fp:
pickle.dump(userinfo, fp)
# 从文件中加载数据重新转换为用户对象
@staticmethod
def load_user():
pathname = os.path.join(os.getcwd(), 'userinfo.db')
if os.path.exists(pathname):
with open(pathname, 'rb') as fp:
userinfo = pickle.load(fp)
return userinfo
else:
return {}
# 管理的类
class Admin:
def __init__(self, account='admin', password='123456'):
self.account = account
self.password = password
# 显示欢迎页面
def welcome(self):
print('*' * 30)
print(' ' * 5 + '欢迎使用xx银行管理系统')
print('*' * 30)
# 登录页面
def login(self):
account = input('请输入管理员账号:')
password = input('请输入管理员密码:')
if account == 'admin' and password == '123456':
return True
return False
# 操作菜单
def menu(self):
print('*' * 30)
print('开户[1] 销户[2] 查询[3] 存款[4]')
print('取款[5] 转账[6] 该密[7] 锁定[8]')
print('解锁[9] 退出[0] 显示所有用户[10]')
print('*' * 30)
# 银行卡的类
class Card:
def __init__(self, cid, pwd):
self.cid = cid # 卡号
self.pwd = pwd # 密码
self.money = 0 # 余额
self.is_lock = False # 是否锁定
from user import User
from card import Card
from helper import Helper
# 操作的类
class Operate:
def __init__(self, userinfo={}):
# 用户保存所有的用户信息
self.userinfo = userinfo
def new_user(self):
name = input('请输入用户名:')
uid = input('请输入身份证:')
pwd = input('请输入密码:')
# 加密密码
pwd = Helper.generate_password_hash(pwd)
# 先创建银行卡对象
# 生成卡号
cid = Helper.generate_cid()
card = Card(cid, pwd)
user = User(uid, name, card)
# 保存到字典中
self.userinfo[cid] = user
# 保存所有用户信息
User.save_user(self.userinfo)
print('开户成功')
def del_user(self):
print('销户')
def query_money(self):
cid = input('请输入您的卡号:')
user = self.userinfo.get(cid)
if user:
# 查看是否已经锁定
if user.card.is_lock:
print('该卡已锁定,请先解锁')
return
print('账户余额:¥{}元'.format(user.card.money))
else:
print('无效的卡号')
def get_money(self):
print('取钱成功')
def save_money(self):
cid = input('请输入您的卡号:')
user = self.userinfo.get(cid)
if user:
# 查看是否已经锁定
if user.card.is_lock:
print('该卡已锁定,请先解锁')
return
# 记录密码错误次数
count = 0
while count < 3:
pwd = input('请输入您的密码:')
# 校验密码
if Helper.check_password_hash(pwd, user.card.pwd):
money = input('请输入您要存的金额:')
user.card.money += int(money)
User.save_user(self.userinfo)
print('存款成功')
break
else:
count += 1
print('密码有误,请重新输入')
else:
user.card.is_lock = True
User.save_user(self.userinfo)
print('密码错误已达上限,该卡已锁定')
else:
print('无效的卡号')
# 转账
def pass_money(self):
cid = input('请输入您的卡号:')
user = self.userinfo.get(cid)
if user:
pwd = input('请输入您的密码:')
if Helper.check_password_hash(pwd, user.card.pwd):
dst_cid = input('请输入收款人账号:')
dst_user = self.userinfo.get(dst_cid)
if dst_user:
money = int(input('请输入您的转账金额:'))
# 判断账户余额是否充足
if user.card.money >= money:
user.card.money -= money
dst_user.card.money += money
User.save_user(self.userinfo)
print('转账成功')
else:
print('余额不足,请及时充值')
else:
print('收款人信息有误')
else:
print('密码错误')
else:
print('无效的卡号')
def change_pwd(self):
print('修改密码成功')
# 变相已经帮大家实现了
def lock_user(self, user):
print('用户锁定')
def unlock_user(self):
cid = input('请输入您的卡号:')
user = self.userinfo.get(cid)
if user:
uid = input('请出示您的身份证:')
if uid == user.uid:
user.card.is_lock = False
User.save_user(self.userinfo)
print('解锁成功')
else:
print('身份证是不是带错了@_@')
else:
print('卡是不是折了:卡号错了')
def show_users(self):
for user in self.userinfo:
print(self.userinfo[user])
from random import randint
import hashlib
class Helper:
@staticmethod
# 生成随机的银行卡号
def generate_cid(length=8):
cid = ''
for i in range(length):
cid += str(randint(0, 9))
return cid
# 生成md5加密后的内容
@staticmethod
def generate_password_hash(password):
# 创建一个md5对象
m = hashlib.md5()
# 设置加密的原始内容
m.update(password.encode('utf-8'))
# 返回加密后的字符串
return m.hexdigest()
# 校验使用md5加密的密码
@staticmethod
def check_password_hash(pwd, pwd_hash):
m = hashlib.md5()
m.update(pwd.encode('utf-8'))
return m.hexdigest() == pwd_hash
if __name__ == '__main__':
print(Helper.generate_password_hash('123456'))
from user import User
from admin import Admin
from operate import Operate
if __name__ == '__main__':
# 创建管理员对象
admin = Admin()
# 定义变量,记录密码错误次数
count = 0
while True:
# 展示欢迎页面
admin.welcome()
# 展示登录页面
ret = admin.login()
if ret:
print('登录成功')
# 加载用户信息
userinfo = User.load_user()
# 创建操作对象
op = Operate(userinfo)
# 定义变量,记录是否退出操作
is_quit = False
while True:
# 显示操作菜单
admin.menu()
# 获取用户的操作
num = int(input('请输入您的操作:'))
if num == 0:
print('彻底退出')
is_quit = True
break
elif num == 1:
op.new_user()
elif num == 2:
op.del_user()
elif num == 3:
op.query_money()
elif num == 4:
op.save_money()
elif num == 5:
op.get_money()
elif num == 6:
op.pass_money()
elif num == 7:
op.change_pwd()
elif num == 8:
op.lock_user()
elif num == 9:
op.unlock_user()
elif num == 10:
op.show_users()
else:
print('操作有误,请重新输入')
if is_quit:
break
else:
print('登录失败')
count += 1
if count >= 3:
print('错误已达上限,禁止登录')
break
结果: