python调用函数实现银行ATM

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


# 查询余额
def check_money():
    print(f"""----------------查询余额---------------\n
    {name}，你好，你的余额剩余：{money}元\n
""")
    return main_diet()


# 存款
def saving():
    saving = int(input("请输入存款余额："))
    global money
    money += saving
    print(f"""
    ----------------存款---------------\n
    {name}，你好，你存款{saving}元成功\n
    {name}，你好，你的余额剩余：{money}元\n

""")
    return main_diet()


# 取款
def get_money():
    global money
    get_money = int(input("请输入取款余额："))
    money -= get_money
    print(f"""
    ----------------取款---------------\n
    {name}，你好，你取款{get_money}元成功\n
    {name}，你好，你的余额剩余：{money}元\n
""")
    return main_diet()


def main_diet():
    print(
        f"""
    ----------------主菜单---------------\n
    {name}，你好，欢迎来到黑马银行ATM，请选择操作：\n
    查询余额\t[输入1]\n
    存款\t\t[输入2]\n
    取款\t\t[输入3]\n
    退出\t\t[输入4]\n
        """
    )
    choice = int(input("请输入你的选择："))
    if choice == 1:
        check_money()
    elif choice == 2:
        saving()
    elif choice == 3:
        get_money()
    else:
        print("程序结束！")


main_diet()