# 假設(shè)文件中的數(shù)據(jù)按照這樣來寫：藍(lán)莓，10,20（一行的數(shù)據(jù)）
import re


def read_goods(fn):
? ? fdic = {}
? ? with open(fn, "r") as f:
? ? ? ? for i in f.readlines():
? ? ? ? ? ? goodl = list(re.split(r"[,，]", i))
? ? ? ? ? ? goodl = [x.strip() for x in goodl]
? ? ? ? ? ? fdic[goodl[0]] = goodl
? ? return fdic


def add_goods(fdic, fn):
? ? goods_list = list(re.split('[,，]', input("請輸入商品信息（商品名稱，單價，數(shù)量），直接輸入回車退出:")))
? ? if goods_list[0] == "":
? ? ? ? return 0
? ? elif len(goods_list) == 3:
? ? ? ? try:
? ? ? ? ? ? a = eval(goods_list[1]) + eval(goods_list[2]) ?# 防止輸入價格時輸入非數(shù)字符號
? ? ? ? except NameError:
? ? ? ? ? ? print("價格請輸入數(shù)字符號")
? ? ? ? else:
? ? ? ? ? ? if goods_list[0] in fdic.keys():
? ? ? ? ? ? ? ? print("該商品已存在")
? ? ? ? ? ? else:
? ? ? ? ? ? ? ? fdic[goods_list[0]] = goods_list
? ? ? ? ? ? ? ? with open(fn, "a") as f:
? ? ? ? ? ? ? ? ? ? f.writelines(','.join(goods_list))
? ? ? ? ? ? ? ? ? ? f.write("\n")
? ? ? ? ? ? add_goods(fdic, fn)
? ? else:
? ? ? ? if goods_list[0] in fdic.keys():
? ? ? ? ? ? print("該商品已存在")
? ? ? ? else:
? ? ? ? ? ? print("輸入錯誤請重新輸入")
? ? ? ? add_goods(fdic, fn)


def find_goods(fdic):
? ? while True:
? ? ? ? good_name = input("請輸入查詢商品名稱，直接輸入回車退出：")
? ? ? ? if good_name == "":
? ? ? ? ? ? break
? ? ? ? else:
? ? ? ? ? ? for k in fdic.keys():
? ? ? ? ? ? ? ? if k == good_name:
? ? ? ? ? ? ? ? ? ? print("{},{:.2f}".format(k, eval(fdic[k][2])))
? ? ? ? ? ? ? ? ? ? find_goods(fdic)
? ? ? ? ? ? ? ? ? ? return 0
? ? ? ? ? ? print("商品庫中無該商品")


def count(fdic, fn):
? ? type_amount = len(fdic)
? ? good_amount, total_sales, sum_price, ave_price = 0, 0, 0, 0
? ? for v in fdic.values():
? ? ? ? good_amount += eval(v[2])
? ? ? ? total_sales += eval(v[2]) * eval(v[1])
? ? ? ? sum_price += eval(v[1])
? ? try:
? ? ? ? ave_price = sum_price / type_amount
? ? ? ? with open(fn, "w") as f:
? ? ? ? ? ? text = "商品種類： " + str(type_amount) + \
? ? ? ? ? ? ? ? ? ?"\n商品總數(shù)： " + str(good_amount) + \
? ? ? ? ? ? ? ? ? ?"\n銷售總額： " + str(total_sales) + \
? ? ? ? ? ? ? ? ? ?"\n商品均價： " + str(ave_price)
? ? ? ? ? ? f.write(text)
? ? except ZeroDivisionError:
? ? ? ? with open(fn, "w+") as f:
? ? ? ? ? ? f.seek(0)
? ? ? ? ? ? text = "商品種類： 0\n商品總數(shù)： 0 \n銷售總額： 0\n商品均價： 0"
? ? ? ? ? ? f.write(text)

? ? return print("商品統(tǒng)計數(shù)據(jù)已寫入統(tǒng)計文件")


def main():
? ? goodinfo = "C:\\Users\\13935\\Desktop\\goodinformation.txt" ?# 換成自己的路徑
? ? goodtotal = "C:\\Users\\13935\\Desktop\\goodtotle.txt" ?# 換成自己的路徑
? ? goods_dict = read_goods(goodinfo)
? ? print("1. 查詢商品\n2. 添加商品\n3. 統(tǒng)計商品\n4. 退出\n")
? ? while True:
? ? ? ? try:
? ? ? ? ? ? info = eval(input('請輸入您的選擇：'))
? ? ? ? ? ? if info == 1:
? ? ? ? ? ? ? ? find_goods(goods_dict)
? ? ? ? ? ? elif info == 2:
? ? ? ? ? ? ? ? add_goods(goods_dict, goodinfo)
? ? ? ? ? ? elif info == 3:
? ? ? ? ? ? ? ? count(goods_dict, goodtotal)
? ? ? ? ? ? elif info == 4:
? ? ? ? ? ? ? ? break
? ? ? ? ? ? else:
? ? ? ? ? ? ? ? print("輸入錯誤請重新輸入")
? ? ? ? except NameError:
? ? ? ? ? ? print("輸入錯誤請重新輸入")
? ? return 0


main()