简易银行取钱代码

money = 5000000         #起始资金
name = input("请输入您的姓名:")
number = input("请设置密码:")
def query(m):        #查余额函数
        print("-------------查询余额-------------")
    print(f"{name},您好，您的余额剩余:{money}元")
def saving(num1):        #存款函数
    global money
    money += num1
    print("-------------存款-------------")
    print(f"{name}，您好，您存款{num1}元成功")
    query(False)
def get_money(num2):         #取款函数
    global money

    if money < num2:
        print("余额不足，无法取出该金额,请重新输入取款金额")
    else:
        money -= num2
        print(f"{name},您好,您取款{num2}元成功")
        query(False)
def main():         #初始界面
    print("----------------主菜单----------------")
    print(f"{name},您好，欢迎来到黑马ATM，请选择操作")
    print("查询余额\t【输入1】")
    print("存款\t\t【输入2】")
    print("取款\t\t【输入3】")
    print("退出\t\t【输入4】")
    return input("请输入您的选择：")
while True:
    a = main()          #无限循环
    if a == "1":
        query(True)
        continue
    elif a == "2":
        a = input("请输入密码：")
        if a == number:
            saving_person=(float(input("请输入存款金额：")))
            saving(saving_person)
            continue
        else:
            print("密码错误")
    elif a == "3":
        a = input("请输入密码：")
        if a == number:
            get_money_person=(float(input("请输入取款金额：")))
            get_money(get_money_person)
            continue
        else:
            print("密码错误")
    elif a == "4":
        print("您已成功退出")
        break