­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ 3 ;jE@sddlZddlZddlZddlZddlmZddlmZmZmZm Z m Z ddl m Z m Z mZmZmZmZddZdd d Zd d Zd dZdddZddZddZeZe jeddddZdS)N)SSLError)config constantserrors log_utilsutils) HTTPErrorRequestURLErrorhttplib std_urlopenurlparsec Ost|dr|j}n |}t|}|jdi}|jtjtjdx |jD]\}}|j ||qJWt j dj ||ydd|krt j|d<t j rttddrtj}d|_tj|_||d<t|f||St|f||Stk r&}z0|jd krtj|j|j|j|j|jWYdd}~Xn~tk r}z`|j r|t|j d d r||j d j!t!j"kr|tj|d t#|ddd j |||_$||_WYdd}~XnXdS) N get_full_urlheaders)z KC-VersionzKC-Patch-Versionz#Requesting url: `{0}`. Headers: {1}timeoutHAS_SNIFcontextirerrnozRequest for `{0}` failed: {1})%hasattrrr popupdaterVERSIONKC_PATCH_VERSIONitems add_headerrlogdebugformatr HTTP_TIMEOUTCHECK_SSL_CERTSgetattrsslcreate_default_contextcheck_hostname CERT_NONE verify_moder r coderNotFoundurlmsghdrsfpr argsrENOENTstrreason) r(r,kwargsZ request_urlrheadervaluectxexr50/usr/libexec/kcare/python/kcarectl/http_utils.py urlopen_bases>      .r7Tcsfdd}|S)Ncsbt|tro|jdkSt|ttjttjfr2dSt |dr^t |j dkr^|j dt j kr^dSdS)NiTr,r) isinstancer r&r r HTTPExceptionrsocketrrlenr,rZ ECONNRESET)estate) retry_on_500r5r6check_functionCs  (z3check_urlopen_retry_factory..check_functionr5)r?r@r5)r?r6check_urlopen_retry_factoryBs rAcCst|dr|j}|jdS)Nrzfile:)rr startswith)r(r5r5r6 is_local_urlOs rCcOsV|jdd}|jdtj}t|r2t|f||Stjt|d|dt|f||S)Nr?T retry_count)r?)count)rr RETRY_COUNTrCr7rretryrA)r(r,r0r?rDr5r5r6urlopenUs  rHcCsHt||d}tj r*|r*|jddj|tj rD|rD|jtj||S)N)method Authorizationz Basic {0})r rUPDATE_FROM_LOCALrrrAUTH_TOKEN_HEADER)r( auth_string auth_tokenrIrequestr5r5r6 http_request]s    rPcCs<|dkrtjdptjdS|dkr8tjdp6tjdSdS)NhttpZ http_proxyZ HTTP_PROXYhttpsZ https_proxyZ HTTPS_PROXY)osgetenv)schemer5r5r6get_proxy_from_envhsrVcCsttdpttdS)NrQrR)boolrVr5r5r5r6 proxy_is_usedosrX)Z check_retrycCstjj|}|stdj|ttj|}|j}|j }|jp@d}|dkrRtd|dkrl|j dkrhdnd}|j dkr~t j }n|j dkrt j }ntd |||tjd } i} |rd j|| d <d | d<t|| d<tj| d<z| jd|x | jD]\} } | j| | qW| jt|d} | j| WdQRX| j}|jdkry |j}Wntk rnd}YnXdj|j}|dk r|dj|7}tj|t||j|ddWd| j XdS)aUpload a file to the given URL using HTTP PUT with chunked streaming. Note: The standard library urllib doesn't support PUT with data We need to use httplib directly for this This function uses streaming upload to support large files up to 1GB without loading the entire file into memory. :param file_path: Path to the file to upload :param upload_url: Full URL to upload the file to. Query params are ignored. :param auth_string: Optional authentication string for Basic Auth :return: None if upload succeeded :raises HTTPError: If upload fails with HTTP status >= 400 :raises ValueError: If URL is invalid z"Refusing to upload empty file: {0}/NzInvalid URL: missing hostnamerRiPrQzInvalid URL: unsupported scheme)rz Basic {0}rJzapplication/octet-streamz Content-TypezContent-Lengthz KC-VersionPUTrbizFailed to upload file: HTTP {0}z - {0})!rSpathgetsize ValueErrorrrrnstrhostnameportrUr HTTPConnectionHTTPSConnectionrHTTP_UPLOAD_TIMEOUTr.rr putrequestr putheader endheadersopensend getresponsestatusread Exceptionrlogerrorr close)Z file_path upload_urlrMZ file_sizeparsedhostrbZurl_pathZconn_clsconnrr1r2fresponseZ error_body error_msgr5r5r6 upload_filevsT              rx)T)NN)N)rrSr;r!rrrrrrpy23r r r r r rr7rArCrHrPrVrXcheck_urlopen_retryrGrxr5r5r5r6s   0