黑马程序员python教程，8天python从入门到精通，学python看这套就

p61 函数综合案例

money = 5000000
name = input("请输入用户姓名：")

def main():
    print("-" * 15, "主菜单", "-" * 15)
    print(f"{name}，您好，欢迎来到黑马银行ATM，请选择操作：")
    print("查询余额\t[输入1]")
    print("存款\t\t[输入2]")
    print("取款\t\t[输入3]")
    print("退出\t\t[输入4]")
    return int(input("请输入您的选择："))

def query(x):
    if x:
        print("-" * 15, "查询余额", "-" * 15)
    print(f"{name}，您好，您的余额剩余：{money}元")

def savings(x):
    global money
    print("-" * 15, "存款", "-" * 15)
    money += x
    print(f"{name}，您好，您存款{x}元成功")
    query(False)

def teller(x):
    global money
    print("-" * 15, "取款", "-" * 15)
    if x > money:
        print(f"{name}，您好，您的余额不足以取款")
    else:
        money -= x
        print(f"{name}，您好，您存款{x}元成功")
    query(False)

while True:
    m = main()
    if m == 1:
        query(True)
        continue
    elif m == 2:
        savings(int(input("请输入存款金额：")))
    elif m == 3:
        teller(int(input("请输入取款金额：")))
    else:
        print("退出")
        break