Google
Edit File: http_utils.pyo
� !S�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