python實現庫存商品管理系統

# 假設檔案中的資料按照這樣來寫：藍莓，10,20（一行的資料）
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("請輸入商品資訊（商品名稱，單價，數量），直接輸入回車退出:")))
    if goods_list[0] == "":
        return 0
    elif len(goods_list) == 3:
        try:
            a = eval(goods_list[1]) + eval(goods_list[2])  # 防止輸入價格時輸入非數位符號
        except NameError:
            print("價格請輸入數位符號")
        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商品總數： " + 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 = "商品種類： 0n商品總數： 0 n銷售總額： 0n商品均價： 0"
            f.write(text)

    return print("商品統計資料已寫入統計檔案")


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. 統計商品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()