eomsangdon
f99f1f68d3
파일을 올리면 지표면 확정 직후 "노선 설계로 이동하세요"를 적고 그 뒤에 초기 설계 체인이 돈다. 그 사이 B05에 들어가 만지면 반쯤 계산된 값을 편집하게 되고, 그 편집이 섞인 채 초기값 스냅샷이 찍힌다. 계산이 끝날 때까지 준비 화면(B11)에서 기다리게 한다. - initial_snapshot: 마커(initial_design.lock) 생성·해제·조회와 스냅샷 무효화, 편집 정본 제거를 한곳에 둔다. 마커는 스냅샷 4트리 밖(프로젝트 루트)이라 복원에 딸려 들어가지 않는다. - 체인 두 곳(자동·재설계): 시작에 마커, finally에서 해제. 재설계 체인은 끝에 스냅샷을 무효화한다 — 사용자 입력을 유지한 채 재계산하므로 그 결과를 찍으면 가짜 초기값이 된다. 다음 [초기화]가 폴백을 타며 새 초기값을 만들고 그때 촬영한다. - /surface/status: 마커가 있으면 in_progress + initial_design으로 답한다. stage 1이 COMPLETE면 진행률 파일을 무시하는 기존 분기를 건드리지 않으려고 별도 확인이다. - 진입 게이트(routeAfterPreloadCheck)와 B11이 그 상태를 보고 기다린다. 상태 조회가 실패하면 막지 않고, B11 대기는 20분 상한을 둔다. - [초기화] 폴백: 체인 전에 structures.json과 drainage/edits를 지운다. 남기면 구조물 측점이 사용자 편집분에서 다시 파생돼 초기값이 오염된다(2026-08-29 실측). 테스트 4건 추가. 249 passed·8 failed(기슭막이 이관 때 생긴 기존 실패, 변경 전 동일).
219 lines
8.8 KiB
Python
219 lines
8.8 KiB
Python
"""초기값 스냅샷 — 자동설계 체인이 만든 첫 결과를 그대로 떠 두고 [초기화]가 되돌린다.
|
|
|
|
CLAUDE.md 5장(조작·데이터 흐름 정책)의 **초기값** 층이다. 자동설계 체인이 끝난 직후
|
|
한 번 찍고, 그 뒤로는 읽기 전용이다 — 어떤 저장 경로도 이 폴더에 쓰지 않는다.
|
|
|
|
[초기화]가 재계산이 아니라 복원이어야 하는 이유: [저장]·[확정]이 새 행을 만들지 않고
|
|
최신 `routes` 행을 제자리 갱신하고 종·횡단 정본 파일도 덮어쓰므로, 초기 상태는 따로
|
|
떠 두지 않으면 남지 않는다. 재계산으로 되살리려 해도 `structures.json`·
|
|
`pipe_points.json` 편집분이 그대로 남아 초기값과 다른 결과가 나온다(2026-08-29).
|
|
"""
|
|
|
|
import json
|
|
import shutil
|
|
from pathlib import Path
|
|
from typing import Any
|
|
|
|
import aiomysql
|
|
|
|
# 스냅샷 폴더는 워크플로우 단계가 아니므로 PROJECT_STORAGE_LAYOUT_V2에 넣지 않는다.
|
|
SNAPSHOT_DIRNAME = "initial_snapshot"
|
|
_DB_DUMP_NAME = "db.json"
|
|
# 초기 설계 체인이 도는 동안만 존재하는 마커(진입 차단 판정용).
|
|
DESIGNING_LOCK_NAME = "initial_design.lock"
|
|
|
|
# 프로젝트 루트 기준 상대 경로 — 정본 파일이 사는 자리 전부.
|
|
_FILE_TREES = (
|
|
"B05_Profile/route",
|
|
"B06_Section/longitudinal",
|
|
"B06_Section/cross_sections",
|
|
"B04_PreProcess/drainage/edits",
|
|
)
|
|
|
|
# `routes.id`를 참조하는 표는 스키마상 이 넷이 전부다(001_create_schema.sql:539~556).
|
|
_CHILD_TABLES = ("route_points", "route_statistics", "longitudinal_sections", "cross_sections")
|
|
|
|
|
|
def snapshot_dir(project_root: Path) -> Path:
|
|
return Path(project_root) / SNAPSHOT_DIRNAME
|
|
|
|
|
|
def designing_lock_path(project_root: Path) -> Path:
|
|
"""초기 설계 체인이 도는 동안만 존재하는 마커.
|
|
|
|
이게 있으면 B05·B06은 아직 들어갈 때가 아니다(CLAUDE.md 5장 — 계산 중 편집이 섞이면
|
|
초기값이 오염된다). 스냅샷 대상 4트리 **밖**인 프로젝트 루트에 두어 복원에 딸려
|
|
들어가지 않게 한다. `workflow_state`의 stage 2는 체인이 끝나도 IN_PROGRESS라 판정에
|
|
못 쓰고, route 존재 여부도 체인 3단계에서 이미 생겨 못 쓴다.
|
|
"""
|
|
return Path(project_root) / DESIGNING_LOCK_NAME
|
|
|
|
|
|
def is_designing(project_root: Path) -> bool:
|
|
return designing_lock_path(project_root).is_file()
|
|
|
|
|
|
def mark_designing(project_root: Path) -> None:
|
|
path = designing_lock_path(project_root)
|
|
try:
|
|
path.parent.mkdir(parents=True, exist_ok=True)
|
|
path.write_text("designing", encoding="utf-8")
|
|
except OSError:
|
|
pass # 마커 실패가 체인을 막지는 않는다 — 문이 일찍 열릴 뿐이다.
|
|
|
|
|
|
def clear_designing(project_root: Path) -> None:
|
|
try:
|
|
designing_lock_path(project_root).unlink(missing_ok=True)
|
|
except OSError:
|
|
pass
|
|
|
|
|
|
def discard_initial_snapshot(project_root: Path) -> bool:
|
|
"""초기값을 무효화한다 — 지표면·노선이 바뀌어 옛 초기값이 더는 기준이 아닐 때.
|
|
|
|
그 자리에서 다시 뜨지 않는다. 재설계 체인은 **사용자 입력을 유지한 채** 재계산하므로
|
|
그 결과를 찍으면 사용자 편집이 섞인 가짜 초기값이 된다. 지워 두면 다음 [초기화]가
|
|
재계산 폴백을 타면서 새 초기값을 만들고, 체인이 그때 촬영한다(2026-08-29 사용자 확정).
|
|
"""
|
|
target = snapshot_dir(Path(project_root))
|
|
if not target.exists():
|
|
return False
|
|
shutil.rmtree(target, ignore_errors=True)
|
|
return True
|
|
|
|
|
|
def has_initial_snapshot(project_root: Path) -> bool:
|
|
return (snapshot_dir(project_root) / _DB_DUMP_NAME).is_file()
|
|
|
|
|
|
def _copy_tree(source: Path, target: Path) -> None:
|
|
if not source.is_dir():
|
|
return
|
|
if target.exists():
|
|
shutil.rmtree(target)
|
|
shutil.copytree(source, target)
|
|
|
|
|
|
async def _dump_rows(
|
|
connection: aiomysql.Connection, table: str, route_id: int
|
|
) -> list[dict[str, Any]]:
|
|
async with connection.cursor(aiomysql.DictCursor) as cursor:
|
|
# 표 이름은 이 모듈의 상수에서만 오므로 자리표시자 대상이 아니다.
|
|
await cursor.execute(f"SELECT * FROM {table} WHERE route_id = %s", (route_id,)) # noqa: S608
|
|
return list(await cursor.fetchall())
|
|
|
|
|
|
def _json_safe(rows: list[dict[str, Any]]) -> list[dict[str, Any]]:
|
|
"""TIMESTAMP 등 JSON이 모르는 값을 문자열로 낮춘다."""
|
|
return [
|
|
{key: (value if _is_json_native(value) else str(value)) for key, value in row.items()}
|
|
for row in rows
|
|
]
|
|
|
|
|
|
def _is_json_native(value: Any) -> bool:
|
|
return value is None or isinstance(value, (bool, int, float, str, list, dict))
|
|
|
|
|
|
async def save_initial_snapshot(
|
|
connection: aiomysql.Connection, project_root: Path, route_id: int
|
|
) -> None:
|
|
"""자동설계 체인 성공 직후 한 번 부른다. 이미 있으면 덮어쓰지 않는다."""
|
|
root = Path(project_root)
|
|
target = snapshot_dir(root)
|
|
if has_initial_snapshot(root):
|
|
return
|
|
target.mkdir(parents=True, exist_ok=True)
|
|
|
|
for tree in _FILE_TREES:
|
|
_copy_tree(root / tree, target / tree.replace("/", "__"))
|
|
|
|
async with connection.cursor(aiomysql.DictCursor) as cursor:
|
|
await cursor.execute("SELECT * FROM routes WHERE id = %s", (route_id,))
|
|
route = await cursor.fetchone()
|
|
if not route:
|
|
return
|
|
dump: dict[str, Any] = {"routes": _json_safe([dict(route)])}
|
|
for table in _CHILD_TABLES:
|
|
dump[table] = _json_safe(await _dump_rows(connection, table, route_id))
|
|
(target / _DB_DUMP_NAME).write_text(json.dumps(dump, ensure_ascii=False), encoding="utf-8")
|
|
|
|
|
|
def wipe_edited_masters(project_root: Path) -> list[str]:
|
|
"""사용자 편집 정본을 걷어낸다 — 재계산으로 **진짜 초기값**을 만들기 위한 사전 정리.
|
|
|
|
자동설계 체인은 계획노선 CSV와 config 기본값으로 도는 결정적 절차라, 이 둘만 없으면
|
|
원래 나왔어야 할 값이 나온다. 남겨 두면 `resolve_extra_stations`가 사용자가 편집한
|
|
관·구조물에서 측점을 다시 파생해 "초기값"이 오염된다(2026-08-29 실측에서 확인).
|
|
|
|
스냅샷 복원 경로에서는 부르지 않는다 — 거기서는 스냅샷본이 통째로 덮어쓴다.
|
|
"""
|
|
removed: list[str] = []
|
|
root = Path(project_root)
|
|
for rel in ("B05_Profile/route/structures.json", "B04_PreProcess/drainage/edits"):
|
|
path = root / rel
|
|
try:
|
|
if path.is_dir():
|
|
shutil.rmtree(path)
|
|
removed.append(rel)
|
|
elif path.is_file():
|
|
path.unlink()
|
|
removed.append(rel)
|
|
except OSError:
|
|
pass # 지우지 못하면 그만큼 초기값이 덜 깨끗할 뿐, 재계산은 계속한다.
|
|
return removed
|
|
|
|
|
|
def restore_snapshot_files(project_root: Path) -> None:
|
|
"""스냅샷 파일을 작업본 자리에 되돌린다. 스냅샷에 없던 트리는 비워 둔다."""
|
|
root = Path(project_root)
|
|
source = snapshot_dir(root)
|
|
for tree in _FILE_TREES:
|
|
_copy_tree(source / tree.replace("/", "__"), root / tree)
|
|
|
|
|
|
async def restore_initial_snapshot(
|
|
connection: aiomysql.Connection, project_root: Path, project_id: str
|
|
) -> int | None:
|
|
"""`routes`와 자식 4표를 스냅샷 값으로 다시 세우고 새 route id를 돌려준다.
|
|
|
|
호출자가 기존 `routes` 행을 지운 **뒤에** 부른다(자식은 FK CASCADE로 함께 지워진다).
|
|
파일 복원은 트랜잭션 밖이라 `restore_snapshot_files()`를 따로 부른다.
|
|
"""
|
|
dump_path = snapshot_dir(Path(project_root)) / _DB_DUMP_NAME
|
|
if not dump_path.is_file():
|
|
return None
|
|
dump = json.loads(dump_path.read_text(encoding="utf-8"))
|
|
route = (dump.get("routes") or [None])[0]
|
|
if not route:
|
|
return None
|
|
|
|
route = dict(route)
|
|
route.pop("id", None)
|
|
route["project_id"] = project_id
|
|
new_id = await _insert_row(connection, "routes", route)
|
|
|
|
for table in _CHILD_TABLES:
|
|
for row in dump.get(table, []):
|
|
child = dict(row)
|
|
child.pop("id", None)
|
|
child["route_id"] = new_id
|
|
if "project_id" in child:
|
|
child["project_id"] = project_id
|
|
await _insert_row(connection, table, child)
|
|
return new_id
|
|
|
|
|
|
async def _insert_row(connection: aiomysql.Connection, table: str, row: dict[str, Any]) -> int:
|
|
columns = list(row.keys())
|
|
placeholders = ", ".join(["%s"] * len(columns))
|
|
names = ", ".join(f"`{name}`" for name in columns)
|
|
async with connection.cursor() as cursor:
|
|
# 표 이름은 상수, 열 이름은 스냅샷이 뜬 실제 스키마에서 온다.
|
|
await cursor.execute(
|
|
f"INSERT INTO {table} ({names}) VALUES ({placeholders})", # noqa: S608
|
|
tuple(row[name] for name in columns),
|
|
)
|
|
return int(cursor.lastrowid)
|