66 lines
2.0 KiB
Python
66 lines
2.0 KiB
Python
from openpyxl import Workbook
|
||
|
||
# 从图片人工识别到的口味与尼古丁浓度（mg）。
|
||
# 目前未录入库存数量；如需要，可再新增一列并补充。
|
||
|
||
inventory_rows = []
|
||
|
||
# 6mg
|
||
for flavor in [
|
||
"Apple Mint","Blueberry","Citrus","Coffee","Cool Mint",
|
||
"Mango","Peach","Spearmint","Watermelon","Wintergreen",
|
||
"Cola","Sour Blue Razz","Ice Wintergreen"
|
||
]:
|
||
inventory_rows.append((flavor, 6))
|
||
|
||
# 9mg
|
||
for flavor in [
|
||
"Cool Mint","Wintergreen","Citrus","Peppermint","Spearmint",
|
||
"Smooth","Apple Mint","Blueberry","Grape","Mango",
|
||
"Black Cherry","Double Mint","Menthol","Ice Wintergreen",
|
||
"Max Polar Mint","Cinnamon","Energy Drink","Orange","Peach",
|
||
"Watermelon","Coffee","Dragon Fruit","Cola","Sour Blue Razz",
|
||
"Strawberry Kiwi","Pineapple"
|
||
]:
|
||
inventory_rows.append((flavor, 9))
|
||
|
||
# 12mg
|
||
for flavor in [
|
||
"Apple Mint","Black Cherry","Blueberry","Coffee","Cool Mint",
|
||
"Mango","Peach","Pink Lemonade","Spearmint","Wintergreen",
|
||
"Ice Wintergreen","Grape","Watermelon"
|
||
]:
|
||
inventory_rows.append((flavor, 12))
|
||
|
||
# 15mg
|
||
for flavor in [
|
||
"Cool Mint","Wintergreen","Citrus","Peppermint","Spearmint",
|
||
"Chill","Coffee","Blueberry","Dragon Fruit","Mango",
|
||
"Apple Mint","Citrus Chill","Grape","Double Mint","Ice Wintergreen",
|
||
"Strawberry Watermelon","Max Polar Mint","Vanilla","Pink Lemonade"
|
||
]:
|
||
inventory_rows.append((flavor, 15))
|
||
|
||
|
||
def build_excel(output_path: str):
|
||
wb = Workbook()
|
||
|
||
# Sheet1: 图片转Excel（口味-浓度）
|
||
ws1 = wb.active
|
||
ws1.title = "图片转Excel"
|
||
ws1.append(["Flavor", "Strength(mg)"])
|
||
for flavor, mg in inventory_rows:
|
||
ws1.append([flavor, mg])
|
||
|
||
# Sheet2: 口味列表（去重）
|
||
ws2 = wb.create_sheet(title="口味列表")
|
||
ws2.append(["Flavor"])
|
||
unique_flavors = sorted({f for f, _ in inventory_rows})
|
||
for f in unique_flavors:
|
||
ws2.append([f])
|
||
|
||
wb.save(output_path)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
build_excel("flavors_inventory.xlsx") |