Интегрирован wesp в сайт

WESP_REL/app/lab/services/racion_reference.py

@@ -0,0 +1,155 @@
+"""Импорт и загрузка справочников RACION из БД."""
+
+from __future__ import annotations
+
+import json
+from dataclasses import dataclass, field
+from pathlib import Path
+from typing import Any
+
+from app import db
+from app.lab.calc.racion.tables import clear_tables_cache
+from app.lab.models.racion_normy import (
+    LabRacionNormyInfo,
+    LabRacionNormyMoskwa,
+    LabRacionNormyMoskwaMeta,
+    LabRacionNormyPiter,
+    LabRacionNormyPiterMeta,
+)
+from app.models.base import default_uuid
+
+_SEED_DIR = Path(__file__).resolve().parents[3] / "data" / "seed" / "racion"
+
+
+@dataclass
+class RacionReferenceImportStats:
+    moskwa_rows: int = 0
+    piter_rows: int = 0
+    info_rows: int = 0
+    errors: list[str] = field(default_factory=list)
+
+
+def _read_seed(name: str) -> dict:
+    path = _SEED_DIR / name
+    return json.loads(path.read_text(encoding="utf-8"))
+
+
+def import_racion_reference(*, replace: bool = True) -> RacionReferenceImportStats:
+    stats = RacionReferenceImportStats()
+    try:
+        moskwa = _read_seed("moskwa_lactir.json")
+        piter = _read_seed("piter_lactir.json")
+        info = _read_seed("normy_info.json")
+    except (OSError, json.JSONDecodeError) as exc:
+        stats.errors.append(str(exc))
+        return stats
+
+    if replace:
+        LabRacionNormyMoskwa.query.delete(synchronize_session=False)
+        LabRacionNormyMoskwaMeta.query.delete(synchronize_session=False)
+        LabRacionNormyPiter.query.delete(synchronize_session=False)
+        LabRacionNormyPiterMeta.query.delete(synchronize_session=False)
+        LabRacionNormyInfo.query.delete(synchronize_session=False)
+
+    for row in moskwa.get("rows") or []:
+        db.session.add(
+            LabRacionNormyMoskwa(
+                id=default_uuid(),
+                npitv=int(row["npitv"]),
+                pom=int(row.get("pom") or 1),
+                koef=row.get("koef"),
+                popr_k_json=json.dumps(row.get("popr_k") or [], ensure_ascii=False),
+            )
+        )
+        stats.moskwa_rows += 1
+
+    db.session.add(
+        LabRacionNormyMoskwaMeta(
+            id=default_uuid(),
+            meta_key="udoy_boundaries",
+            meta_json=json.dumps(moskwa.get("udoy_boundaries") or [], ensure_ascii=False),
+        )
+    )
+
+    for entry in piter.get("entries") or []:
+        db.session.add(
+            LabRacionNormyPiter(
+                id=default_uuid(),
+                npitv=int(entry["npitv"]),
+                konc=float(entry["konc"]),
+                udoy=float(entry["udoy"]),
+                normy_json=json.dumps(entry.get("normy") or [], ensure_ascii=False),
+            )
+        )
+        stats.piter_rows += 1
+
+    db.session.add(
+        LabRacionNormyPiterMeta(
+            id=default_uuid(),
+            meta_key="mass_kg_values",
+            meta_json=json.dumps(piter.get("mass_kg_values") or [], ensure_ascii=False),
+        )
+    )
+
+    for row in info.get("rows") or []:
+        db.session.add(
+            LabRacionNormyInfo(
+                id=default_uuid(),
+                nperem=int(row["nperem"]),
+                znachenie_json=json.dumps(row.get("znachenie") or [], ensure_ascii=False),
+            )
+        )
+        stats.info_rows += 1
+
+    db.session.commit()
+    clear_tables_cache()
+    return stats
+
+
+def load_moskwa_lactir_from_db() -> dict | None:
+    if LabRacionNormyMoskwa.query.count() == 0:
+        return None
+    rows = []
+    for r in LabRacionNormyMoskwa.query.order_by(LabRacionNormyMoskwa.npitv, LabRacionNormyMoskwa.pom).all():
+        rows.append(
+            {
+                "npitv": r.npitv,
+                "pom": r.pom,
+                "koef": r.koef,
+                "popr_k": json.loads(r.popr_k_json or "[]"),
+            }
+        )
+    meta = LabRacionNormyMoskwaMeta.query.filter_by(meta_key="udoy_boundaries").first()
+    boundaries = json.loads(meta.meta_json or "[]") if meta else []
+    return {"udoy_boundaries": boundaries, "rows": rows}
+
+
+def load_piter_lactir_from_db() -> dict | None:
+    if LabRacionNormyPiter.query.count() == 0:
+        return None
+    entries = []
+    for r in LabRacionNormyPiter.query.order_by(
+        LabRacionNormyPiter.npitv, LabRacionNormyPiter.konc, LabRacionNormyPiter.udoy
+    ).all():
+        entries.append(
+            {
+                "npitv": r.npitv,
+                "konc": r.konc,
+                "udoy": r.udoy,
+                "normy": json.loads(r.normy_json or "[]"),
+            }
+        )
+    meta = LabRacionNormyPiterMeta.query.filter_by(meta_key="mass_kg_values").first()
+    masses = json.loads(meta.meta_json or "[]") if meta else [400, 450, 500, 550, 600, 650, 700, 750]
+    return {"mass_kg_values": masses, "entries": entries}
+
+
+def load_normy_info_from_db() -> dict | None:
+    if LabRacionNormyInfo.query.count() == 0:
+        return None
+    rows = []
+    for r in LabRacionNormyInfo.query.order_by(LabRacionNormyInfo.nperem).all():
+        rows.append({"nperem": r.nperem, "znachenie": json.loads(r.znachenie_json or "[]")})
+    by_nperem = {row["nperem"]: row["znachenie"] for row in rows}
+    mass_kg_values = by_nperem.get(14) or by_nperem.get(13) or [400, 450, 500, 550, 600, 650, 700, 750]
+    return {"rows": rows, "mass_kg_values": mass_kg_values}