'''

购物车

用户入口

、商品信息存在文件里

、已购商品、余额记录，长期保存，

即第一次启动是需要输入预算的，以后就拿剩下的钱买东西

商家入口

、可以添加商品，修改商品价格

按q可以退出

'''

import os

#定义函数

#客户端

def customer(budgetfile,productfile):

    #显示或获取当前预算

    budgetfd = open(budgetfile, 'r', encoding="utf-8")

    print("目前已购买：")

    cartdata=budgetfd.readline()

    cartlist=[]

    budget=''

    while cartdata:

        cartlist.append(cartdata.split(" ")[].strip())

        print(cartdata.split(" ")[])

        budget=cartdata.split(" ")[]

        cartdata=budgetfd.readline()

    flag=True

    if budget=='':

        while flag:

            budget=input("请输入购物预算： ")

            if budget.isdigit():

                budget=int(budget)

                flag=False

            else:

                print("必须输入数字。")

                continue

    else:

        print("您的余额是："+budget)

        budget = int(budget)

    budgetfd.close()

    shopping=True

    while shopping:

        # 先显示当前商品列表

        productfd = open(productfile, 'r', encoding='utf-8')

        print("目前商品列表：")

        data = productfd.readline()

        datadic = {}

        while data:

            datadic[data.split(" ")[].strip()] = data.split(" ")[].strip()

            print(data)

            data = productfd.readline()

        productfd.close()

        choice=input("输入你要买的商品名称，按q退出：")

        if choice=='q':

            shopping=False;

            return True

        elif choice in datadic:

            if int(datadic[choice])<budget:

                fd=open(budgetfile,'a',encoding='utf-8')

                budget-=int(datadic[choice])

                print("您的余额是："+str(budget))

                cartlist.append(choice)

                fd.write(choice+' '+str(budget)+'\n')

                fd.close()

                print("目前已购买：")

                for key in cartlist:

                    print(key)

            else:

                print("预算不足，不能购买。")

        else:

            print("不存在该商品。")

#商户端

def boss(productfile):

    #打印目前商品列表

    fd=open(productfile,'r',encoding='utf-8')

    fd.close()

    while True:

        #先显示当前商品列表

        fd.close()

        fd = open(productfile, 'r+', encoding='utf-8')

        print("目前商品列表：")

        data = fd.readline()

        datadic = {}

        while data:

            datadic[data.split(" ")[].strip()] = data.split(" ")[].strip()

            print(data.split(" ")[].strip(), data.split(" ")[].strip())

            data = fd.readline()

        fd.close()

        operation = input("请输入您要做的操作：1表示增加商品、2表示删除商品、3表示修改商品价格、q表示退出程序： ")

        if operation=='':

            fd = open(productfile, 'a', encoding='utf-8')

            string=input("请输入要添加的商品及价格，用空格分隔：")

            key=string.split(" ")[].strip()

            value=string.split(" ")[].strip()

            if key not in datadic:

                fd.write(string+'\n')

                datadic[key]=value

            else:

                print("已存在该商品，不能添加")

            fd.close()

        elif operation=='':

            string = input("请输入要删除的商品名称: ")

            fd=open(productfile,'w',encoding='utf-8')

            if string in datadic:

                datadic.pop(string)

            else:

                print("要删除的商品不存在。")

            for key in datadic:

                fd.write(key + ' ' + datadic[key] + '\n')

            fd.close()

        elif operation=='':

            string=input("请输入要修改的商品名称和价钱，用空格分隔：")

            k=string.split(" ")[]

            v=string.split(" ")[]

            if k in datadic:

                datadic[k]=v

            else:

                print("要修改的商品不存在。")

            fd = open(productfile, 'w', encoding='utf-8')

            for key in datadic:

                fd.write(key+' '+datadic[key]+'\n')

            fd.close()

        elif operation=='q':

            fd.close()

            flag=False

            return True

        else:

            print("请输入准确的命令。")

            operation = input("请输入您要做的操作：1表示增加商品、2表示删除商品、3表示修改商品价格、q表示退出程序： ")

#脚本主程序

exitflag=False

Ident=input("您是客户还是商户：")

while not exitflag:

    if Ident=="客户":

        exitflag=customer("cartlist.txt","productlist.txt")

    elif Ident=="商户":

        exitflag=boss("productlist.txt")

    elif Ident=="q":

        exitflag=True

        print("退出程序。")

    else:

        print("请输入准确的字眼。")

        Ident = input("您是客户还是商户：")