Python学习之购物车

实现功能：

1. 程序启动，提示用户输入用户名和密码，程序读取余额文件last_salary.txt内容（文件不存在则自动创建），若文件内容为空则提示“首次登录，请输入工资”；
2. 用户可以输入商品编号进行购买；
3. 用户选择商品后，自动计算总价，若总价超过账户余额salary，则提示余额不足；若总价未超过余额，则自动扣除；
4. 用户可在购买过程中随时退出“q”；
5. 关键信息高亮显示；
6. 用户退出时，余额信息存入文件last_salary.txt，购买商品信息存入文件last_bucket.txt,文件不存在则自动创建；
7. 用户再次登录时，读取并显示当前账户余额信息；
8. 用户可在购买过程中输入“h”查询历史消费信息；
9. 支持多个用户；
10. 支持显示当前购物车商品数量；
11. 当购物车为空，退出时显示提示信息。

示例代码：

# -*- coding: utf-8 -*-

import os
from datetime import datetime

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

now = datetime.now().strftime('%Y-%m-%d %H:%M:%S')      #获取当前时间
bucket =  []        #定义空列表存储购买的商品
price_list = {}     #将商品及其价格以“键值对”形式存入字典
total_price = item_price = 0
exit_flag = False
user_info = {'admin':'admin','alex':'123456'}       #定义用户名密码

#生成商品-价格字典
for i in goods:
    a = list(i.values())
    price_list.setdefault(a[0],a[1])

#生成商品名称列表
product_list = list(price_list)

#开始
while not exit_flag:
    username = input("用户名：").strip()
    if username in user_info:
        while not exit_flag:
            userpasswd = input("密码：").strip()
            if userpasswd == user_info.get(username):
                if not os.path.isfile(username + '_last_salary.txt'):       #判断username + '_last_salary.txt'是否存在，不存在则创建，存在则读取文件内容
                    with open(username + '_last_salary.txt', 'w+', encoding='utf-8') as f:
                        s = f.read()
                else:
                    with open(username + '_last_salary.txt','r',encoding= 'utf-8') as f:      #若username + '_last_salary.txt'文件存在，则读取内容
                        s = f.read()
                if len(s):          #如果username + '_last_salary.txt'不存在或为空，则表示用户首次登录，要求输入工资
                    salary = int(s)
                    print("\033[1;31;m*欢迎回来！账户余额：%d\033[0m " % salary)
                else:
                    while True:
                        salary = input("\033[1;31;m*首次登录，请输入工资：\033[0m").strip()     #首次登陆需输入工资
                        if salary.isdigit():
                            salary = int(salary)
                            break
                        else:
                            print("输入错误，请重试！")
                print("-------商品列表------")
                for n,i in enumerate(price_list):
                    print(n,i,price_list[i])
                while not exit_flag:        #要求用户输入商品编号
                    select = input("请选择商品：").strip()
                    if select.isdigit() and int(select) < len(goods):       #判断输入的商品编号是否是数字，且在范围内
                        select = int(select)
                        p = product_list[select]
                        item_price = price_list[p]
                        total_price += item_price
                        if salary < item_price:              #若余额小于所选商品价格，则提示余额不足
                            print("\033[1;31;m余额不足！\033[0m")
                            print(salary)
                        else:                               #若余额大于或等于所选商品价格，则扣除相应该商品对应价格并加入购物车
                            salary = salary - item_price
                            bucket.append(p)
                            print("\033[1;31;m \'%s\'已加入购物车！ \033[0m" % p )
                    elif select == 'q':                 #若用户输入q退出，则：打印已购买商品列表；打印账户余额；将余额和购买记录分别存入不同文件
                        print("\033[1;31;m-------已购买商品-------\033[0m")
                        if len(bucket):
                            for n,p in enumerate(set(bucket)):
                                print(p,tuple(bucket).count(p))
                        else:
                            print("当前购物车为空！")
                        print("\033[1;31;m-----------------------\033[0m")
                        print("\033[1;31;m账户余额:%d\033[0m" % salary)
                        with open(username + '_last_salary.txt','w',encoding= 'utf-8') as f1:
                            f1.write(str(salary))               #将余额存入文件username + '_last_salary.txt'
                        with open(username + '_last_bucket.txt','a+',encoding= 'utf-8') as f2:
                            for i in bucket:                    #将消费记录存入文件username + '_last_bucket.txt',若文件不存在则创建,指定写入字符串编码为utf-8
                                f2.write(now + ' ' + i + '\n')
                        exit_flag = True
                    elif select == 'h':         #当用户输入'h',读取并打印历史消费信息
                        print("--------历史购买商品-------")
                        if os.path.isfile(username + '_last_bucket.txt'):
                            with open(username + '_last_bucket.txt','r',encoding= 'utf-8') as f:
                                l = f.read()
                                print(l)
                        else:
                            with open(username + '_last_bucket.txt', 'w', encoding='utf-8') as f:
                                print("历史记录为空！")
                    else:
                        print("请输入商品编号！")
                        continue
            else:
                print("密码错误，请重试！")
    else: print("用户不存在，请重试！")

代码写的有点冗余繁杂，后期再优化。