python實現簡單的商品數據管理系統

# 一個商品 名稱 價格 庫存 總銷量
# 存放商品的數據類型  str set list tuple dict
# 綜合考慮，選擇字典

dict_data = {}


# {名稱: {price: 價格, inventory: 庫存, sales: 總銷量}}


def exist_goods(goods_name):  # 有無keys都行應該是
    """判斷商品是否已經存在"""
    if goods_name in dict_data:
        return True
    else:
        return False


# 補貨，添加商品數據
def add_goods():
    """補貨"""
    goods_name = input('商品的名稱:')
    goods_count = int(input('商品的數量:'))
    # 該商品是否已經存在
    if exist_goods(goods_name):  # 有無keys都行應該是
        # 如果該商品已經存在
        # 就添加對應的數量
        dict_data[goods_name]['inventory'] += goods_count  # 添加對應的庫存
    else:
        goods_price = float(input('商品的價格:'))
        # 把商品數據添加到字典
        dict_data[goods_name] = {'price': goods_price, 'inventory': goods_count, 'sales': 0}
    print(f'{goods_name}補貨成功，當前庫存為{dict_data[goods_name]["inventory"]}')


# 出售商品
def sell_goods():
    """出售商品，庫存減少，銷量增加"""
    goods_name = input('商品的名稱:')
    if exist_goods(goods_name):
        goods_count = int(input('商品的數量:'))
        dict_data[goods_name]['inventory'] -= goods_count  # 添加對應的庫存
        dict_data[goods_name]['sales'] += goods_count  # 統計總銷量
        print('出售成功')
    else:
        print('商品名稱不存在')


# 修改商品的價格
def modify_price():
    """修改商品的價格"""
    goods_name = input('商品的名稱:')
    if exist_goods(goods_name):
        good_price = float(input('商品的價格:'))
        dict_data[goods_name]['price'] = good_price  # 修改價格
        print('修改成功')
    else:
        print('商品名稱不存在')


# 查詢單個商品的信息
def select_goods():
    """查詢單個商品的信息"""
    goods_name = input('商品的名稱:')
    if exist_goods(goods_name):
        print(dict_data[goods_name])
    else:
        print('商品不存在')


# 查詢所有商品的信息
def select_all_goods():
    """查詢所有商品的信息"""
    for i in dict_data.items():
        print(i)


while True:
    print('''----------------------------------------------------------------------------
1.增加商品   2.出售商品   3.修改商品價格   4.查詢單個商品數據   5.查詢所有商品數局   6.退出''')
    q = input('輸入要操作的對象的id:')
    if q == '1':
        add_goods()
    elif q == '2':
        sell_goods()
    elif q == '3':
        modify_price()
    elif q == '4':
        select_goods()
    elif q == '5':
        select_all_goods()
    elif q == '6':
        break
    else:
        print('操作id無效')