实现一个会员购物系统-简易版

需求一、需要打印一个好看的欢迎页面
需求二、可以充值
需求三、可以买东西
需求四、可以查看当前卡上余额
需求五、输入0表示退出程序
代码如下:

#准备一个欢迎界面
wel_page="""
欢迎进入会员专属页面,请选择要完成的操作:
1、充值
2、购买商品
3、查看当前卡上余额
0、退出程序
"""
def show_balance(balance):
    """
    展示账户余额
    :param balance: 账户余额
    :return: None
    """
    print(f"当前账户余额为{balance}")
def chongzhi(balance):
    """
    完成充值功能
    :param balance: 账户余额
    :return: 账户余额
    """
    num = int(input("请输入你要充值的金额:"))
    balance += num
    print(f"充值成功,本次充值{num}元")
    show_balance(balance)
    return balance

def shopping(balance):
    """
    完成消费购买操作
    :param balance:余额
    :return: balance
    """
    # 打印商品列表
    goods_list = {
        "A": 11111,
        "B": 22222,
        "C": 33333,
        "D": 44444
    }
    print("欢迎进入商品界面", goods_list)
    # 选择要购买的商品
    choice_good = input("请输入你要购买的商品:")
    num=int(input("请输入你要购买的数量:"))
    # 判断余额是否充足:
    pricea = goods_list[choice_good]*num
    while True:
        if balance >= pricea:
            # 如果账户余额足扣除消费金额
            balance -= pricea
            print(f"消费成功,本次消费{pricea}元")
            show_balance(balance)
            return balance
        else:
            # 否则打印余额不足,询问是否需要充值
            choice2 = input("当前账户余额不足,是否需要进行充值(y/n):")
            # 如果需要充值,进入充值模块
            if choice2 == "y":
                balance = chongzhi(balance)

            # 如果不需要,则重新选择操作
            else:
                print("余额不足,交易取消")
                return balance

#设定账户起始值
balance=0
#让程序循环起来
while   True:
    #打印欢迎界面
    print(wel_page)
    #用户输入要进行的操作
    u_choice=input("请输入你要使用的功能:")
    #判断用户的输入,展示相应的功能
    # 输入为1,进入充值界面
    if u_choice=="1":
        balance=chongzhi(balance)

    #输入为2,进入购买商品界面
    elif u_choice=="2":
        balance=shopping(balance)

    #输入为3,查看当前账户余额
    elif u_choice=="3":
        show_balance(balance)

    #输入为0,退出程序
    elif u_choice=="0":
        print("程序即将退出")
        break
    else:
        print("输入有误,请重新输入:")