Python学习的第二周作业--购物车

作业需求： 
（1）启动程序后，输入用户名密码后，如果是第一次登录，让用户输入工资，然后打印商品列表
（2）允许用户根据商品编号购买商品
（3）用户选择商品后，检测余额是否够，够就直接扣款，不够就提醒 
（4）可随时退出，退出时，打印已购买商品和余额
（5）在用户使用过程中， 关键输出，如余额，商品已加入购物车等消息，需高亮显示
（6）用户下一次登录后，输入用户名密码，直接回到上次的状态，即上次消费的余额什么的还是那些，再次登录可继续购买
（7）允许查询之前的消费记录

#!/usr/bin/env python
# -*- coding:utf-8 -*-
# Author:Ahappier

set = False
f =open ("user","r+",encoding="utf-8") #读取user文件
f_str = str(f.read())  #文件内容转字符串
for line in f_str:
    f_str_str= str(f_str)   #逐行读取
data = eval(f_str_str)  #字符串转字典
name = input ("请输入账号：")
password = input ("请输入密码：")
while True:
    if name in data:  #判断用户名在字典
        if password in data[name]:  #判断密码在字典内与用户对应
            salary = int (data[name][password])
            print ("\033[31;1m欢迎登陆，当前余额：%s\033[0m"%salary)
            break
        else:
            print ("输入密码错误，请重新输入")
            password = input ("请重新输入密码：")
            continue
    else:
        salary_p ={} #定义密码、工资空字典
        salary_input = input("请输入你的工资：")
        salary = int (salary_input) #输入的工资转成数字
        salary_p[password] = salary #密码与工资对应
        data[name] =salary_p  #用户名和密码 工资对应
        f.seek(0) #文件读取移到开头
        f.write(str(data))  #写入字典
        f.tell()  #返回到当前位置
        break

h =open("history","r+",encoding="utf-8")
h_str =str(h.read())
for line in h_str:
    h_str_str=str(h_str)
data1=eval(h_str_str) #把history文件内容转成字典
if name not in data1:
    data1[name]=[]         #如果用户名第一次登陆名字不在字典内，历史记录为空
shoppinglist = data1[name]
shoppinglist_new =[]   #再次购物的记录为空
choose = input ("\033[32;1m是否需要查询历史购物记录（y/n）\033[0m") #询问是否查询历史记录
if choose == "y":
    print("----历史购物记录-----")
    print (shoppinglist)


list =[
    ["Apple Iphone",6000],
    ["Apple Watch",4600],
    ["Books",600],
    ["Bike",750],
    ["cups",120],
    ["Apple",50],
    ["banana",60],
]                       #购物清单

while not set:
    print ("----商品清单----")
    for index,i in enumerate(list):
        print(index,i)        #输入商品清单
    number = input("输入你选择的商品标号")  #选择编号
    if number =="q":     #当输入q时，退出，并打印清单
        set =True
        data[name][password]= salary #工资给对应的用户密码位置
        f.seek(0)
        f.write(str(data))    #用户信息写入到文件中
        f.tell()
        print("----已购商品清单----")
        print (shoppinglist_new)  #打印清单
        print ("\033[31;1m你的余额：%s\033[0m"%salary)#打印余额
        shoppinglist.extend(shoppinglist_new)#本次购物记录追加到购物列表中
        data1[name]=shoppinglist   #购物列表和用户名对应
        h.seek(0)
        h.write(str(data1))  #购物记录写入文件
        h.tell()
    elif number.isdigit == False:
        print("\033[31;1m输入的不是编号，请重新输入\033[0m")
    elif int(number)>int(len(list)) or int(number)<0:
        print("\033[31;1m你购买的商品不在清单中\033[0m")
    else:
        number_buy =int(number)-1
        if list[number_buy][1]<salary:
            salary=salary-int(list[number_buy][1])
            shoppinglist_new.append(list[number_buy])
        else:
            print ("\033[31;1m你的余额不足\033[0m")