Initial commit: svg backend

backend/app/repositories/scheme_sectors.py

@@ -0,0 +1,74 @@
+import json
+from uuid import uuid4
+
+from sqlalchemy import asc, delete, select
+
+from app.db.session import AsyncSessionLocal
+from app.models.scheme_sector import SchemeSectorRecord
+
+
+async def replace_scheme_version_sectors(
+    *,
+    scheme_id: str,
+    scheme_version_id: str,
+    sectors: list[dict],
+) -> None:
+    async with AsyncSessionLocal() as session:
+        await session.execute(
+            delete(SchemeSectorRecord).where(
+                SchemeSectorRecord.scheme_version_id == scheme_version_id
+            )
+        )
+
+        for item in sectors:
+            row = SchemeSectorRecord(
+                sector_record_id=uuid4().hex,
+                scheme_id=scheme_id,
+                scheme_version_id=scheme_version_id,
+                element_id=item.get("id"),
+                sector_id=item.get("sector_id"),
+                name=item.get("sector_id") or item.get("id") or "unnamed-sector",
+                classes_raw=json.dumps(item.get("classes", []), ensure_ascii=False),
+            )
+            session.add(row)
+
+        await session.commit()
+
+
+async def list_scheme_version_sectors(scheme_version_id: str) -> list[SchemeSectorRecord]:
+    async with AsyncSessionLocal() as session:
+        result = await session.execute(
+            select(SchemeSectorRecord)
+            .where(SchemeSectorRecord.scheme_version_id == scheme_version_id)
+            .order_by(asc(SchemeSectorRecord.id))
+        )
+        return list(result.scalars().all())
+
+
+async def clone_scheme_version_sectors(
+    *,
+    source_scheme_version_id: str,
+    target_scheme_version_id: str,
+) -> None:
+    async with AsyncSessionLocal() as session:
+        result = await session.execute(
+            select(SchemeSectorRecord).where(
+                SchemeSectorRecord.scheme_version_id == source_scheme_version_id
+            )
+        )
+        rows = list(result.scalars().all())
+
+        for row in rows:
+            session.add(
+                SchemeSectorRecord(
+                    sector_record_id=uuid4().hex,
+                    scheme_id=row.scheme_id,
+                    scheme_version_id=target_scheme_version_id,
+                    element_id=row.element_id,
+                    sector_id=row.sector_id,
+                    name=row.name,
+                    classes_raw=row.classes_raw,
+                )
+            )
+
+        await session.commit()