o it,@s(ddlZddlZddlZddlmZddlmZddlmZddl m Z m Z ddl Z ddl Z ddlmZmZddlmZddlmZmZddlmZd d lmZd d lmZmZmZmZmZd d lm Z d d l!m"Z"m#Z#m$Z$d dl%m&Z&hdZ'ddddddde(deee)Be(BeBdBde*dBde+de*de*de,ej-ddfddZ.dddde(de*de,ej-de j/fdd Z0dddd!d"e(de*de+de,ej-dej1f d#d$Z2dddde(de*de,ej-defd%d&Z3d'e(de(d(e+de(fd)d*Z4ddddd+de(d,e(d-e(dBde*dBde+de,ej-dej5fd.d/Z6dS)0N)BytesIO)Path)IO)urljoinurlparse) ClientErrorContentTypeError) InputImplTypes)get_output_directory)request_logger)default_base_urlget_auth_headeris_processing_interruptedsleep_with_interruptto_aiohttp_url)_diagnose_connectivity)ApiServerErrorLocalNetworkErrorProcessingInterrupted)bytesio_to_image_tensor>?g@)timeout max_retries retry_delay retry_backoffclsurldestr r!r"r#r$returncs: t|ttfst|dstdd}|}i} t|} | js;| js;|dur)tdtt dd| d}t |} |d7}t d ||} tj|d } t|ttf} d}d}dd}d}zztttj| d |d Wdn1sywYtj| d }tfd d}t|}t|jt|| d}tj||htjdIdH\}}||vr||vr|tt|IdHWdtd1swYtdz|IdH}Wn tjytddw|4IdHz|j dkrttJz |!IdH}Wn$t"tfy2|#IdH}t$|dkr(|ndt$|d}Ynwtj| d ||j t%|j&|d|j dWdn 1sRwY|j t'vr||krt(||dddIdH||9} WdIdHWWdur)|r|tt |IdHWdn 1swY|r|*s|tt |IdHWdn 1swY|rtt|+IdHWdn 1swY|rtt|,|+Wdn 1s wYq;td|j d| rItt|}tt|j-j.dddWdn 1sd&|t?d'|d}~wwWdur})|r|tt |IdHWdn 1swY|r|*s|tt |IdHWdn 1swY|rtt|+IdHWdn 1swY|rtt|,|+Wdn 1swYndur)|r3|tt |IdHWdn 1s.wY|rZ|*sZ|tt |IdHWdn 1sUwY|rztt|+IdHWdn 1suwY|rtt|,|+Wdw1swYwq<)(aStream-download a URL to `dest`. `dest` must be one of: - a BytesIO (rewound to 0 after write), - a file-like object opened in binary write mode (must implement .write()), - a filesystem path (str | pathlib.Path), which will be opened with 'wb'. If `url` starts with `/proxy/`, `cls` must be provided so the URL can be expanded to an absolute URL and authentication headers can be applied. Raises: ProcessingInterrupted, LocalNetworkError, ApiServerError, Exception (HTTP and other errors) writezNdest must be a path (str|Path) or a binary-writable object providing .write().rNz._monitor)headers) return_whenzTask cancellediiz[text z bytes]zHTTP )r,r-r.response_status_coderesponse_headersresponse_content error_messagezFailed to download (HTTP z).)parentsexist_okwbirz [streamed z bytes to dest])r,r-r.r:r;r<z: z (will retry))r,r-r.r=internet_accessiblezVUnable to connect to the network. Please check your internet connection and try again.z4The remote service appears unreachable at this time.)@ isinstancestrrhasattr ValueErrorrschemenetlocrrrstriplstripr_generate_operation_idaiohttp ClientTimeout contextlibsuppress Exceptionr log_request_response ClientSessionr0Event create_taskgetrwaitFIRST_COMPLETEDcancelrr2statusjsonrtextlendictr8 _RETRY_STATUSrsetdonecloseflushparentmkdiropenwait_forcontentread TimeoutErrorrat_eofr(rseekrOSErrortype__name__rrr)r%r&r r!r"r#r$attemptdelayr8 parsed_urlop_id timeout_cfg is_path_sinkfhandlesession monitor_taskreq_taskr7r`pendingrespbodyr[psinkwrittenchunkediagr3r4r6download_url_to_bytesio s             &  W                    W       $B              W                rr r$cs&t}t||||dIdHt|S)z@Downloads an image from a URL and returns a [B, H, W, C] tensor.rN)rrrr%r r$resultr3r3r6download_url_to_image_tensorsrr r!r$ video_urlcs*t}t|||||dIdHt|S)z:Downloads a video from a URL and returns a `VIDEO` output.rN)rrr VideoFromFile)rr r!r$rr3r3r6download_url_to_video_outputs rcs"t}t||||dIdH|S)zFDownloads content from a URL and returns a new BytesIO (rewound to 0).rN)rrrr3r3r6download_url_as_bytesiosrmethodroc Csvzt|}|jdddp|jpdddd}Wn ty&d}Ynw|d|d|dtj ddS)Nr)r download__try) rpathrsplitrHstripreplacerPuuiduuid4hex)rr%roparsedslugr3r3r6rKs. (rK)task_idr r!r$ file_formatrc sx|d}t}t|||||dIdH|dur5tt}||d|}|||dt j ||dS)zDownloads a 3D model file from a URL into memory as BytesIO. If task_id is provided, also writes the file to disk in the output directory for backward compatibility with the old save-to-disk behavior. .rNr)sourcer) rJlowerrrrr write_bytesgetvaluerkr File3D) r%rrr r!r$data output_dir output_pathr3r3r6download_url_to_file_3d s    r)7r0rNriorpathlibrtypingr urllib.parserrrLtorchaiohttp.client_exceptionsrrcomfy_api.latestCOMFY_IOr r folder_pathsr r _helpersrrrrrclientrcommon_exceptionsrrr conversionsrr^rDbytesfloatintrm ComfyNoderTensorrrrrrKrrr3r3r3r6s         ?