+j=ddlZddlZddlZddlmZddlmZddlmZm Z m Z m Z m Z m Z mZddlmZddlmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%ddl&m'Z'ddl(m)Z)dd l*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2dd l3m4Z4 d3d e5d e5de,dzfdZ6 d4d e5de5dzdee5dzde0de5d e5de5dzde5dzde,fdZ7 d5d e5d e5de8de8fdZ9 d6d e5de5dzd e5de,fdZ:de5de8fdZ;de5de+dzfdZd Z< d7d e5d$ee5dzd%ee5dzd&e5dzd'e=dzd(e>d)e>d*e5d+e5d,e5dzde/fd-Z?d.ede@fd/ZAd*e5d+e5de5dzfd0ZB d3de5d e5de.dzfd1ZC d3d e5d e5de.fd2ZDdS)8N)timezone)Sequence) CursorPayloadInvalidCursorError decode_cursordecode_cursor_intdecode_cursor_time encode_cursorencode_cursor_from_time)Asset)asset_exists_by_hashreference_exists_for_asset_iddelete_reference_by_idfetch_reference_and_assetsoft_delete_reference_by_idfetch_reference_asset_and_tagsget_asset_by_hashget_reference_by_idget_reference_with_owner_checklist_references_pagelist_all_file_paths_by_asset_idlist_references_by_asset_idset_reference_metadataset_reference_previewset_reference_tagsupdate_asset_hash_and_mimeupdate_reference_access_timeupdate_reference_nameupdate_reference_updated_at)select_best_live_path)compute_relative_filename) AssetDataAssetDetailResultAssetSummaryDataDownloadResolutionResultListAssetsResult UserMetadataextract_asset_dataextract_reference_data)create_session reference_idowner_idreturnct5}t|||}|s ddddS|\}}}tt|t ||cdddS#1swxYwYdS)Nr,r-refassettags)r*rr#r)r()r,r-sessionresultr2r3r4s D/home/wildlama/comfy/ComfyUI/app/assets/services/asset_management.pyget_asset_detailr87s    W/ %             "UD &s++$U++                     sA01A00A47A4manualnamer4 user_metadata tag_origin mime_type preview_idcPt5}t|||} d} ||| jkrt|||d} | jrt | jnd} d} |t |} n3| r1| jpi} | d| krt | } | | r| | d<t||| d} |t||||d} |t|| j |}|rd} |t|||d} | r|t|| t||| }|st!d |\} }}t#t%| t'|| }||cdddS#1swxYwYdS) NF)r,r:Tfilename)r,r;)r,r4origin)asset_idr=r,preview_reference_idr,r0zState changed during updater1)r*rr:r file_pathr!dictr;getrrrrBrrr RuntimeErrorr#r)r(commit)r,r:r4r;r<r-r=r>r5r2touchedcomputed_filenamenew_meta current_metaupdatedr6r3tag_listdetails r7update_asset_metadatarRLs   EW,WlHMM   0 0 !' 4 P P P PGHK _5cmDDD[_ $  $M**HH  .,2L ++/@@@ --    9'8$ "l(    G   )!      G  0#,)G   ! !)%/     G  L}, 'l K K K K/ %     ><== =%UH"&s++$U++    KEEEEEEEEEEEEEEEEEEsE?FF"FTdelete_content_if_orphanct5}|s4t|||}||cdddSt||}|r|jnd}|r|jnd}t |||}|s"| ddddS|s"| ddddSt||}|r"| ddddSt||} |r| || t|} | | | || D]h} tjt5| r3t j| rt!j| dddn #1swxYwYi dddn #1swxYwYdS)aDelete an asset reference. With ``delete_content_if_orphan=False`` (a soft delete), the reference is hidden and the underlying content is preserved. With ``True``, the content is also removed once it becomes orphaned. Note: the public DELETE /api/assets/{id} endpoint always soft-deletes (passes ``False``); the orphan-reclamation path is intentionally internal-only, retained for a future GC/admin caller. r0NrEFTrB)r*rrJrrBrFrrrappendrHr delete contextlibsuppress Exceptionospathisfileremove) r,r-rSr5deletedref_rowrBrF still_exists file_paths asset_rowps r7delete_asset_referenceres   .!W' 1lXG NN   .!.!.!.!.!.!.!.!&gLIII'.87##D)0:G%%d ( ,     NN   %.!.!.!.!.!.!.!.!(  NN   -.!.!.!.!.!.!.!.!05WxPPP   NN   7.!.!.!.!.!.!.!.!>5WxPPP  )   i ( ( (KKx00  NN9 % % % ! !A$Y// ! !!**!IaLLL ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! !W.!.!.!.!.!.!.!.!.!.!.!.!.!.!.!` 4sO*G"AG"#G"(G"%>eTTTT"l22CNNE }t "4hl)K)KSV[` a a aarct5}t||}|s ddddSt||j}fd|D}t |}|s ddddSt j|}|D]}|j|kr|j r |j }n|j ptj |dpd} dddn #1swxYwYt|| |S)a$Resolve a blake3 hash to an on-disk file path. Only references visible to *owner_id* are considered (owner-less references are always visible). Returns a DownloadResolutionResult with abs_path, content_type, and download_name, or None if no asset or live path is found. NrUc@g|]}|jdks |jk|Sr+)r-).0rr-s r7 z(resolve_hash_to_path..s<   zR1:#9#9 #9#9#9rrapplication/octet-streamabs_path content_type download_name)r*rmrrr r[r\basenamerFr:r= mimetypes guess_typer%) rhr-r5r3rvisibler display_namer2ctypes ` r7resolve_hash_to_pathrs    W)':>>         +7UXFFF       )11         w''11   C}((SX("x  O *#L11!4 *) #               , $"   sC"6C"4A"C""C&)C&c t5}t|||}|std|d|\}}|jr,tj|jr|j}nJt||j}t|}|s#td|d|jd|j d|j }|j } t|| || ptj|p|d pd } |ptj|} t%|| | cdddS#1swxYwYdS) Nr0zAssetReference z not foundrUz No live path for AssetReference z (asset id=z, name=)rErrr)r*r ValueErrorrFr[r\r]rrr FileNotFoundErrorr:r=rrJrrrr%) r,r-r5pairr2r3rrref_name asset_mimerrs r7resolve_asset_for_downloadrs   ' W( ,    IG|GGGHH H U = RW^^CM:: }HH/wJJJD,T22H '>|>>!&>>25(>>> 8_ $W<HHHH  *#H$899!< *)  !>BG$4$4X$>$> ''   G' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' sD/E  EEr)NNNr9r+NN)T)Nr+) r+NNNNrqrrnrrN)ErXrr[datetimertypingrapp.assets.services.cursorrrrrr r r app.assets.database.modelsr app.assets.database.queriesr rrrrrrrmrrrrrrrrrrrrapp.assets.helpersr app.assets.services.path_utilsr!app.assets.services.schemasr"r#r$r%r&r'r(r)app.database.dbr*rr8rRboolrergrkrrGintrobjectrrrrrr7rs -,,,,,*544444DDDDDD                    +*****         .!%"& !OOO *O 3-$ O O  O  OTzOd OOOOOj&*????#? ????H(,* >DSDTDDDD )#))d*:))))C)-)- $#'MSMSMS3-$&MS3-$&MS: MS D[ MS  MS MS MS MS :MSMSMSMSMS`=Vb3bsbsTzbbbb.&&&&$&&&&V+ + + + + + + + + + r