Python--银行系统

# 定义全局变量 money name
money = 5000000
name = input("请输入您的姓名：")

# 定义查询函数
def check():

    print ("-----查询余额-----")
    print (f"{name},您好，您的账户余额为：{money}元")
    return

# 定义存款函数
def add():
    s = float(input("请输入您的存款金额："))
    global money
    money += s
    print ("-----存款-----")
    print (f"{name},您好，您存款{s}元成功。")
    check()  #存款结束后调用查询函数查看账户余额

# 定义取款函数
def WD():
    m = float(input("请输入您的取款金额："))
    global money
    money -= m
    print ("-----取款-----")
    print (f"{name},您好，您取款{m}元成功。")
    check()  #存款结束后调用查询函数查看账户余额

# 定义主菜单函数
def menu():
    print("--------主菜单--------")
    print(f"{name}您好，欢迎来到淄职ATM，请选择操作：")
    print("查询余额\t[输入1]")

    print("存款\t\t[输入2]")

    print("取款\t\t[输入3]")

    print("退出\t\t[输入4]")

    return input("请输入您的操作：")

# 设置无限循环，确保程序不退出
while True:
    enter = menu()
    if enter == '1':
        check()
        continue
    elif enter == '2':
        add()
        continue
    elif enter == '3':
        WD()
        continue
    else:
        print ("程序退出")
        break