o i5@sdZddlmZddlmZGdddZehdZehdZde d e fd d Z d d Z de d e fddZdZde d e fddZde d eeeee ffddZde de d e fddZdede d e fddZd:d e d!e d"e d e fd#d$Zde d eeee ffd%d&Zd'ee d(e d)e d ee fd*d+Zd e d,ed-ed.e d ee f d/d0Z 1 1 2 3 1 d;d,ed-ed.e d4eee d5ee d(e d)e d6eed7ed eee effd8d9Zd1S)3dtextaudiovideoimages>.fbx.glb.obj.gltf.usdzfilenamereturncs|tfddtDS)Nc3|]}|VqdSNendswith.0extlowerrr z#has_3d_extension..)r*anyTHREE_D_EXTENSIONSr rr)rhas_3d_extensionsr0cCsB|durdSt|trt|r|ddddSdSt|tr|SdS)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) isinstancestrr0dict)itemrrrnormalize_output_item"s  r:outputsc Csi}|D]J\}}t|ts|||<qi}|D]1\}}|dks't|ts,|||<qg}|D]}|dur7q0t|} || durC| n|q0|||<q|||<q|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. animatedN)itemsr6r8listr:append) r; normalizednode_id node_outputsnormalized_node media_typer=normalized_itemsr9normrrrnormalize_outputs3s&   rGivaluecCs&t|tkr d|iS|dtddS)zCreate a text preview dict with optional truncation. Returns: dict with 'content' and optionally 'truncated' flag contentNT)rI truncated)lenTEXT_PREVIEW_MAX_LENGTH)rHrrr_create_text_previewRs  rM extra_datacCs0|d}|di}|did}||fS)zqExtract create_time and workflow_id from extra_data. Returns: tuple: (create_time, workflow_id) create_time extra_pnginfoworkflowidget)rNrOrP workflow_idrrr_extract_job_metadata`s  rVrDr9csb|tvrdS|dd}|r|ds|drdS|ddtfddtDr/dSd S) 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) Tformatr2zvideo/zaudio/r c3r"r#r$r&r/rrr+r,z!is_previewable..F)PREVIEWABLE_MEDIA_TYPESrT startswithr*r-r.)rDr9fmtrr/ris_previewablels  r[statuscCs0|\}}}}}t|\}}t||||d|dS)zvConvert queue item tuple to unified job dict. Expects item with sensitive data already removed (5 elements). r)rRr\priorityrO outputs_countrU)rVr)r9r\r] prompt_id_rNrOrUrrrnormalize_queue_items raFr_ history_iteminclude_outputsc Csp|d}|\}}}}}t|\}} |di} | r| dnd} |di} t| \} }d}d}d}d}| r|| dg}|D]=}t|ttfr{t|dkr{|d |d }}t|tr{|d kre|d }q>|d vr{|d }|dkru|}q>|dkr{d}q>| dkrtj }n| dkr|rtj ntj }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). promptr\ status_strNr;Fmessagesrexecution_start timestamp)execution_successexecution_errorexecution_interruptedrlrmTsuccesserror) rRr\r]rOexecution_start_timeexecution_end_timerlr^preview_outputrUexecution_status)rdrNrQ)rVrTget_outputs_summaryr6r>tuplerKr8rrrrrrG)r_rbrc prompt_tupler]r`rdrNrOrU status_inforer;r^rrrlrprqwas_interruptedrfentry event_name event_datar\jobrrrnormalize_history_itemsd          r}c CsDd}d}d}|D]\}}t|tsq |D]\}}|dks%t|ts&q|D]q}t|tskt|} | duri|dkrh|d7}|durht|trQ|rN|dnd} nt|} t| } i| ||d} |durh| }q(| }|d7}|durtq(t||ri|d|i} d |vr|| d <| d d kr| }q(|dur| }q(qq ||p|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<rrhr2)nodeIdr5r~r5r3r1) r=r6r8r>r:rur7rMr[rT) r;countrrfallback_previewrArBrDr=r9r@ text_value text_previewenrichedrrrrtsb      /rtjobssort_by sort_ordercCs0|dk}|dkr dd}ndd}t|||dS)z,Sort jobs list by specified field and order.descexecution_durationcSs,|dd}|dd}|r|r||SdS)NrprrqrS)r|startendrrr get_sort_key"s  z#apply_sorting..get_sort_keycSs |ddS)NrOrrS)r|rrrr's )keyreverse)sorted)rrrrrrrr apply_sortings  rrunningqueuedhistorycCsj||vr t|||ddS|D]}|d|krt|tjSq|D]}|d|kr2t|tjSq"dS)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)rcrhN)r}rarrr)r_rrrr9rrrget_job-s   rN created_atr status_filterrUlimitoffsetc s g} |dur tj}tj|vr|D] } | t| tjqtj|vr/|D] } | t| tjq#tjtjtjh} | t |@} | rY| D]\} }t | |}| d| vrX| |qCrdfdd| D} t | ||} t| }|dkrx| |d} |dur| d|} | |fS)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) Nr\csg|] }|dkr|qSrUrS)r'jrrr zsz get_all_jobs..r)rrrr?rarrrrsetr=r}rTrrK)rrrrrUrrrrrr9history_statusesrequested_history_statusesr_rbr| total_countrrr get_all_jobsHs4        r)F)NNrrNr)r typingrcomfy_api.internalrr frozensetrXr.r7boolr0r:r8rGrLrMruintrVr[rar}rtr>rrrrrrrsZ   " A"B"