j"UdZddlmZddlZddlZddlZddlZddlZddlmZm Z ddl m Z ddl m Z mZmZmZmZddlmZe rddlmZej,eZddddd Zd ed <ej6d Zdd ZddZddZGddZ y)uCaptureQueue — append/flush 로 중간 생성물을 image_asset 으로 영속화 (Task A4). 저수준 capture(sink)는 spool 파일 경로 + 메타만 append 한다. step/runner 의 generation_context 종료 시 finally 에서 flush() 가: 1. 독립 SessionLocal() 세션을 열고 2. 각 항목을 ImageAsset(is_intermediate=True, ...) 로 insert 3. spool 파일을 최종 generated 경로로 rename 4. commit / close 한다. 전체를 try/except 로 감싸 어떤 실패도 raise 하지 않는다(독립 non-fatal 세션 — business transaction rollback 무관). 실패 시 rollback + CAPTURE_DIAG 증가. ★ reference_image_ids 는 절대 건드리지 않는다(char/prop lineage 보존). 전체 입력 lineage 는 input_image_ids 컬럼에 JSON 으로 저장한다. ) annotationsN)datetimetimezone)Path) TYPE_CHECKINGAnyDictListTuple) SessionLocal)GenerationContext)skipped_no_contextskipped_no_flush_scopeskipped_capture_off flush_errorszDict[str, int] CAPTURE_DIAGz[^A-Za-z0-9_.-]cbtjd|xsd}|dvst|dhk(ry|S)ustage 를 안전한 단일 path 세그먼트로. DB 엔 원문 저장, 파일 경로에만 사용. 허용 문자 ``[A-Za-z0-9_.-]`` 외는 ``_`` 치환. ``.``/``..``/빈 문자열 등 상위 디렉터리 탈출 위험 값은 ``unknown`` 으로(generated root 밖 금지)._)r.z..runknown)_UNSAFE_SEG_REsubset)stagesegs V/Users/manta/Documents/Projects/TheRoad-I1/backend/app/services/image_capture/queue.py _safe_segmentr3s7   S%+2 .C oScU!2 Jcdtjtjj SN)rnowrutc isoformatrr_nowr&>s << % / / 11rczddlm}t|j|z }|r|dz |z }|dz dz t |z S)u최종 generated 디렉터리(절대). episode 무관 자산은 project-level fallback. stage 는 ``_safe_segment`` 로 정화해 경로 탈출을 막는다(Codex BLOCKING 2).r)settingsepisodesimages generated)app.core.configr(r projects_dirr) project_id episode_idrr(bases r_generated_dirr1BsG) %% & 3Dj :- (?[ (=+? ??rc(eZdZdZddZddZddZy) CaptureQueueuW단일 capture scope 의 flush 큐. append=적재, flush=독립세션 일괄 영속화.cR||_g|_tj|_yr!)_ctx_items threadingLock_lock)selfctxs r__init__zCaptureQueue.__init__Qs  8: ^^% rc|j5|jj|t|xsifdddy#1swYyxYw)uFspool 파일 경로 + 메타를 적재(DB 접근 없음). thread-safe.N)r9r6appenddict)r: spool_pathmetas rr>zCaptureQueue.appendYs3 ZZ KK   D,<= >ZZs +AA c |j5|js dddy|j}g|_ddd|j}d}d} ddlm}|ddlm}t}D]\}}ttj} |jdxs |j} t|j|j| } | j!dd| | dz } t#|} | j%| |jd }|jd }| t/|nd}|j0 |i}|j3d |j0|d$id | d |jd|jddddd| d|jdd |t5j6|ndd|jdd|jdd|jdd|jdd|t5j6|ndd|j8d|j:d|j<d|jddt| dddt?}|jA||d z }|jC|| |jESS#1swYxYw#t&$rD| j)| j+ | j-n#t&$rYnwxYwYwxYw#tF$rYSwxYw#tF$r{tHd!xxd z cc<tJjMd"tOd#|! |jQn#tF$rYnwxYwY|! |jEy#tF$rYywxYwywxYw#|! |jEw#tF$rYwwxYwwxYw)%u1독립 세션으로 적재 항목을 ImageAsset 으로 insert + spool→final rename. Returns: insert 한 행 수. 어떤 실패도 raise 하지 않는다(non-fatal). 실패 시 rollback + CAPTURE_DIAG["flush_errors"] 증가 후 0 반환. 큐는 항상 비운다. Nr)register_models) ImageAssetrT)parentsexist_okz.pnginput_image_idspipeline_metadata scene_indexidr.r/ asset_typer+is_intermediate pipeline_rolegeneration_call_idcandidate_index disposition attempt_indexpipeline_metadata_json entity_idstill_id shot_index prompt_usedprompt file_pathstatus created_atrumCaptureQueue.flush failed (non-fatal) — %d item(s) dropped (renamed files may be orphaned in generated dir))exc_infor%))r9r6r5app.core.databaserCapp.models.projectrDr struuiduuid4getrr1r.r/mkdirrrenameOSError write_bytes read_bytesunlinkr?rI setdefaultjsondumpsrSrTrUr&addcommitclose Exceptionrloggerwarninglenrollback)r:itemsr;sessioninsertedrCrDr@rAaidrgen_dir final_pathsp input_ids pipeline_metapmrows rflushzCaptureQueue.flush^sZZ;;ZKKEDK  ii^  :   5"nG$) D$**,')6SYY(O dT :$#d|3 *%IIj)!HH%67 $)< = -:,ET-(4??.zMM-A "~~ #~~ +  %)   #'((?";2;1F 9-D(,xx0D'E%)HH->$?!% 7#'((?"; +-. 2d!$"mm%&!\\'( #~~)*!% 2+."*o/0'12 $v36 C A u%*x NN $"MMO#GZ>**2==?; " N!)   (A - ( NNCE    "$$& "MMO ## ""MMO #sJ4J4B)L :K FL "L4J>(L*K;:L; LLLL L LL  LL =N$M/.N$/ M;8N$:M;;N$>N'N NN#N$$N''O +N<;O < OO OO N)r;z'GenerationContext'returnNone)r@r_rAzDict[str, Any]rr)rint)__name__ __module__ __qualname____doc__r<r>rr%rrr3r3Nsa&? nrr3)r str | Nonerr_)rr_)r.r_r/rrr_rr)!r __future__rrjloggingrer7r`rrpathlibrtypingrrr r r r]r "app.services.image_capture.contextr getLoggerrrpr__annotations__compilerrr&r1r3r%rrrs #   '88*D   8 $   n./2 @~~r