+j ddlZddlZddlZddlZddlZddlmZddlmZddl m Z ddl m Z ddl mZddlmZdd lmZejd Zejd Zd efd Zdeejd efdZdeejd eeeffdZd efdZ ddddedeejdzdedzdedzde dzdeeejee e dzgdfdzf dZ!ded efdZ"deezd e fdZ#ded e fdZ$dS) N)Callable)BytesIO)URL)argsprocessing_interrupted)IO)ProcessingInterruptedz%[0-9A-Fa-f]{2}z%(?![0-9A-Fa-f]{2})returnctS)z3Return True if user/runtime requested interruption.r=/home/wildlama/comfy/ComfyUI/comfy_api_nodes/util/_helpers.pyis_processing_interruptedrs ! # ##rnode_clsc|jjSN)hidden unique_idrs r get_node_idrs ? $$rct|jjrdd|jjiS|jjrd|jjiSiS)N AuthorizationzBearer z X-API-KEY)rauth_token_comfy_orgapi_key_comfy_orgrs rget_auth_headerrsL+S!Q8?+O!Q!QRR(@X_>?? Irc.ttddS)Ncomfy_api_basezhttps://api.comfy.org)getattrrrrrdefault_base_urlr!&s 4)+B C CCr)display_callbacksecondslabelstart_tsestimated_totalr"c Ktj|z} trtdtj}|S|rQ|rOt jt 5|||t||z |dddn #1swxYwY||krdStj td||z d{V)z Sleep in 1s slices while: - Checking for interruption (raises ProcessingInterrupted). - Optionally emitting time progress via display_callback (if provided). TzTask cancelledNg?) time monotonicrr contextlibsuppress Exceptionintasynciosleepmin)r#rr$r%r&r"endnows rsleep_with_interruptr3*s7 .  W $C 1 $ & & :'(899 9n  E .> $Y// X X  5#cHn2E2EWWW X X X X X X X X X X X X X X X #:: EmCS3Y//000000000 1s)BBB mime_typec\|ddS)z)Converts a MIME type to a file extension./)splitlower)r4s rmimetype_to_extensionr:Es% ??3   # ) ) + ++rpath_or_objectct|trtj|St |Sr) isinstancestrospathgetsizelengetvalue)r;s rget_fs_object_sizerDJsB.#&&/w~... ~&&(( ) ))rurlc,td|Dstd|Drt|St|r+t|st|dSt|S)aRIf `url` appears to be already percent-encoded (contains at least one valid %HH escape and no malformed '%' sequences) and contains no raw whitespace/control characters preserve the original encoding byte-for-byte (important for signed/presigned URLs). Otherwise, return `URL(url)` and allow yarl to normalize/quote as needed.c3>K|]}|VdSr)isspace.0cs r z!to_aiohttp_url..Us* $ $1199;; $ $ $ $ $ $rc3<K|]}t|dkVdS) N)ordrIs rrLz!to_aiohttp_url..Us,+G+GaCFFTM+G+G+G+G+G+GrT)encoded)anyr _HAS_PCT_ESCsearch _HAS_BAD_PCT)rEs rto_aiohttp_urlrUPs  $ $ $ $ $$$+G+G3+G+G+G(G(G3xx3& (;(;C(@(@&3%%%% s88Or)NNN)%r.r*r?rer(collections.abcrioryarlrcomfy.cli_argsrcomfy.model_managementrcomfy_api.latestr common_exceptionsr compilerRrTboolrtype ComfyNoder>rdictrr!floatr-r3r:rDrUrrrrds $$$$$$999999444444rz,-- rz011 $4$$$$ %$r|,%%%%%d2<0T#s(^D#DDDD!"& 1[_111 12< 4'1 :1dl 1 4Z 1R\ 2CcDjI4OPSWW11116,S,S,,,, *sW}*****        r