260705_2

B06_wf3_ProfileCross/B06_wf3_ProfileCross_Repository.py

@@ -0,0 +1,145 @@
+"""B06 종횡단 결과의 aiomysql Raw SQL 접근.
+
+longitudinal_sections(종단면 1건), cross_sections(측점별 다건) 테이블에
+메타데이터와 상대 경로를 기록한다. 상세 샘플 데이터는 파일에 저장하고 DB에는
+요약 data(JSON)와 경로만 기록한다.
+"""
+
+import json
+from pathlib import PurePosixPath
+from typing import Any
+from uuid import UUID
+
+import aiomysql
+
+_STAGE_ROOT = "B06_wf3_ProfileCross"
+
+
+def _validate_stage_path(relative_path: str) -> str:
+    normalized = PurePosixPath(relative_path.replace("\\", "/"))
+    if normalized.is_absolute() or ".." in normalized.parts:
+        raise ValueError("DB에는 프로젝트 루트 기준 상대 경로만 저장할 수 있습니다.")
+    if not normalized.parts or normalized.parts[0] != _STAGE_ROOT:
+        raise ValueError(f"B06 산출물 경로는 {_STAGE_ROOT} 아래여야 합니다.")
+    return normalized.as_posix()
+
+
+async def delete_sections_for_route(connection: aiomysql.Connection, route_id: int) -> None:
+    """경로 재생성 전에 기존 종횡단 레코드를 삭제한다 (멱등 재실행)."""
+    async with connection.cursor() as cursor:
+        await cursor.execute("DELETE FROM cross_sections WHERE route_id = %s", (route_id,))
+        await cursor.execute("DELETE FROM longitudinal_sections WHERE route_id = %s", (route_id,))
+
+
+async def create_longitudinal_section(
+    connection: aiomysql.Connection,
+    *,
+    project_id: UUID,
+    route_id: int,
+    data: dict[str, Any] | None,
+    longitudinal_file_path: str,
+    status: str = "DRAFT",
+) -> int:
+    """종단면 메타데이터를 저장하고 생성된 ID를 반환한다."""
+    file_rel = _validate_stage_path(longitudinal_file_path)
+    async with connection.cursor() as cursor:
+        await cursor.execute(
+            """
+            INSERT INTO longitudinal_sections (
+                project_id, route_id, computed_at, data, longitudinal_file_path, status
+            )
+            VALUES (%s, %s, CURRENT_TIMESTAMP, %s, %s, %s)
+            """,
+            (
+                str(project_id),
+                route_id,
+                json.dumps(data, ensure_ascii=False) if data is not None else None,
+                file_rel,
+                status,
+            ),
+        )
+        new_id = cursor.lastrowid
+    if not new_id:
+        raise RuntimeError("longitudinal_sections 레코드 생성 결과에 ID가 없습니다.")
+    return int(new_id)
+
+
+async def insert_cross_sections(
+    connection: aiomysql.Connection,
+    *,
+    project_id: UUID,
+    route_id: int,
+    sections: list[dict[str, Any]],
+) -> int:
+    """측점별 횡단면 레코드를 일괄 저장하고 저장 건수를 반환한다.
+
+    각 section dict: {chainage_m, sequence_num, data, cross_section_file_path, status?}
+    """
+    if not sections:
+        return 0
+    rows = []
+    for section in sections:
+        file_rel = _validate_stage_path(section["cross_section_file_path"])
+        data = section.get("data")
+        rows.append(
+            (
+                str(project_id),
+                route_id,
+                section.get("chainage_m"),
+                section.get("sequence_num"),
+                json.dumps(data, ensure_ascii=False) if data is not None else None,
+                file_rel,
+                section.get("status", "DRAFT"),
+            )
+        )
+    async with connection.cursor() as cursor:
+        await cursor.executemany(
+            """
+            INSERT INTO cross_sections (
+                project_id, route_id, chainage_m, sequence_num,
+                data, cross_section_file_path, status
+            )
+            VALUES (%s, %s, %s, %s, %s, %s, %s)
+            """,
+            rows,
+        )
+    return len(rows)
+
+
+async def get_longitudinal_section(
+    connection: aiomysql.Connection, project_id: UUID, route_id: int
+) -> dict[str, Any] | None:
+    """경로의 종단면 메타데이터를 조회한다 (없으면 None)."""
+    async with connection.cursor() as cursor:
+        await cursor.execute(
+            """
+            SELECT id, longitudinal_file_path, status, computed_at
+            FROM longitudinal_sections
+            WHERE project_id = %s AND route_id = %s
+            ORDER BY id DESC
+            LIMIT 1
+            """,
+            (str(project_id), route_id),
+        )
+        row = await cursor.fetchone()
+    if not row:
+        return None
+    return {
+        "id": int(row[0]),
+        "longitudinal_file_path": row[1],
+        "status": row[2],
+        "computed_at": row[3].isoformat() if row[3] else None,
+    }
+
+
+async def confirm_sections_for_route(connection: aiomysql.Connection, route_id: int) -> None:
+    """경로의 종횡단면 상태를 CONFIRMED로 변경한다."""
+    async with connection.cursor() as cursor:
+        await cursor.execute(
+            "UPDATE longitudinal_sections SET status = 'CONFIRMED' WHERE route_id = %s",
+            (route_id,),
+        )
+        await cursor.execute(
+            "UPDATE cross_sections SET status = 'CONFIRMED' WHERE route_id = %s",
+            (route_id,),
+        )