自定义函数模拟银行存取款

题目

 

代码一（自己写的）：

money = 5000000 # 银行卡余额
name = input("请输入您的姓名：")
def check(show): # 定义查询余额函数
    if show:
        print("----------------查询余额-----------------")
    print(f"{name}，您好，您的余额剩余：{money}元")

def deposit(): # 定义存款函数
    num1 = int(input("请输入您要存款的数目："))
    global money # 把money变成全局变量
    money += num1
    print("----------------存款-----------------")
    print(f"{name}，您好，您存款{num1}元成功")
    check(False)

def withdraw_money(): # 定义取款函数
    num2 = int(input("请输入您要取款的数目："))
    global money
    money -= num2
    print("----------------取款-----------------")
    print(f"{name}，您好，您取款{num2}成功")
    check(False)

def main_menu(): # 定义主菜单
    print("----------------主菜单-----------------")
    print(f"{name},您好，欢迎来到黑马银行ATM，请选择操作：")
    print("查询余额\t[输入1]\n存款\t\t[输入2]\n取款\t\t[输入3]\n退出\t\t[输入4]")
    global select
    select = int(input("请输入您的选择"))
    if select == 1:
        check(True)
        main_menu()
    elif select == 2:
        deposit()
        main_menu()
    elif select == 3:
        withdraw_money()
        main_menu()

main_menu()

代码二：

money = 5000000 # 银行卡余额
name = None
name = input("请输入您的姓名：")
# 定义查询函数
def query(show_header):
    if show_header:
        print("-------------查询余额-------------")
    print(f"{name},您好，您的余额剩余：{money}元")

# 定义存款函数
def saving(num):
    global 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]\n存款\t\t[输入2]\n取款\t\t[输入3]\n退出\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退出循环

自己写的虽然也可以循环程序，输入4和输入错误可以退出，但是感觉有一点低级，循环还是用while语句更规范吧