"""WESP-shaped recipe write/calculate for orchestrator static UI.""" from __future__ import annotations from datetime import UTC, datetime from typing import Any from uuid import uuid4 from fastapi import HTTPException, status from sqlalchemy import select from app.core.database import session_scope from app.modules.sync.engine import SyncEngine from app.modules.sync.schemas import ChangeEventIn from app.modules.zootech.catalog_apply import load_catalog_row from app.modules.zootech.catalog_models import ZootechUnloadingGroup from app.modules.zootech.models import ZootechComponent, ZootechIngredient, ZootechRecipe from app.modules.zootech.recipe_calculator import calculate_recipe from app.modules.zootech.sync_content_hash import compute_content_hash class RecipeWriteError(Exception): def __init__(self, message: str, status_code: int = 400): self.message = message self.status_code = status_code super().__init__(message) def _normalize_groups(raw: list[Any]) -> list[dict[str, Any]]: out: list[dict[str, Any]] = [] for group in raw: if not isinstance(group, dict): continue g = dict(group) if "distributionType" not in g and "distribution_type" in g: g["distributionType"] = g.get("distribution_type") out.append(g) return out def calculate_recipe_wesp(body: dict[str, Any]) -> dict[str, Any]: if not body: raise RecipeWriteError("Данные не предоставлены", 400) try: heads_count = int(body.get("headsCount") or body.get("heads_count") or body.get("headsPerTrip") or 0) trip_percent = float(body.get("tripPercent") or body.get("trip_percent") or 100) except (TypeError, ValueError) as exc: raise RecipeWriteError("Некорректные числовые параметры", 400) from exc ingredients = body.get("ingredients") or [] unloading_groups = _normalize_groups(body.get("unloadingGroups") or body.get("unloading_groups") or []) calculate_from_dry_matter = bool( body.get("calculateFromDryMatter") if body.get("calculateFromDryMatter") is not None else body.get("calculate_from_dry_matter", False) ) component_ids = [i.get("component_id") for i in ingredients if isinstance(i, dict) and i.get("component_id")] component_dry_matter_map: dict[str, float] = {} if component_ids: with session_scope() as db: for comp in db.scalars(select(ZootechComponent).where(ZootechComponent.id.in_(component_ids))): component_dry_matter_map[comp.id] = float(comp.dry_matter or 0) result = calculate_recipe( ingredients=[i for i in ingredients if isinstance(i, dict)], heads_count=heads_count, trip_percent=trip_percent, unloading_groups=unloading_groups, component_dry_matter_map=component_dry_matter_map or None, calculate_from_dry_matter=calculate_from_dry_matter, ) def _to_float2(x: Any) -> float: try: return round(float(x), 2) except (TypeError, ValueError): return 0.0 def _truncate2(x: Any) -> float: try: v = float(x) return float(int(v * 100)) / 100.0 except (TypeError, ValueError): return 0.0 for ing in result.get("ingredients") or []: for key in ("weightPerHead", "tripWeight", "totalWeight", "dryMatterPerHead"): if key in ing and ing[key] is not None: raw_val = ing[key] v = _truncate2(raw_val) if key == "weightPerHead" else _to_float2(raw_val) ing[key] = f"{v:.2f}" if key == "weightPerHead" else v totals = result.get("totals") or {} for key in ("totalWeight", "totalTripWeight", "totalDryMatterPerHead", "totalWeightPerHead"): if key in totals and totals[key] is not None: totals[key] = _to_float2(totals[key]) result["totals"] = totals return result def _push_change( enterprise_id: str, table: str, record_id: str, payload: dict[str, Any], *, action: str = "upsert", ) -> None: version = int(payload.get("version") or 1) content_hash = compute_content_hash(payload) payload = {**payload, "version": version, "content_hash": content_hash} event = ChangeEventIn( event_id=str(uuid4()), seq=0, domain="global", table=table, record_id=record_id, action=action, version=version, content_hash=content_hash, payload=payload, emitted_at=datetime.now(UTC), origin_site_id=SyncEngine.ORCHESTRATOR_SITE_ID, ) SyncEngine(enterprise_id, SyncEngine.ORCHESTRATOR_SITE_ID).process_push([event], origin="orchestrator") def _resolve_component(enterprise_id: str, ing: dict[str, Any]) -> ZootechComponent | None: comp_id = ing.get("component_id") with session_scope() as db: if comp_id: row = db.scalar( select(ZootechComponent).where( ZootechComponent.enterprise_id == enterprise_id, ZootechComponent.id == str(comp_id), ZootechComponent.is_deleted.is_(False), ) ) if row: return row name = (ing.get("name") or "").strip() if name: return db.scalar( select(ZootechComponent).where( ZootechComponent.enterprise_id == enterprise_id, ZootechComponent.name == name, ZootechComponent.is_deleted.is_(False), ) ) return None def update_recipe_wesp(enterprise_id: str, recipe_id: str, data: dict[str, Any]) -> dict[str, Any]: if not data: raise RecipeWriteError("Данные не предоставлены", 400) for key in ("name", "heads_count", "mixing_time"): if key not in data: raise RecipeWriteError(f'Отсутствует обязательное поле "{key}"', 400) with session_scope() as db: recipe = db.scalar( select(ZootechRecipe).where( ZootechRecipe.enterprise_id == enterprise_id, ZootechRecipe.id == recipe_id, ZootechRecipe.is_deleted.is_(False), ) ) if not recipe: raise RecipeWriteError("Рецепт не найден", 404) existing_ings = list( db.scalars( select(ZootechIngredient).where( ZootechIngredient.enterprise_id == enterprise_id, ZootechIngredient.recipe_id == recipe_id, ZootechIngredient.is_deleted.is_(False), ) ) ) existing_groups = list( db.scalars( select(ZootechUnloadingGroup).where( ZootechUnloadingGroup.enterprise_id == enterprise_id, ZootechUnloadingGroup.recipe_id == recipe_id, ZootechUnloadingGroup.is_deleted.is_(False), ) ) ) recipe_trip_percent = float(recipe.trip_percent or 100) recipe_dry_matter_locked = bool(recipe.dry_matter_locked) recipe_unloading_link_broken = bool(recipe.unloading_link_broken) recipe_target_component_id = recipe.target_component_id recipe_version = int(recipe.version or 1) + 1 ing_by_id = { str(i.id): {"id": str(i.id), "name": i.name, "version": int(i.version or 1)} for i in existing_ings } grp_by_id = { str(g.id): {"id": str(g.id), "name": g.name, "version": int(g.version or 1)} for g in existing_groups } recipe_payload = { "id": recipe_id, "name": str(data["name"]), "heads_per_trip": int(data["heads_count"]), "mixing_time": int(data["mixing_time"]), "trip_percent": float(data.get("trip_percent") or recipe_trip_percent), "dry_matter_locked": bool(data.get("dry_matter_locked", recipe_dry_matter_locked)), "unloading_link_broken": bool(data.get("unloading_link_broken", recipe_unloading_link_broken)), "target_component_id": data.get("target_component_id", recipe_target_component_id), "version": recipe_version, } _push_change(enterprise_id, "recipe", recipe_id, recipe_payload) seen_ing_ids: set[str] = set() for idx, ing in enumerate([x for x in (data.get("ingredients") or []) if isinstance(x, dict)], start=1): component = _resolve_component(enterprise_id, ing) if not component: raise RecipeWriteError( f'Компонент не найден (id="{ing.get("component_id", "")}", name="{ing.get("name", "")}")', 400, ) ing_id = str(ing.get("id") or "").strip() or str(uuid4()) existing = ing_by_id.get(ing_id) order_value = int(ing.get("order") or idx) wph = float(ing.get("weight_per_head") or ing.get("weightPerHead") or 0) amount = float(ing.get("amount") or 0) dm = float(ing.get("dry_matter") or component.dry_matter or 0) dm_ph = ing.get("dry_matter_per_head") if dm_ph is None: dm_ph = ing.get("dryMatterPerHead") dm_ph_f = float(dm_ph) if dm_ph not in (None, "") else (wph * (dm / 100.0) if wph and dm else 0.0) version = int(existing.get("version") or 1) + 1 if existing else 1 payload = { "id": ing_id, "recipe_id": recipe_id, "component_id": component.id, "name": component.name, "amount": amount, "weight_per_head": wph, "dry_matter": dm, "dry_matter_per_head": dm_ph_f, "order": order_value, "version": version, } _push_change(enterprise_id, "ingredient", ing_id, payload) seen_ing_ids.add(ing_id) for raw_id in data.get("deleted_ingredient_ids") or []: ing_id = str(raw_id or "").strip() if not ing_id or ing_id not in ing_by_id: continue existing = ing_by_id[ing_id] version = int(existing["version"] or 1) + 1 payload = { "id": ing_id, "recipe_id": recipe_id, "name": existing["name"], "version": version, } _push_change(enterprise_id, "ingredient", ing_id, payload, action="delete") for ing_id, existing in ing_by_id.items(): if ing_id not in seen_ing_ids and ing_id not in {str(x) for x in (data.get("deleted_ingredient_ids") or [])}: version = int(existing["version"] or 1) + 1 _push_change( enterprise_id, "ingredient", ing_id, {"id": ing_id, "recipe_id": recipe_id, "name": existing["name"], "version": version}, action="delete", ) seen_grp_ids: set[str] = set() groups_in = [x for x in (data.get("unloading_groups") or data.get("unloadingGroups") or []) if isinstance(x, dict)] for idx, group in enumerate(groups_in, start=1): try: gname = str(group["name"]) gdist = str(group.get("distribution_type") or group.get("distributionType") or "percent") gval = float(group.get("value") or 0) except (KeyError, TypeError, ValueError) as exc: raise RecipeWriteError(f"Некорректная группа выгрузки (order={idx}): {exc}", 400) from exc grp_id = str(group.get("id") or "").strip() or str(uuid4()) existing = grp_by_id.get(grp_id) order_value = int(group.get("order") or idx) weight_raw = group.get("weight") weight = float(weight_raw) if weight_raw not in (None, "") else None version = int(existing.get("version") or 1) + 1 if existing else 1 payload = { "id": grp_id, "recipe_id": recipe_id, "name": gname, "distribution_type": gdist, "value": gval, "weight": weight, "order": order_value, "version": version, } _push_change(enterprise_id, "unloading_group", grp_id, payload) seen_grp_ids.add(grp_id) for raw_id in data.get("deleted_unloading_group_ids") or []: grp_id = str(raw_id or "").strip() if not grp_id or grp_id not in grp_by_id: continue existing = grp_by_id[grp_id] version = int(existing["version"] or 1) + 1 _push_change( enterprise_id, "unloading_group", grp_id, {"id": grp_id, "recipe_id": recipe_id, "name": existing["name"], "version": version}, action="delete", ) for grp_id, existing in grp_by_id.items(): if grp_id not in seen_grp_ids and grp_id not in {str(x) for x in (data.get("deleted_unloading_group_ids") or [])}: version = int(existing["version"] or 1) + 1 _push_change( enterprise_id, "unloading_group", grp_id, {"id": grp_id, "recipe_id": recipe_id, "name": existing["name"], "version": version}, action="delete", ) with session_scope() as db: saved = db.scalar( select(ZootechRecipe).where( ZootechRecipe.enterprise_id == enterprise_id, ZootechRecipe.id == recipe_id, ZootechRecipe.is_deleted.is_(False), ) ) if not saved: raise RecipeWriteError("Рецепт не найден после сохранения", 404) return { "success": True, "message": "Рецепт обновлен", "id": recipe_id, "stats": { "ingredients": len(seen_ing_ids), "unloading_groups": len(seen_grp_ids), }, } def recipe_write_http_error(exc: RecipeWriteError) -> HTTPException: return HTTPException(status_code=exc.status_code, detail={"message": exc.message, "error": True})