Datei-Button volle Breite, Import als Aktionsbutton, Batch-Upsert statt Einzel-INSERTs. Co-authored-by: Cursor <cursoragent@cursor.com>
47 lines
1.4 KiB
Python
47 lines
1.4 KiB
Python
from io import BytesIO
|
|
|
|
from openpyxl import Workbook
|
|
|
|
from bls.import_service import should_persist_value
|
|
from bls.parser import parse_components_xlsx, parse_foods_xlsx
|
|
|
|
|
|
def _xlsx(rows):
|
|
wb = Workbook()
|
|
ws = wb.active
|
|
for row in rows:
|
|
ws.append(row)
|
|
buf = BytesIO()
|
|
wb.save(buf)
|
|
return buf.getvalue()
|
|
|
|
|
|
def test_parse_components_dynamic():
|
|
data = _xlsx([
|
|
["Code", "Name_DE", "Name_EN", "Einheit"],
|
|
["ENERCC", "Energie", "Energy", "kcal/100g"],
|
|
["NA", "Natrium", "Sodium", "mg/100g"],
|
|
])
|
|
attrs = parse_components_xlsx(data)
|
|
keys = {a["attr_key"] for a in attrs}
|
|
assert "ENERCC" in keys
|
|
assert "NA" in keys
|
|
|
|
|
|
def test_parse_foods_keeps_bls_code():
|
|
data = _xlsx([
|
|
["BLS Code", "Name", "Food name", "ENERCC Energie [kcal/100g]", "ENERCC Herkunft", "ENERCC Referenz"],
|
|
["C131000", "Hafer roh", "Oats raw", 350, "Analyse", "MRI"],
|
|
])
|
|
parsed = parse_foods_xlsx(data)
|
|
assert parsed["foods"][0]["bls_code"] == "C131000"
|
|
assert parsed["foods"][0]["name_de"] == "Hafer roh"
|
|
vals = {v["attr_key"]: v["value_num"] for v in parsed["foods"][0]["values"]}
|
|
assert vals.get("ENERCC") == 350
|
|
|
|
|
|
def test_persist_only_numeric_or_trace():
|
|
assert should_persist_value({"value_num": 1.2, "is_trace": False})
|
|
assert should_persist_value({"value_num": None, "is_trace": True})
|
|
assert not should_persist_value({"value_num": None, "is_trace": False})
|