o i @sddlZddlZddlZddlZddlZddlmZddlmZddl m Z ddl m Z ddl mZddlmZdd lmZed Zed Zd efd dZdeejd efddZdeejd eeeffddZd efddZ   d'dddedeejdBdedBdedBde dBdeeejee e dBgdfdBf ddZ!ded efdd Z"d!eeBd e fd"d#Z#d$ed e fd%d&Z$dS)(N)Callable)BytesIO)URL)argsprocessing_interrupted)IO)ProcessingInterruptedz%[0-9A-Fa-f]{2}z%(?![0-9A-Fa-f]{2})returncCstS)z3Return True if user/runtime requested interruption.rr r ;/mnt/c/Users/fbmor/ComfyUI/comfy_api_nodes/util/_helpers.pyis_processing_interruptedsrnode_clscCs|jjSN)hidden unique_idrr r r get_node_idsrcCs2|jjr dd|jjiS|jjrd|jjiSiS)N AuthorizationzBearer z X-API-KEY)rauth_token_comfy_orgapi_key_comfy_orgrr r r get_auth_headers  rcCs ttddS)Ncomfy_api_basezhttps://api.comfy.org)getattrrr r r r default_base_url&s r)display_callbacksecondslabelstart_tsestimated_totalrcst|} trtdt}|dur;|r;|r;tt|||t|||Wdn1s6wY||krAdSt t d||IdHq)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)rrrrr rendnowr r r sleep_with_interrupt*s  r, mime_typecCs|ddS)z)Converts a MIME type to a file extension./)splitlower)r-r r r mimetype_to_extensionEsr2path_or_objectcCs"t|tr tj|St|Sr) isinstancestrospathgetsizelengetvalue)r3r r r get_fs_object_sizeJs   r;urlcCsTtdd|Dstdd|Drt|St|r&t|s&t|ddSt|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.css|]}|VqdSr)isspace.0cr r r Usz!to_aiohttp_url..css|] }t|dkVqdS) N)ordr>r r r rAUsT)encoded)anyr _HAS_PCT_ESCsearch _HAS_BAD_PCT)r<r r r to_aiohttp_urlPs $ rI)NNN)%r'r#r6rer!collections.abcrioryarlrcomfy.cli_argsrcomfy.model_managementrcomfy_api.latestrcommon_exceptionsr compilerFrHboolrtype ComfyNoder5rdictrrfloatr&r,r2r;rIr r r r sN