p61 函數(shù)綜合案例

money = 5000000
name = input("請(qǐng)輸入用戶姓名：")

def main():
    print("-" * 15, "主菜單", "-" * 15)
    print(f"{name}，您好，歡迎來(lái)到黑馬銀行ATM，請(qǐng)選擇操作：")
    print("查詢余額\t[輸入1]")
    print("存款\t\t[輸入2]")
    print("取款\t\t[輸入3]")
    print("退出\t\t[輸入4]")
    return int(input("請(qǐng)輸入您的選擇："))

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("請(qǐng)輸入存款金額：")))
    elif m == 3:
        teller(int(input("請(qǐng)輸入取款金額：")))
    else:
        print("退出")
        break