银行系统ATM

# 管理的类
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)


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


# 银行卡的类
class Card:
    def __init__(self, cid, pwd):
        self.cid = cid          # 卡号
        self.pwd = pwd          # 密码
        self.money = 0          # 余额
        self.is_lock = False    # 是否锁定

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 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):
        cid = input('请输入您的卡号:')
        if cid in self.userinfo:
            self.userinfo.pop(cid)
            User.save_user(self.userinfo)
            print('销户成功')
        else:
            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):
        cid = input('请输入卡号：')
        user = self.userinfo.get(cid)
        if user:
            pwd = input('请输入密码：')
            count = 0
            while count < 3:
                if Helper.check_password_hash(pwd,user.card.pwd):
                    get_momey = float(input('请输入要取的金额：'))
                    if user.card.money > get_momey:
                        user.card.money -= get_momey
                        User.save_user(self.userinfo)
                        break
                    else:
                        print('卡内余额不足，请及时充值')
                else:
                    count +=1
                    print('密码有误，重新输入')
        else:
            user.card.is_lock = True
            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):
        cid = input('请输入卡号：')
        user = self.userinfo.get(cid)
        if not self.userinfo.get(cid):
            print('改卡不存在')
        if user.card.is_lock ==  True:
            print('该卡已锁定，请先解锁')
            if user :
                uid = input('请出示您的身份证：')
                if uid == user.uid:
                    new_pwd = Helper.generate_password_hash(input('请输入新密码：'))
                    user.card.pwd = new_pwd
                    User.save_user(self.userinfo)
                    print('修改密码成功')
            else:
                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])


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)#  把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 {}