黑马程序员2022新版python教程补充（P61）

        我也是一个新手小白，跟着黑马程序员在学，下面是我对P61集函数综合案例的一些改进。如果你仔细看了这集，会发现有些小问题没有处理，没看过或者没有印象的可以重新看一遍。（别喷）

 问题：1.取款数额大的情况没有考虑；

            2.最后if-elif-elif-else没有考虑输入数据不是1,2,3,4的情况。

下面是我做的改进。

money = 500000
name = None
name=input("请输入你的名字:")
def check(show_header):
    if(show_header):
        print("--------查询余额--------")
    print(f"{name},你的余额剩余{money}\n")

def save_money(num):
    global money
    money += num
    print("--------存款--------")
    print(f"{name}你存款{num}元成功")
    check(False)

def get_money(num):
    global money
    if num > money:
        print("你的余额不足,请重新输入：")
        num = int(input())
        get_money(num)
    else:
        money -= num
        print("--------取款--------")
        print(f"{name},你取款{num}元成功")
        check(False)

def main():
    print("--------主菜单--------")
    print("查询余额\t请输入1")
    print("存款\t\t请输入2")
    print("取款\t\t请输入3")
    print("退出\t\t请输入4")
    return input("请输入你的选择： ")

while 1:
    key_board = main()
    if key_board == '1':
        check(True)
        continue
    elif key_board == "2":
        num=int(input("请问你想存入多少？\n请输入："))
        save_money(num)
        continue
    elif key_board == '3':
        num = int(input("请问你想取出多少？\n请输入："))
        get_money(num)
        continue
    elif key_board == '4':
        print("程序退出了")
        break
    else:
        print("输入数据有误，请重新输入你的选择： ")