Files
Pier_v1/RAG/design-model-box-sample_2/tools/recover_wing_geometry_checkpoint.py
T
ahnyongjooandClaude Sonnet 5.5 964a841de1 chore(rebar): Pier_T_rebar 작업 폴더 초기 구성 및 외부 의존 파일 통합
- Code\rebar_pipeline, review_pipeline, pier-program을 정션에서 실제 폴더로 전환
- Code\rhino_mcp 및 CLAUDE_rhino배경.md 추가, .mcp.json은 PYTHONPATH로 새 위치 사용
- pier-program 내부 .git 제거(이력은 backup\pier-program_20261028.bundle에 보관)
- worktree 가이드 반영(CLAUDE.local.md, HANDOFF.md)
- D:\Rhino-MCP 경로 참조를 새 위치로 치환

Co-Authored-By: Claude Sonnet 5.5 <noreply@anthropic.com>
2026-10-28 11:20:18 +09:00

85 lines
4.1 KiB
Python

"""Recover a stopped shape checkpoint using only current drawing adapters."""
import argparse
import copy
import json
import os
from pathlib import Path
import sys
import tempfile
import time
sys.path.insert(0, str(Path(__file__).resolve().parents[1] / 'src' / 'backend'))
import server
def recover(job):
if job.get('status') not in ('failed', 'cancelled') or job.get('result'):
raise ValueError('only stopped shape checkpoints without a final result')
request = job['requestBody']
if request.get('concreteApproved'):
raise ValueError('approved/rebar models are outside this repair')
partial = job['partialResult']
original = partial['exteriorEvidence']
exterior = copy.deepcopy(original)
drawing = copy.deepcopy(partial['drawingEvidence'])
analysis = server.rehydrate_drawing_set_analysis(request['analysis'])
exterior = server.normalize_direct_dxf_wing_footprints(analysis, exterior, drawing)
exterior = server.enforce_wing_haunch_object_contract(exterior)
exterior = server.align_wing_root_bottoms_to_side_walls(exterior, analysis)
exterior = server.enforce_wing_haunch_object_contract(exterior)
audit = exterior['wing_haunch_object_contract_audit']
if not audit['pass'] or audit['haunch_count'] != audit['wing_count'] or any(
r['status'] != 'resolved' for r in audit['contact_rows']):
raise ValueError('wing/haunch geometry or contact still unresolved')
before = {c['id']: c for c in original['components']}
after = {c['id']: c for c in exterior['components']}
for key, value in before.items():
if value.get('kind') not in ('wingwall', 'wing_wall', 'wingwall_haunch'):
assert after.get(key) == value, 'unrelated concrete changed'
assert exterior.get('rebar_groups') == original.get('rebar_groups')
checkpoint = server.exterior_feedback_snapshot(drawing, exterior)
result = dict(summary='날개벽 외곽 및 접합헌치 복구 · 전체 형상 검토 대기',
category_id='box_culvert', concrete_components=copy.deepcopy(exterior['components']),
rebar_groups=copy.deepcopy(exterior.get('rebar_groups') or []),
drawingEvidence=checkpoint['drawingEvidence'],
unresolved=copy.deepcopy(exterior.get('unresolved') or []),
ready_for_geometry=False, awaiting_concrete_confirmation=True,
workflowStage='shape', wingGeometryRecoveryAudit=audit)
updated = copy.deepcopy(job)
updated['previousFailureDiagnostic'] = updated.pop('failureDiagnostic', None)
updated.update(status='awaiting_feedback', error=None, verified=False, progress=100,
stage='날개벽·접합헌치 복구 완료 · 전체 형상 검토 대기',
result=result, partialResult=checkpoint, geometryRevision=time.time_ns(),
pipelineVersion=server.MODEL_PIPELINE_VERSION,
sourceContext=server.model_job_source_context(request))
return updated
def main():
parser = argparse.ArgumentParser(description=__doc__)
parser.add_argument('job', type=Path)
parser.add_argument('--apply', action='store_true')
args = parser.parse_args()
server.require_modeling_rules()
target = args.job.resolve()
raw = target.read_bytes()
updated = recover(json.loads(raw))
print(json.dumps(updated['result']['wingGeometryRecoveryAudit'], ensure_ascii=False))
if not args.apply:
return
if target.read_bytes() != raw:
raise ValueError('concurrent job update')
backup = target.with_name(target.stem + f'.before-wing-recovery-{time.time_ns()}.json')
with backup.open('xb') as stream:
stream.write(raw)
with tempfile.NamedTemporaryFile(mode='w', dir=target.parent, delete=False) as stream:
json.dump(updated, stream, ensure_ascii=False, allow_nan=False)
temp = stream.name
if target.read_bytes() != raw:
raise ValueError('concurrent update before commit')
os.replace(temp, target)
print(json.dumps({'saved': str(target), 'backup': str(backup)}))
if __name__ == '__main__':
main()