用python编写购物车程序(增加卖家入口，用户信息保存)

要求：

• 卖家入口：

1. 商家先将添加商品

2. 可以添加商品修改商品价格

• 买家入口：

1. 允许多个买家注册

2. 第一次启动才需要输入工资，之后信息被存储，已购商品、余额长时间留存

3. 启动程序后，打印用户已购买的商品列表

4. 允许用户根据商品编号购买商品

5. 用户选择商品后，检测余额是否充足，够就直接扣款，不够就提醒

6. 可随时推出，退出时打印以购买商品，购买商品数量及余额

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

import pickle
import os

buyer_or_seller=input('您好!请登录(买家输入’买家‘卖家输入’卖家‘)：')


#--------------------------------买家入口----------------------------------------------
if buyer_or_seller=='买家':

    #判断是否是新用户
    new_or_existuser=input('买家注册or登陆(注册输入’注册‘登陆输入’登陆‘)：')



    #--------------------------------新用户注册输入薪水---------------------------------
    if new_or_existuser=='注册':
        buyername=input('请输入用户名(由字母数字下划线组成)：')
        buyerpasswd=input('请输入密码(由字母数字下划线组成)：')

        #输入初始工资
        initial_salary=int(input('注册成功！请输入您的工资:'))

        #将用户密码存入字典
        user_passwd={}
        user_passwd[buyername]=buyerpasswd

        #将用户密码字典写入文件
        w_user_passwd = open(buyername+'passwd'+'.txt', 'wb')
        pickle.dump(user_passwd, w_user_passwd)
        w_user_passwd.close()


        #将用户信息存入字典
        user_info = {}
        user_info[buyername]={'balance':initial_salary,'boughtgoods':[]}

        #将用户信息字典写入文件
        w_user_info = open(buyername+'info'+'.txt', 'wb')
        pickle.dump(user_info, w_user_info)
        w_user_info.close()

        # 将用户信息暂时导入到temp_user_info里
        temp_user_info=user_info



    #------------------------------旧用户登陆并导入用户信息-------------------------------
    elif new_or_existuser=='登陆':

        while True:
            buyername = input('请输入用户名：')
            buyerpasswd = input('请输入密码：')

            #读出用户信息，登陆密码
            r_user_passwd = open(buyername+'passwd'+'.txt', 'rb')
            read_user_passwd = pickle.load(r_user_passwd)
            r_user_passwd.close()

            # 判断输入的账号是否存在密码是否正确
            if buyername in read_user_passwd and buyerpasswd==read_user_passwd[buyername]:

                # 读出用户信息，余额已购买商品
                r_user_info = open(buyername+'info'+'.txt', 'rb')
                read_user_info= pickle.load(r_user_info)
                r_user_info.close()

                # 打印用户信息，余额、已购买商品
                print('''
用户{_user}您好，您的余额为{_balance}
                '''.format(_user=buyername,_balance=read_user_info[buyername]['balance']))
                print('------您已够买以下商品------')
                for i in read_user_info[buyername]['boughtgoods']:
                    print(i)

                #将用户信息暂时导入到temp_user_info里
                temp_user_info=read_user_info
                break

            else:
                print('用户名不存在或密码错误，请重新登录')


    #用户注册或登陆之后已经有了用户信息，包括用户名、薪水或余额、已购买商品存在temp_user_info字典里

    #从文件中读出商品信息
    r_shoppinggoods = open('shoppinggoods.txt', 'rb')
    read_shoppinggoods = pickle.load(r_shoppinggoods)
    r_shoppinggoods.close()

    dict_of_bought={}

    #从字典temp_user_info读出用户信息，包括用户名、薪水或余额、已购买商品(以购买商品用数列存储)
    balance=temp_user_info[buyername]['balance']
    list_of_bought=temp_user_info[buyername]['boughtgoods']#以购买商品用数列存储




    #---------------------------------开始购买流程-------------------------------------
    while True:

        #输出商品及其编号
        print('--------支持购买的商品-------')
        for index,item in enumerate(read_shoppinggoods):#enumerate取出下标
            print(index,item)
            #print(read_shoppinggoods.index(item),item)
        choice_of_user=input('选择购买商品编号输入q退出:')

        #判断输入是否合法
        if choice_of_user.isdigit():
            choice_of_user=int(choice_of_user)

            #判断编号是否有对应商品
            if 0<=choice_of_user<len(read_shoppinggoods):

                #判断余额是否足够买此商品
                if read_shoppinggoods[choice_of_user][1]<= balance:

                    #加入购物清单
                    list_of_bought.append(read_shoppinggoods[choice_of_user][0])

                    #计算余额`
                    balance-=read_shoppinggoods[choice_of_user][1]

                    print('''添加{_boughtgood}到您的购物车,此刻您的余额为{_balance}.
                    '''.format(_boughtgood=read_shoppinggoods[choice_of_user][0],_balance=balance))
                else:
                    print('您的余额不足,此实余额为%s,不够购买此商品',balance)
            else:
                print('商品不存在')

        elif choice_of_user=='q':

            #统计购买的商品及数量
            category_of_bought=set(list_of_bought)
            for item in category_of_bought:
                dict_of_bought[item]=list_of_bought.count(item)
            print('您购买的商品及数量分别为',dict_of_bought)

            print('您的余额为:',balance)

            #将用户余额和购买商品更新存入字典
            temp_user_info[buyername] = {'balance': balance, 'boughtgoods': list_of_bought}

            #将用户余额和购买商品更新写入文件
            w_user_userinfo = open(buyername+'info'+'.txt', 'wb')
            pickle.dump(temp_user_info, w_user_userinfo)
            w_user_userinfo.close()

            exit()
        else:
            print('输入不合法')








#--------------------------------卖家入口----------------------------------------------
elif buyer_or_seller=='卖家':

    #卖家注册(因为只有一个店，所以只有一个卖家)
    if os.path.getsize('sellerpasswd.txt') == 0:
        print('新卖家请注册')

        sellername = input('请输入用户名(由字母数字下划线组成)：')
        sellerpasswd = input('请输入密码(由字母数字下划线组成)：')

        #将卖家密码存入字典
        seller_passwd = {}
        seller_passwd[sellername] = sellerpasswd

        # 将用户密码字典写入文件
        w_seller_passwd = open('sellerpasswd.txt', 'wb')
        pickle.dump(seller_passwd, w_seller_passwd)
        w_seller_passwd.close()

        print('恭喜！注册成功！请重新登录。')

    #卖家登陆
    while True:
        print('请登录')
        sellername = input('请输入用户名：')
        sellerpasswd = input('请输入密码：')

        # 读出卖家信息，登陆密码
        r_seller_passwd = open('sellerpasswd.txt', 'rb')
        read_seller_passwd = pickle.load(r_seller_passwd)
        r_seller_passwd.close()

        # 判断输入的账号是否存在密码是否正确
        if sellername in read_seller_passwd and sellerpasswd == read_seller_passwd[sellername]:
            print('欢迎')

            #从商品文件中读出已添加商品，若文件中存在商品则打印提示是否添加，若文件中不存在商品则提示添加
            if os.path.getsize('shoppinggoods.txt') > 0:

                # 从文件中读出已添加商品
                r_shoppinggoods = open('shoppinggoods.txt', 'rb')
                read_shoppinggoods = pickle.load(r_shoppinggoods)
                r_shoppinggoods.close()

                temp_shoppinggoods=read_shoppinggoods


                #显示出已添加商品
                print('------您已添加商品及其价格如下------')
                for i in read_shoppinggoods:
                    print(i)
            else:
                temp_shoppinggoods = []
                print('您还未添加任何商品')

            #添加商品
            add_or_not = input('是否添加商品？添加请输入A退出输入其它')
            if add_or_not == 'A':
                while True:
                    add_goodname = input('请输入您要添加的商品：')
                    add_goodprice = int(input('请输入您要添加的商品价格：'))
                    add_good = [add_goodname, add_goodprice]

                    temp_shoppinggoods.append(add_good)

                    continue_or_not = input('是否继续添加商品？继续请输入C')
                    if continue_or_not == 'C':
                        pass
                    else:
                        break

            #将商品列表写入文件
            print('已添加商品:',temp_shoppinggoods)
            w_seller_shoppinggoods = open('shoppinggoods.txt', 'wb')
            pickle.dump(temp_shoppinggoods, w_seller_shoppinggoods)
            w_seller_shoppinggoods.close()

            break

        else:
            print('用户名不存在或密码错误，请重新登录')