010511-黑马ATM

定义一个全局变量：money，用来记录银行卡余额（默认5000000）
定义一个全局变量：name，用来记录客户姓名（启动程序时输入）
定义如下的函数：
查询余额函数 check_money
存款函数 deposit
取款函数 withdrawal
主菜单函数 main
要求：
程序启动后要求输入客户姓名
查询余额、存款、取款后都会返回主菜单
存款、取款后，都应显示一下当前余额
客户选择退出或输入错误，程序会退出，否则一直运行

money = 5000000  # 银行卡余额
user_name = input('请输入您的名字：')  # 用户名


def main(name):
    """
    主菜单
    :param name:用户名
    :return: 调用功能序号/余额
    """
    print('-----主菜单-----')
    print(f'{name},您好，欢迎来到黑马银行ATM。请选择操作：\n查询余额 [输入1]\n存款\t[输入2]\n取款\t[输入3]\n退出\t[输入4]')
    number = int(input('请输入您的选择：'))

    if number == 1:
        check_money(name)
    elif number == 2:
        deposit(name, money)
    elif number == 3:
        withdrawal(name, money)
    elif number == 4:
        exit()
    else:
        print('出错了！')


def check_money(name):
    """
    查询余额
    :param name:用户名
    :return: 余额
    """
    print('-----查询余额-----')
    print(f'{name}，您好，您的 余额剩余：{money}元')


def withdrawal(name, i):
    """
    取款
    :param name:用户名
    :param i:当前余额
    :return: 余额
    """
    print('-----取款-----')
    number = int(input('请输入金额：'))
    i -= number
    print(f'{name}，您好，您取款{number}元成功，当前余额为：{i}元')
    return i


def deposit(name, i):
    """
    存款
    :param name:用户名
    :param i: 当前余额
    :return: 余额
    """
    print('-----存款-----')
    number = int(input('请输入金额：'))
    i += number
    print(f'{name}，您好，您存款{number}元成功，当前余额为：{i}元')
    return i


main(user_name)

"""
老师原版



# 定义全局变量money name
money = 5000000
name = None
# 要求客户输入姓名
name = input("请输入您的姓名：")
# 定义查询函数
def query(show_header):
    if show_header:
        print("-------------查询余额------------")
    print(f"{name}，您好，您的余额剩余：{money}元")


# 定义存款函数
def saving(num):
    global money    # money在函数内部定义为全局变量
    money += num
    print("-------------存款------------")
    print(f"{name}，您好，您存款{num}元成功。")

    # 调用query函数查询余额
    query(False)

# 定义取款函数
def get_money(num):
    global money
    money -= num
    print("-------------取款------------")
    print(f"{name}，您好，您取款{num}元成功。")

    # 调用query函数查询余额
    query(False)
# 定义主菜单函数
def main():
    print("-------------主菜单------------")
    print(f"{name}，您好，欢迎来到黑马银行ATM。请选择操作：")
    print("查询余额\t[输入1]")
    print("存款\t\t[输入2]")
    print("取款\t\t[输入3]")    # 通过\t制表符对齐输出
    print("退出\t\t[输入4]")
    return input("请输入您的选择：")

# 设置无限循环，确保程序不退出
while True:
    keyboard_input = main()
    if keyboard_input == "1":
        query(True)
        continue    # 通过continue继续下一次循环，一进来就是回到了主菜单
    elif keyboard_input == "2":
        num = int(input("您想要存多少钱？请输入："))
        saving(num)
        continue
    elif keyboard_input == "3":
        num = int(input("您想要取多少钱？请输入："))
        get_money(num)
        continue
    else:
        print("程序退出啦")
        break       # 通过break退出循环

    
"""