File: //usr/libexec/kcare/python/kcarectl/http_utils.pyc
�
�� 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� t j
r� 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 formatt PY2_6R 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_functionD s !:( ( RA RB ( ( RA s0 /usr/libexec/kcare/python/kcarectl/http_utils.pyt check_urlopen_retry_factoryC 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_urlP s c O sq | j d t � } | j d t j � } t | � rC t | | | � St j t d | � d | �t � | | | � S( NRA t retry_countt count( R R< R t RETRY_COUNTRE R8 R t retryRC ( R* R. R2 RA RF ( ( s0 /usr/libexec/kcare/python/kcarectl/http_utils.pyt urlopenV 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_tokenRK 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_envi s c C s" t t d � � p! t t d � � S( NRS RV ( t boolR\ ( ( ( s0 /usr/libexec/kcare/python/kcarectl/http_utils.pyt
proxy_is_usedp s t check_retryc C s� 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 � � t j r� i n
i t j d 6} | | | | � }
i } | r@d
j | � | d <n d | d
<t | � | d <t j | d <z|
j d | � x* | j � D] \ } }
|
j | |
� q�W|
j � t | d � � } |
j | � Wd QX|
j � } | j d k rzy | j � } Wn t k
rd } n Xd j | j � } | d k rO| 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 hostnameRV i� iP RS s Invalid URL: unsupported schemeR s Basic {0}RL s application/octet-streams Content-Types Content-Lengths
KC-Versiont PUTt rbNi� s Failed to upload file: HTTP {0}s - {0}(# RY t patht getsizet
ValueErrorR R
R t nstrt hostnamet portR" R[ R t HTTPConnectiont HTTPSConnectionR R R t HTTP_UPLOAD_TIMEOUTR0 R t
putrequestR t putheadert
endheaderst opent sendt getresponset statust readt ExceptionR t logerrorR t close( t file_patht
upload_urlRO t file_sizet parsedt hostRh t url_patht conn_clst conn_kwargst connR R4 R5 t ft responset
error_bodyt error_msg( ( s0 /usr/libexec/kcare/python/kcarectl/http_utils.pyt upload_filew sV
"(" R RY R; R! R R$ t typingR R t R R R R R t py23R R R
R R R
R8 R< RC RE RJ R" RR R\ R^ t check_urlopen_retryRI R� ( ( ( s0 /usr/libexec/kcare/python/kcarectl/http_utils.pyt <module> s$ (. 1