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':''}       #定义用户名密码

 #生成商品-价格字典
 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("用户不存在，请重试！")

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