+j ddlZddlZddlmZddlmZddlmZmZm Z m Z ddl Z ddl m Z n#e $re jdYnwxYwdZe gefZeGdd Zed eeezd e eeeeffd Zeddfd eeezd ededzdedzd eedzedzff dZdS)N)contextmanager) dataclass)IOAnyCallableIterator)blake3z%WARNING: blake3 package not installedicDeZdZUdZeed<eed<dZeed<dZeed<dS)HashCheckpointz9Saved state for resuming an interrupted hash computation.bytes_processedhasherrmtime_ns file_sizeN) __name__ __module__ __qualname____doc__int__annotations__rrr;/home/wildlama/comfy/ComfyUI/app/assets/services/hashing.pyr r sMCC KKKHcIsrr fpreturnc#Kt|drt|dd}d}|rH |}|dkr|dn#tj$rd}YnwxYw |dfV|||dSdS#|||wwxYwt tj|d5}|dfVddddS#1swxYwYdS) z=Yield (file_object, is_path) with appropriate setup/teardown.readseekablecdS)NFrrrrz#_open_for_hashing.. s5rNrFrbT) hasattrgetattrtellseekioUnsupportedOperationopenosfspath)rrorig_posfs r_open_for_hashingr,swr6972z==99;;   7799q==GGAJJJ*     "e)OOO#!!!!!$#x#!!!!$")B-- & & !T'MMM                  s//A""A65A6:BB5C..C25C2 chunk_sizeinterrupt_check checkpointc|dkrt}t|5\}}|+|r)||j|j}|j}nt }d} |9|r/|rdt ||fcdddS ddddS||}|sn(|||t|z }{| dfcdddS#1swxYwYdS)aCompute BLAKE3 hash of a file, with optional checkpoint support. Args: fp: File path or file-like object chunk_size: Size of chunks to read at a time interrupt_check: Optional callable that returns True if the operation should be interrupted (e.g. paused or cancelled). Must be non-blocking so file handles are released immediately. Checked between chunk reads. checkpoint: Optional checkpoint to resume from (file paths only) Returns: Tuple of (hex_digest, None) on completion, or (None, checkpoint) on interruption (file paths only), or (None, None) on interruption of a file object rNT)r r )NN) DEFAULT_CHUNKr,r$r r r r rupdatelen hexdigest) rr-r.r/r+is_pathhr chunks rcompute_blake3_hashr83s,Q" 2  #,1g  !g ! FF:- . . .!A(8OOAO **/@/@*(7 """######## "!########"FF:&&E  HHUOOO s5zz )O *{{}}d"/##################sA"C< C<AC<<DD)r%r( contextlibr dataclassesrtypingrrrrloggingr ModuleNotFoundErrorwarningr1boolInterruptCheckr strbytestupler,rr8rrrrDs %%%%%%!!!!!!............====GO;<<<<<= "d(#  #5 /huRY_7M.N0$-1(, 0#0# bi0#0#$d*0#% 0#  3:~, ,- 0#0#0#0#0#0#s-AA