/
usr
/
libexec
/
kcare
/
python
/
kcarectl
/
Upload File
HOME
� �;jc @ 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