路飞学城14天集训营作业3—购物车

作业需求：

数据结构：
goods = [
{"name": "电脑", "price": 1999},
{"name": "鼠标", "price": 10},
{"name": "游艇", "price": 20},
{"name": "美女", "price": 998},
......
]

功能要求：
基础要求：

1、启动程序后，输入用户名密码后，让用户输入工资，然后打印商品列表

2、允许用户根据商品编号购买商品

3、用户选择商品后，检测余额是否够，够就直接扣款，不够就提醒

4、可随时退出，退出时，打印已购买商品和余额

5、在用户使用过程中， 关键输出，如余额，商品已加入购物车等消息，需高亮显示


扩展需求：

1、用户下一次登录后，输入用户名密码，直接回到上次的状态，即上次消费的余额什么的还是那些，再次登录可继续购买

2、允许查询之前的消费记录

代码：

# -*- coding:utf-8 -*-
# author: heimu

import os,sys
import json

user_dic = {}       # 保存账户信息
shopping_list = {}  # 保存账户购物清单
salary_list = {}    # 保存账户余额

goods = [
{"name": "电脑", "price": 1999},
{"name": "鼠标", "price": 10},
{"name": "游艇", "price": 20},
{"name": "美女", "price": 998},
]

def init_all():
    '''
    初始化 user_dic, shopping_list, salary_list
    :return:
    '''
    user_dic['heimu_1'] = '111111'
    user_dic['heimu_2'] = '222222'
    user_dic['heimu_3'] = '333333'
    for key in user_dic.keys():
        shopping_list[key] = []
        salary_list[key] = 0
    save_data()                     # 保存初始化的数据到本地文件

def save_data():
    '''
    保存数据到本地文件
    :return:
    '''
    with open("user_dic.txt",'w') as f:
        json.dump(user_dic,f)
    with open("shopping_list.txt",'w') as f:
        json.dump(shopping_list,f)
    with open("salary_list.txt",'w') as f:
        json.dump(salary_list,f)

def download_data():
    '''
    从本地文件下载数据
    :return:
    '''
    with open("user_dic.txt",'r') as f:
        user_dic = json.load(f)
    with open("shopping_list.txt",'r') as f:
        shopping_list = json.load(f)
    with open("salary_list.txt",'r') as f:
        salary_list = json.load(f)
    return user_dic,shopping_list,salary_list

def inquire_salary():
    '''
    账户查询余额，高亮打印账户余额
    :return:
    '''
    print('33[31;1m余额：%d33[0m' % salary_list[user_name_input])

def inquire_shopping_list():
    '''
    账户查询历史购物清单，打印历史购物清单
    :return:
    '''
    print("-----------------你的历史购物清单------------")
    [print(shopping_list[user_name_input][i]) for i in range(len(shopping_list[user_name_input]))]

def exit_print():
    '''
    账户退出，打印购物清单和账户余额
    :return:
    '''
    print("----------------你的购物清单-------------")
    [print(shopping_list[user_name_input][i]) for i in range(len(shopping_list[user_name_input]))]
    print("-----------------你的账户余额------------")
    print('33[31;1m余额：%d33[0m' % salary_list[user_name_input])

def account_recharge():
    '''
    账户充值服务
    :return:
    '''
    recharge_number = input("请输入您要充值的金额：").strip()
    while True:
        if recharge_number.isdigit():
            salary_list[user_name_input]+=int(recharge_number)
            break
        else:
            recharge_number = input("输入错误，请重新输入您要充值的金额").strip()
    print("33[31;1m充值成功，您的账户当前余额为%d33[0m" % salary_list[user_name_input])

def start_shopping():
    '''
    账户购物流程
    :return:
    '''
    while True:
        print('-----------商品清单----------')
        [print(goods[i]) for i in range(len(goods))]    # 打印商品清单

        choice_input = input("请输入你要购买的商品编号(1--4)：").strip()
        if choice_input.isdigit() and 1 <= int(choice_input) <= 4:
            if goods[int(choice_input) - 1]['price'] <= salary_list[user_name_input]:   # 余额充足
                salary_list[user_name_input] -= goods[int(choice_input) - 1]['price']   # 更新账户的余额
                shopping_list[user_name_input].append(goods[int(choice_input) - 1])     # 商品加入购物车，更新购物清单
                print("商品----33[31;1m[%s]33[0m----已经加入购物车" % goods[int(choice_input) - 1]['name'])
                while True:
                    exit_flag = input("退出请按e,继续请按g").strip()
                    if exit_flag.lower() == 'e':
                        save_data()  # 退出保存数据
                        exit_print()  # 退出打印
                        sys.exit()
                    elif exit_flag.lower() == 'g':
                        break
                    else:
                        print('输入错误，请重新输入')

            else:           # 账户余额不足
                while True:
                    print("您的账户当前余额为：33[31;1m%d33[0m" % salary_list[user_name_input])
                    choose_user = input("您的账户余额不足，充值选择c,退出选择q").strip()
                    if choose_user.lower() == 'q':
                        save_data()                 # 退出保存数据
                        exit_print()                # 退出打印
                        sys.exit()
                    elif choose_user.lower() == 'c':
                        account_recharge()
                        break
                    else:
                        print("输入错误，请重新输入")
        else:
            print("输入错误，请重新输入")

if os.path.isfile("user_dic.txt") and os.path.isfile("shopping_list.txt") and os.path.isfile("salary_list.txt"):
    # 如果本地数据文件存在，从本地文件倒数数据
    print('------------导入数据库----------')
    user_dic, shopping_list, salary_list = download_data()
else:
    # 本地数据文件不存在，初始化数据
    print("-----------初始化数据----------")
    init_all()

while True:
    user_name_input = input("请输入账号：").strip()
    pass_words_name = input("请输入密码：").strip()

    if user_name_input in user_dic and pass_words_name == user_dic[user_name_input]:
        if not shopping_list[user_name_input]:       # 如果账户的历史购物清单为空，那么为第一次登陆
            salary_input = input("您是第一次登陆，请输入工资：").strip()
            salary_list[user_name_input] = int(salary_input)        # 更新账户余额
            print('33[31;1m余额：%d33[0m' % salary_list[user_name_input])
            start_shopping()                                        # 开始购物
        else:
            print('您以前有过消费记录')
            while True:
                choose_user = input("查询消费记录请按c，查询余额请按i，继续购物请按g,退出请按e").strip()
                if choose_user.lower() == 'c':
                    inquire_shopping_list()                         # 查询消费记录
                elif choose_user.lower() == 'i':
                    inquire_salary()                                # 查询余额
                elif choose_user.lower() == 'e':
                    save_data()                                     # 退出保存数据
                    exit_print()                                    # 退出打印
                    sys.exit()
                elif choose_user.lower() == 'g':
                    start_shopping()                                # 继续购物
                else:
                    print("输入错误，请重新输入")
    else:
        print("该账户不存在，请重新输入")