ATM的python代码

# 函数案例：ATM
# 定义全局变量
money = 5000000
name = input("请输入姓名：")
# 定义查询函数
def query(show):
    if show:
        print("----------查询----------")
    print(f"{name}您好，您的余额剩余：{money}元")
# 定义存款函数
def save(sum):
    print("----------存款----------")
    global money
    money += sum
    print(f"{name}您好，您存款{sum}元成功")
    query(False)
# 定义取款函数
def get(sum):
    print("----------取款----------")
    global money
    money -= sum
    print(f"{name}您好，您取款{sum}元成功")
    query(False)
# 定义主函数
def main():
    print("----------主菜单----------")
    print(f"{name}您好，欢迎来到ATM。请选择操作")
    print("查询余额\t[输入1]")
    print("存款\t[输入2]")
    print("取款\t[输入3]")
    print("退出\t[输入4]")
    return int(input("请输入您的选择："))
# 循环函数
while True:
    clik = main()
    if clik == 1:
        query(True)
    elif clik == 2:
        sum = int(input("请输入存款金额："))
        save(sum)
    elif clik == 3:
        sum = int(input("请输入取款金额："))
        get(sum)
    else:
        print("程序退出")
        break