PYTHON自学之路_购物车程序升级版

对购物车程序进行了升级。

# Author:Johnson
'''
购物车优化
    用户入口：
    1、商品信息存在文件里。
    2、已购商品，余额记录。

    商家入口：
    1、可以添加商品，修改商品价格。
'''
import sys

ms=input("用户入口，请输入1；商家入口，请输入2。")
if ms=='1':
    sp_list=[]
    lines=open("./sp_list.ini","r",encoding="utf-8").readlines()
    for line in lines:
        sp_list.append([line.strip().split(",")[0],int(line.strip().split(",")[1])])
    gwc=[]
    gz=input("请输入工资：")
    if gz.isdigit():
        gz=int(gz)
        gz_list=open("./yu_e.ini","w",encoding="utf-8")
        gz_list.write(str(gz))
        gz_list.close()
    else:
        print("输入工资错误。")
        exit()
    for i in sp_list:
        print(sp_list.index(i),i[0],i[1])
    while True:
        shuanze=input("输入你要选择的商品编号：")
        if shuanze.isdigit():
            shuanze=int(shuanze)
            if shuanze <  len(sp_list) and shuanze>=0:
                s_item=sp_list[shuanze]
                if s_item[1]<=gz:
                    gwc.append(s_item)
                    gz-=s_item[1]
                    print("你选择了商品%s，还剩下%s元"%(s_item,gz))
                else:
                    print("工资不够。")
            else:
                print("输入的商品不存在。")
        elif shuanze=='Q':
            w_lines = open("./shoping_list.ini", "a", encoding="utf-8")
            for i in gwc:
                print(i)
                w_lines.write(i[0] + ',' + str(i[1]) + '\n')
            print("你还剩下%s元"%(gz))
            w_lines.close()
            exit()
        else:
            print("请输入一个数字")
elif ms=='2':
    mm=input("请输入密码：")
    if mm=="123":
        print("好的")
    else:
        print("密码错误，退出系统。")