File: //usr/libexec/kcare/python/kcarectl/http_utils.pyo
�
%��ic @ s( d d l Z d d l Z d d l Z d d l Z d d l m Z e r_ d d l m Z m Z n d d l m
Z
m Z m Z m
Z
m Z d d l m Z m Z m Z m Z m Z m Z d � Z e d � Z d � Z d
� Z d d d � Z d � Z d
� Z e � Z e j d e � d d � � Z! d S( i����N( t SSLError( t Optionalt Unioni ( t configt constantst errorst log_utilst utils( t HTTPErrort Requestt URLErrort httplibt std_urlopent urlparsec O s t | d � r | j � } n | } t | � } | j d i � } | j i t j d 6t j d 6� x* | j � D] \ } } | j | | � qp Wt
j d j | | � � y� d | k r� t
j | d <n t
j r t t d d � r t j � } t | _ t j | _ | | d <t | | | � St | | | � SWn� t k
r�} | j d k r~t j | j | j | j | j | j � � n � n� t! k
r} | j" r�t | j" d
d � r�| j" d
j# t# j$ k r�t j | d t% | � d d � � n d j | | � | _& | | _ � n Xd S(
Nt get_full_urlt headerss
KC-Versions KC-Patch-Versions# Requesting url: `{0}`. Headers: {1}t timeoutt HAS_SNIt contexti� i t errnos Request for `{0}` failed: {1}(' t hasattrR R t popt updateR t VERSIONt KC_PATCH_VERSIONt itemst
add_headerR t logdebugt formatR t HTTP_TIMEOUTt CHECK_SSL_CERTSt getattrt sslt Nonet create_default_contextt Falset check_hostnamet CERT_NONEt verify_modeR R t codeR t NotFoundt urlt msgt hdrst fpR
t argsR t ENOENTt strt reason( R) R- t kwargst request_urlR t headert valuet ctxt ex( ( s0 /usr/libexec/kcare/python/kcarectl/http_utils.pyt urlopen_base s@
-8$ c s � f d � } | S( Nc s� t | t � r" � o! | j d k St | t t j t t j f � rG t St
| d � r� t | j � d k r� | j d t
j k r� t Sd S( Ni� R- i i ( t
isinstanceR R' R
R t
HTTPExceptionR t socketR t TrueR t lenR- R t
ECONNRESET( t et state( t retry_on_500( s0 /usr/libexec/kcare/python/kcarectl/http_utils.pyt check_functionC s !:( ( R@ RA ( ( R@ s0 /usr/libexec/kcare/python/kcarectl/http_utils.pyt check_urlopen_retry_factoryB s c C s+ t | d � r | j � } n | j d � S( NR s file:( R R t
startswith( R) ( ( s0 /usr/libexec/kcare/python/kcarectl/http_utils.pyt is_local_urlO s c O sq | j d t � } | j d t j � } t | � rC t | | | � St j t d | � d | �t � | | | � S( NR@ t retry_countt count( R R; R t RETRY_COUNTRD R7 R t retryRB ( R) R- R1 R@ RE ( ( s0 /usr/libexec/kcare/python/kcarectl/http_utils.pyt urlopenU s
c C sh t | d | �} t j r> | r> | j d d j | � � n t j rd | rd | j t j | � n | S( Nt methodt
Authorizations Basic {0}( R R t UPDATE_FROM_LOCALR R R t AUTH_TOKEN_HEADER( R) t auth_stringt
auth_tokenRJ t request( ( s0 /usr/libexec/kcare/python/kcarectl/http_utils.pyt http_request] s c C sT | d k r( t j d � p' t j d � S| d k rP t j d � pO t j d � Sd S( Nt httpt
http_proxyt
HTTP_PROXYt httpst https_proxyt HTTPS_PROXY( t ost getenv( t scheme( ( s0 /usr/libexec/kcare/python/kcarectl/http_utils.pyt get_proxy_from_envh s c C s" t t d � � p! t t d � � S( NRR RU ( t boolR[ ( ( ( s0 /usr/libexec/kcare/python/kcarectl/http_utils.pyt
proxy_is_usedo s t check_retryc C st t j j | � } | s0 t d j | � � � n t t j | � � } | j } | j } | j pc d } | d k r� t d � � n | d k r� | j d k r� d n d } n | j d k r� t j
} n' | j d k r� t j } n t d � � | | | d t j �} i }
| r'd
j | � |
d <n d |
d
<t | � |
d <t j |
d <z| j d | � x* |
j � D] \ } } | j | | � qnW| j � t | d � � }
| j |
� Wd QX| j � } | j d k ray | j � } Wn t k
rd } n Xd j | j � } | d k r6| d j | � 7} n t j | � t | | j | d d � � n Wd | j! � Xd S( s� Upload 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
s" Refusing to upload empty file: {0}t /s Invalid URL: missing hostnameRU i� iP RR s Invalid URL: unsupported schemeR s Basic {0}RK s application/octet-streams Content-Types Content-Lengths
KC-Versiont PUTt rbNi� s Failed to upload file: HTTP {0}s - {0}(" RX t patht getsizet
ValueErrorR R
R t nstrt hostnamet portR! RZ R t HTTPConnectiont HTTPSConnectionR t HTTP_UPLOAD_TIMEOUTR/ R R t
putrequestR t putheadert
endheaderst opent sendt getresponset statust readt ExceptionR t logerrorR t close( t file_patht
upload_urlRN t file_sizet parsedt hostRg t url_patht conn_clst connR R3 R4 t ft responset
error_bodyt error_msg( ( s0 /usr/libexec/kcare/python/kcarectl/http_utils.pyt upload_filev sT
"(" R RX R: R R R# t typingR R t R R R R R t py23R R R
R R R
R7 R; RB RD RI R! RQ R[ R] t check_urlopen_retryRH R� ( ( ( s0 /usr/libexec/kcare/python/kcarectl/http_utils.pyt <module> s$ (. 0