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

2026-07-17 12:57:18 +03:00
parent 5dfa06ddbe
commit 355c0ef9f1
883 changed files with 194576 additions and 177 deletions
@@ -0,0 +1,161 @@
+"""Фабрики данных для тестов кормораздатчика (period_recipes)."""
+
+from __future__ import annotations
+
+from datetime import datetime
+from typing import List, Tuple
+
+from sqlalchemy import select
+
+from app import db
+from app.models import (
+    Component,
+    FeedDispenser,
+    FeedingPeriod,
+    Ingredient,
+    PeriodRecipe,
+    Recipe,
+    UnloadingGroup,
+)
+
+E2E_DISP_ID = "e2e-disp-1"
+E2E_PERIOD_A = "e2e-period-a"
+E2E_PERIOD_B = "e2e-period-b"
+E2E_DISP_RECIPE_1 = "e2e-disp-recipe-1"
+E2E_DISP_RECIPE_2 = "e2e-disp-recipe-2"
+E2E_DISP_COMP = "e2e-disp-comp"
+
+
+def seed_dispenser_period_recipes() -> Tuple[str, str, List[str]]:
+    """Кормораздатчик + 2 периода + 2 рецепта в period A. Возвращает (disp_id, period_a_id, recipe_ids)."""
+    now = datetime.now()
+    db.session.add_all(
+        [
+            FeedDispenser(
+                id=E2E_DISP_ID,
+                name="E2E Раздатчик",
+                farm="Ферма",
+                operator="Тест",
+                type="dispenser",
+                content_hash="",
+            ),
+            FeedingPeriod(id=E2E_PERIOD_A, name="E2E Утро", dispenser_id=E2E_DISP_ID),
+            FeedingPeriod(id=E2E_PERIOD_B, name="E2E Вечер", dispenser_id=E2E_DISP_ID),
+            Component(
+                id=E2E_DISP_COMP,
+                name="E2E Disp компонент",
+                type="grain",
+                dry_matter=55.0,
+                protein=0.0,
+                energy=0.0,
+                price=0.0,
+            ),
+            Recipe(
+                id=E2E_DISP_RECIPE_1,
+                name="E2E Рейс 1",
+                heads_per_trip=20,
+                mixing_time=3,
+                trip_percent=100.0,
+                target_component_id=E2E_DISP_COMP,
+                content_hash="",
+            ),
+            Recipe(
+                id=E2E_DISP_RECIPE_2,
+                name="E2E Рейс 2",
+                heads_per_trip=15,
+                mixing_time=4,
+                trip_percent=100.0,
+                target_component_id=E2E_DISP_COMP,
+                content_hash="",
+            ),
+        ]
+    )
+    db.session.flush()
+    for idx, (rid, ing_id) in enumerate(
+        ((E2E_DISP_RECIPE_1, "e2e-disp-ing-1"), (E2E_DISP_RECIPE_2, "e2e-disp-ing-2")),
+        1,
+    ):
+        db.session.add(
+            Ingredient(
+                id=ing_id,
+                name=f"Ing {idx}",
+                weight_per_head=float(idx),
+                amount=float(idx * 10),
+                dry_matter=55.0,
+                component_id=E2E_DISP_COMP,
+                order=1,
+                recipe_id=rid,
+                created_by="system",
+                updated_by="system",
+            )
+        )
+        db.session.add(
+            UnloadingGroup(
+                id=f"e2e-disp-grp-{idx}",
+                name=f"Г{idx}",
+                distribution_type="percent",
+                value=100.0,
+                weight=10.0,
+                order=1,
+                recipe_id=rid,
+                created_by="system",
+                updated_by="system",
+            )
+        )
+    for ord_, rid in enumerate((E2E_DISP_RECIPE_1, E2E_DISP_RECIPE_2), 0):
+        db.session.add(
+            PeriodRecipe(
+                period_id=E2E_PERIOD_A,
+                recipe_id=rid,
+                order=ord_,
+                created_at=now,
+                created_by="system",
+                updated_by="system",
+            )
+        )
+    db.session.commit()
+    return E2E_DISP_ID, E2E_PERIOD_A, [E2E_DISP_RECIPE_1, E2E_DISP_RECIPE_2]
+
+
+def reset_e2e_dispenser_period_state() -> None:
+    """Восстановить рейсы 1–2 в period A после мутирующих E2E (перенос, удаление, порядок)."""
+    now = datetime.now()
+    period_b_rows = db.session.execute(
+        select(PeriodRecipe).where(
+            PeriodRecipe.period_id == E2E_PERIOD_B,
+            PeriodRecipe.is_deleted.is_(False),
+        )
+    ).scalars().all()
+    for row in period_b_rows:
+        db.session.delete(row)
+    db.session.flush()
+    for rid in (E2E_DISP_RECIPE_1, E2E_DISP_RECIPE_2):
+        row = db.session.get(PeriodRecipe, {"period_id": E2E_PERIOD_A, "recipe_id": rid})
+        if row is not None:
+            db.session.delete(row)
+    db.session.flush()
+    for ord_, rid in enumerate((E2E_DISP_RECIPE_1, E2E_DISP_RECIPE_2)):
+        db.session.add(
+            PeriodRecipe(
+                period_id=E2E_PERIOD_A,
+                recipe_id=rid,
+                order=ord_,
+                created_at=now,
+                updated_at=now,
+                created_by="system",
+                updated_by="system",
+            )
+        )
+    db.session.commit()
+
+
+def get_period_recipe_order(period_id: str) -> List[str]:
+    rows = db.session.execute(
+        select(PeriodRecipe.recipe_id)
+        .where(
+            PeriodRecipe.period_id == period_id,
+            PeriodRecipe.is_deleted.is_(False),
+        )
+        .order_by(PeriodRecipe.order.asc())
+    ).scalars().all()
+    return [str(x) for x in rows]