黑马ATM综合案例

money = 2000
name = None

name = input("请输入您的姓名：")


# 查询余额函数
def query(show_hender):
    if show_hender:
        print("--------查询余额--------")
    print(f"{name}先生/女士您好，您的余额剩余{money}元")


# 存款函数
def saving(num):
    global money
    money += num
    print("--------存款--------")
    print(f"{name}先生/女士，存款{money}成功")
    query(False)


# 取款函数
def get_money(num):
    global money
    money -= num
    print("--------取款--------")
    print(f"{name}先生/女士，存款{money}成功")
    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:
    keyboard_input = main()
    if keyboard_input == "1":
        query(True)
        continue
    elif keyboard_input == "2":
        num = int(input("请输入你要存款的金额："))
        saving(num)
        continue
    elif keyboard_input == "3":
        num = int(input("请输入你要取款的金额："))
        if num <= money:
            get_money(num)
        else:
            print("您的余额不足")
        continue
    else:
        print("程序结束")
        break