^(jY4dZddlmZddlmZGddZehdZehdZde d e fd Z d Z d e d e fd ZdZde d e fdZde d eeeee ffdZde de d e fdZdede d e fdZd+de de de d e fdZd e d eeee ffdZdee de de d ee fd Zde d!ed"ed#e d ee f d$Z d,d!ed"ed#e d&eee d'ee de de d(eed)ed eee effd*Zy%)-zp Job utilities for the /api/jobs endpoint. Provides normalization and helper functions for job status tracking. )Optional) prune_dictc2eZdZdZdZdZdZdZdZeeeeegZ y) JobStatuszJob status constants.pending in_progress completedfailed cancelledN) __name__ __module__ __qualname____doc__PENDING IN_PROGRESS COMPLETEDFAILED CANCELLEDALL>/Users/danicosta/Desktop/Flux2/ComfyUI/comfy_execution/jobs.pyrr s.GKI FI KFI >Crr>3dtextaudiovideoimages>.fbx.glb.obj.gltf.usdzfilenamereturncT|jtfdtDS)Nc3@K|]}j|ywNendswith).0extlowers r z#has_3d_extension..sA.@su~~c".@)r,anyTHREE_D_EXTENSIONS)r#r,s @rhas_3d_extensionr1s! NN E A.@A AArct|yt|trt|r|ddddSyt|tr|Sy)zNormalize a single output list item for the jobs API. Returns the normalized item, or None to exclude it. String items with 3D extensions become {filename, type, subfolder} dicts. Noutputr)r#type subfolder mediaType) isinstancestrr1dict)items rnormalize_output_itemr<"sC  |$ D ! $hRVZ[ [$ routputsc<i}|jD]\}}t|ts|||<i}|jD]Q\}}|dk(st|ts|||<!g}|D]%}|t |} |j | | n|'|||<S|||<|S)zNormalize raw node outputs for the jobs API. Transforms string 3D filenames into file output dicts and removes None items. All other items (non-3D strings, dicts, etc.) are preserved as-is. animated)itemsr8r:listr<append) r= normalizednode_id node_outputsnormalized_node media_typer@normalized_itemsr;norms rnormalize_outputsrJ3sJ!(,-".Jw  !-!3!3!5 JZ'z%/F.3 +! <,T2 ''0@dK  +;OJ '"6. 7!"1" rivaluecFt|tkrd|iS|dtddS)zCreate a text preview dict with optional truncation. Returns: dict with 'content' and optionally 'truncated' flag contentNT)rM truncated)lenTEXT_PREVIEW_MAX_LENGTH)rKs r_create_text_previewrQRs5  5z,,5!!112 r extra_datac|jd}|jdi}|jdijd}||fS)zqExtract create_time and workflow_id from extra_data. Returns: tuple: (create_time, workflow_id) create_time extra_pnginfoworkflowidget)rRrTrU workflow_ids r_extract_job_metadatar[`sJ ../KNN?B7M##J377=K  ##rrGr;c|tvry|jdd}|r#|jds|jdry|jddjt fdt Dryy) au Check if an output item is previewable. Matches frontend logic in ComfyUI_frontend/src/stores/queueStore.ts Maintains backwards compatibility with existing logic. Priority: 1. media_type is 'images', 'video', 'audio', or '3d' 2. format field starts with 'video/' or 'audio/' 3. filename has a 3D extension (.obj, .fbx, .gltf, .glb, .usdz) Tformatr4zvideo/zaudio/r#c3@K|]}j|ywr'r()r*r+r#s rr-z!is_previewable..s @-?c8  S !-?r.F)PREVIEWABLE_MEDIA_TYPESrY startswithr,r/r0)rGr;fmtr#s @ris_previewablerblso,, ((8R C x(CNN8,Dxx B'--/H @-? @@ rstatusc R|\}}}}}t|\}}t||||d|dS)zvConvert queue item tuple to unified job dict. Expects item with sensitive data already removed (5 elements). r)rWrcpriorityrT outputs_countrZ)r[r)r;rcre prompt_id_rRrTrZs rnormalize_queue_itemrisG -1)HiJ4Z@K ""  rrg history_iteminclude_outputsc |d}|\}}}}}t|\}} |jdi} | r| jdnd} |jdi} t| \} }d}d}d}d}| r| jdg}|D]}t|tt fst |dk\s)|d |d }}t|tsD|d k(r|jd }[|d vs`|jd }|dk(r|}y|dk(sd}| dk(rtj}n8| dk(r#|rtjntj}ntj}t|||||||| || d }|rt| |d<| |d<||d|d<|S)zConvert history item dict to unified job dict. History items have sensitive data already removed (prompt tuple has 5 elements). promptrc status_strNr=Fmessagesrexecution_start timestamp)execution_successexecution_errorexecution_interruptedrurvTsuccesserror) rWrcrerTexecution_start_timeexecution_end_timerurfpreview_outputrZexecution_status)rmrRrV)r[rYget_outputs_summaryr8rAtuplerOr:rrrrrrJ)rgrjrk prompt_tuplererhrmrRrTrZ status_infornr=rfr{ruryrzwas_interruptedroentry event_name event_datarcjobs rnormalize_history_itemrs  )L)5&HaQ4Z@K""8R0K2=.4Jy"-G$7$@!M>OO??:r2E%$/CJ!O).q58J j$/!%66/9~~k/J,#'hh-7^^K-H*%)::.8O'+BB.2OY$$ w (7$$Y=M=M$$ " 40*&("  C*73I"- $ J Jrc0d}d}d}|jD]\}}t|ts|jD]\}}|dk(st|ts|D]}t|tsYt |} | J|dk(rD|dz }|=t|t r |r|dnd} n t |} t| } i| ||d} || }j| }|dz }|tt||si|d|i} d |vr|| d <|jd d k(r| }|| }||xs|fS) z Count outputs and find preview in a single pass. Returns (outputs_count, preview_output). Preview priority (matching frontend): 1. type="output" with previewable media 2. Any previewable media rNr?rrqr4)nodeIdr7rr7r5r3) r@r8r:rAr<r~r9rQrbrY) r=countr{fallback_previewrDrErGr@r;rC text_value text_previewenricheds rr}r}sw EN!(,- !-!3!3!5 JZ'z%/F!$-!6t!.get_sort_key"s7GG2A6E''.2C"%%3; 6Q 6rc&|jddS)NrTrrX)rs rrz#apply_sorting..get_sort_key's77=!, ,r)keyreverse)sorted)rrrrrs r apply_sortingrs0V#G&& 7  - $L' ::rrunningqueuedhistoryc||vrt|||dS|D]&}|d|k(s t|tjcS|D]&}|d|k(s t|tjcSy)aY Get a single job by prompt_id from history or queue. Args: prompt_id: The prompt ID to look up running: List of currently running queue items queued: List of pending queue items history: Dict of history items keyed by prompt_id Returns: Job dict with full details, or None if not found T)rkrqN)rrirrr)rgrrrr;s rget_jobr-szG%i1CUYZZ 7i 'i.C.CD D 7i 'i.?.?@ @ rN status_filterrZlimitoffsetc g} |tj}tj|vr0|D]+} | jt | tj-tj |vr0|D]+} | jt | tj -tj tjtjh} | t|z} | rI|jD]6\} }t| |}|jd| vs&| j|8|r$| Dcgc]}|jd|k(s|} }t| ||} t| }|dkDr| |d} || d|} | |fScc}w)ae Get all jobs (running, pending, completed) with filtering and sorting. Args: running: List of currently running queue items queued: List of pending queue items history: Dict of history items keyed by prompt_id status_filter: List of statuses to include (from JobStatus.ALL) workflow_id: Filter by workflow ID sort_by: Field to sort by ('created_at', 'execution_duration') sort_order: 'asc' or 'desc' limit: Maximum number of items to return offset: Number of items to skip Returns: tuple: (jobs_list, total_count) NrcrZr)rrrrBrirrrrsetr@rrYrrO)rrrrrZrrrrrr;history_statusesrequested_history_statusesrgrjrj total_counts r get_all_jobsrHse8 D!  -D KK,T93H3HI JM)D KK,T93D3DE F"++Y-=-=y?R?RS!1C 4F!F!'.}} #I|(LACwwx $>> C (7 G4a155#7;#F4G w 3Dd)K zFG} FU| + Hs 'E4E4)F)NN created_atrNr)rtypingrcomfy_api.internalrr frozensetr_r0r9boolr1r<r:rJrPrQr~intr[rbrirr}rArrrrrrrs  )??$$NOIJBsBtB "t8    $d $uXc]HSM5Q/R $s$48ucd$>c>>PT>ae>B?5?5%Xd^0C*D?5D ;T ;S ;c ;d4j ; sT4$8TX>>*.!%= = ==DI& = # =  == C== = 4:s?=r