python 函数 练习

python 函数 练习

import random

"""
综合案例:红浪漫ATM
.定义一个全局变量:money用来记录银行卡余额(默认5000000)定义一个全局变量:name用来记录客户姓名(启动程序时输入)
.定义如下的函数:
    .查询余额函数
    .存款函数
    .取款函数
    .主菜单函数
.要求:
.程序启动后要求输入客户姓名
.查询余额、存款、取款后都会返回主菜单
.存款、取款后，都应显示一下当前余额
.客户选择退出或输入错误，程序会退出，否则一直运行
"""
money = random.randint(1000, 100000)
name = None


def look_money(is_look):
    if is_look:
        print("----------查询余额----------")
    print(f"{name}, 您好, 您的余额剩余:{money}元")


def add_money():
    print("----------存款----------")
    look_money(False)
    global money
    add_num = int(input("请输入存款数: "))
    money += add_num
    print(f"{name}, 您好, 您存款{add_num}元成功")


def subtract_money():
    print("----------取款----------")
    look_money(False)
    global money
    sub_num = int(input("请输入取款数: "))
    money -= sub_num
    print(f"{name}, 您好, 您取款{sub_num}元成功")


def main():
    print("----------主菜单----------")
    print(f"{name}, 您好, 欢迎来到红浪漫银行ATM, 请选择操作:")
    print("查询余额 [输入1]")
    print("存款    [输入2]")
    print("取款    [输入3]")
    print("退出    [输入4]")

    input_num = int(input("请输入您的选择: "))
    return input_num


def into_atm():
    global name
    name = input("请输入您的姓名: ")
    while True:
        input_num = main()
        if input_num == 1:
            look_money(True)
        elif input_num == 2:
            add_money()
            look_money(False)
        elif input_num == 3:
            subtract_money()
            look_money(False)
        else:
            print(f"您已成功退出ATM系统, 欢迎下次光临, 办理业务!")
            break


into_atm()