zyxck

money = 5000
key = "123"  # 保持密码的数据类型为字符串


def check_money():
    """查询余额"""
    print("------查询余额------")
    print(f"您好，您的余额为{money}元")
    menu()


def save_money(x):
    global money
    money += x
    print("-----------存款-----------")
    print(f"您好，存款成功，您的余额为{money}")
    menu()
    return money


def withdraw_money(y):
    global money
    if money < y:
        print("对不起，您的余额不足")
        menu()
    else:
        money -= y
        print("-----------存款-----------")
        print(f"您好，取款成功，您的余额为{money}")
        menu()
    return money


def menu():
    print()  # 换行
    print("查询余额\t[输入数字1]")
    print("存款  \t[输入数字2]")
    print("取款  \t[输入数字3]")
    print("退出  \t[输入数字4]")
    num = int(input())
    if num == 1:
        check_money()
    elif num == 2:
        # 用户输入的金额应该是一个整数，所以在调用int(input())时，将输入的字符串转换为整数。
        save_money(int(input("请输入存款金额")))
    elif num == 3:
        withdraw_money(int(input("请输入取款金额")))
    elif num == 4:
        exit()


key_input = input("您好欢迎来到ATM，请输入您的密码")
if key_input != key:
    print("密码错误，系统退出")
    exit()
else:
    print("密码正确,请选择操作")
    menu()