Python 银行系统模拟

# 导入所需要的模块 os
import  os

# 创建银行类
class bank():
    def __init__(self,name=None,pwd=None,money=None):
        self.name = name
        self.pwd  =pwd
        self.money =money
        self.user_data = {'aa':['1',1000],'bb':['1',2000]}  #初始2个用户，便于测试，可注释掉此行，程序不受影响
        self.login_menu()

    # 登陆菜单
    def login_menu(self):
        while    True:
            print("""
            ******* 欢迎来到《银行管理系统》 *************
            1、注册新用户         2、登陆        3、退出
            """)
            _choose = input("请选择：")
            if _choose == "1":
                self.createBank()
            elif    _choose =="2":
                self.login()
            elif _choose =="3":
                break
            else:
                print("输入错误，请重新输入!")
                self.login()
    # 用户登陆功能
    def login(self):
        _name = input("请输入用户名：")
        _pwd = input("请输入密码：")
        if _name in self.user_data.keys():
            if _pwd ==self.user_data[_name][0]:
                self.name  = _name
                self.menu()
            else:
                print("密码不正确！")
        else:
            print("用户不存在，请重新输入！")

    # 用户登陆后的银行卡操作菜单
    def menu(self):
        while    True:
            print("""
            ********* 欢迎进入《银行管理系统-模拟版》  **********
            1、存钱        2、取钱        3、查询
            4、注销        5、退出
            """)
            choose = input("请选择：")
            if  choose =="1":
                self.saveMoney()
            elif   choose == "2":
                self.drawMoney()
            elif    choose =="3":
                self.queryMoney()
            elif choose=="4":
                self.destory()
                break
            elif    choose  =="5":
                self.quit()
                break
            else:
                print("输入错误，请重新选择~")
                os.system("cls")

    # 开立帐号
    def createBank(self) :
        userName = input("请输入开卡人姓名：")

        if (userName in self.user_data.keys()):
            print("用户已存在，请重新输入：")
            self.createBank()
        else:
            userPwd1 = input("请输入密码：")
            userPwd2 = input("请再次输入密码：")
            money =int(input("请输入初始余额："))

            if userPwd1 == userPwd2 :
                user = {userName:[userPwd1,money]}
                self.user_data.update(user)
            else :
                print("密码输入错误，请重新输入！")
                self.createBank()

            print(f"{userName}开卡成功,当前余额为：{money} ")
            print(self.user_data)

    # 存钱
    def saveMoney(self):
        _money = int(input("请输入存款金额："))
        if _money>0 :
            self.user_data[self.name][1] += _money
            print("存钱成功:",self.user_data)
        else:
            print("你输入的金额不能小于等于‘零’ ！")

    # 取钱
    def drawMoney(self ):
        _money = int(input("请输入取款金额："))
        if _money >0 :
            if  _money<= self.user_data[self.name][1]:
                self.user_data[self.name][1] -= _money
                print(self.user_data)
            else:
                print(f"你的余额不足，当前余额为：{self.user_data[self.name][1]}")
        else:
            print("你输入的金额不能小于等于‘零’ ！")

    # 查询卡内余额
    def queryMoney(self):
        print(f"{self.name} 的当前余额为：{self.user_data[self.name][1]}")

    # 销户
    def destory(self):
        choose = input("确认是否注销？（Y/N)")
        if choose.upper() == 'Y':
            del self.user_data[self.name]
            print(f"{self.name}注销成功!")
            print (self.user_data)

    # 退出
    def quit(self):
        print("已退出系统，欢迎下次光临~~")

# 运行银行系统
b =bank()