| Dir : /usr/libexec/kcare/python/kcarectl/__pycache__/ |
| Server: Linux red.truehostdns.com 4.18.0-553.150.1.el8_10.x86_64 #1 SMP Fri Jul 31 15:23:31 EDT 2026 x86_64 IP: 51.161.15.103 |
| Dir : //usr/libexec/kcare/python/kcarectl/__pycache__/http_utils.cpython-36.pyc |
3
0�jE � @ s� d dl Z d dlZd dlZd dlZd dlmZ ddlmZmZmZm Z m
Z
ddl m Z m
Z
mZmZmZmZ dd� Zdd d
�Zd d � Zd
d� Zddd�Zdd� Zdd� Ze� Ze
jed�ddd��ZdS )� N)�SSLError� )�config� constants�errors� log_utils�utils)� HTTPError�Request�URLError�httplib� std_urlopen�urlparsec O s� t | d�r| j� }n | }t| �} |jdi �}|jtjtjd�� x |j� D ]\}}| j ||� qJW t
j dj ||�� ydd|kr�t
j|d< t
j r�ttdd �r�tj� }d|_tj|_||d< t| f|�|�S t| f|�|�S tk
�r& } z0|jd k�rtj|j|j|j |j |j ��� W Y d d }~X n~ t k
�r� } z`|j �r|t |j d
d ��r||j d
j!t!j"k�r|tj| d t#|�d d ��d j ||�|_$||_� W Y d d }~X nX d S )
N� get_full_url�headers)z
KC-VersionzKC-Patch-Versionz#Requesting url: `{0}`. Headers: {1}�timeout�HAS_SNIF�contexti� r �errnoz Request for `{0}` failed: {1})%�hasattrr r
�pop�updater �VERSION�KC_PATCH_VERSION�items�
add_headerr �logdebug�formatr � HTTP_TIMEOUT�CHECK_SSL_CERTS�getattr�ssl�create_default_context�check_hostname� CERT_NONE� verify_moder
r �coder �NotFound�url�msg�hdrs�fpr �argsr �ENOENT�str�reason) r( r, �kwargsZ request_urlr �header�value�ctx�ex� r5 �0/usr/libexec/kcare/python/kcarectl/http_utils.py� urlopen_base s>
.r7 Tc s � fdd�}|S )Nc sb t | t�r� o| jdkS t | ttjttjf�r2dS t | d�r^t
| j �dkr^| j d t j
kr^dS d S )Ni� Tr, � r )�
isinstancer r& r r �
HTTPExceptionr �socketr r �lenr, r Z
ECONNRESET)�e�state)� retry_on_500r5 r6 �check_functionC s
(z3check_urlopen_retry_factory.<locals>.check_functionr5 )r? r@ r5 )r? r6 �check_urlopen_retry_factoryB s rA c C s t | d�r| j� } | jd�S )Nr zfile:)r r �
startswith)r( r5 r5 r6 � is_local_urlO s
rC c O sV |j dd�}|j dtj�}t| �r2t| f|�|�S tjt|d�|d�t�| f|�|�S )Nr? T� retry_count)r? )�count)r r � RETRY_COUNTrC r7 r �retryrA )r( r, r0 r? rD r5 r5 r6 �urlopenU s
rH c C sH t | |d�}tj r*|r*|jddj|�� tj rD|rD|jtj|� |S )N)�method�
Authorizationz Basic {0})r
r �UPDATE_FROM_LOCALr r r �AUTH_TOKEN_HEADER)r( � auth_string�
auth_tokenrI �requestr5 r5 r6 � http_request] s rP c C s< | dkr t jd�pt jd�S | dkr8t jd�p6t jd�S d S )N�httpZ
http_proxyZ
HTTP_PROXY�httpsZ https_proxyZ HTTPS_PROXY)�os�getenv)�schemer5 r5 r6 �get_proxy_from_envh s rV c C s t td��pt td��S )NrQ rR )�boolrV r5 r5 r5 r6 �
proxy_is_usedo s rX )Z check_retryc C s� t jj| �}|s tdj| ���ttj|��}|j}|j }|jp@d}|dkrRtd��|dkrl|j
dkrhdnd}|j
dkr~t j }n|j
dkr�t j
}ntd ��|||tjd
�} i }
|r�d j|�|
d < d
|
d< t|�|
d< tj|
d< z�| jd|� x |
j� D ]\} } | j| | � q�W | j� t| d��}
| j|
� W dQ R X | j� }|jdk�r�y |j� }W n t k
�rn d}Y nX dj|j�}|dk �r�|dj|�7 }t j |� t ||j|dd��W d| j � X dS )a� 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
z"Refusing to upload empty file: {0}�/Nz Invalid URL: missing hostnamerR i� �P rQ z Invalid URL: unsupported scheme)r z Basic {0}rJ zapplication/octet-streamz Content-TypezContent-Lengthz
KC-Version�PUT�rbi� z Failed to upload file: HTTP {0}z - {0})!rS �path�getsize�
ValueErrorr r r �nstr�hostname�portrU r �HTTPConnection�HTTPSConnectionr �HTTP_UPLOAD_TIMEOUTr. r r �
putrequestr � putheader�
endheaders�open�send� getresponse�status�read� Exceptionr �logerrorr �close)Z file_path�
upload_urlrM Z file_size�parsed�hostrb Zurl_pathZconn_cls�connr r1 r2 �f�responseZ
error_body� error_msgr5 r5 r6 � upload_filev sT
rx )T)NN)N) r rS r; r! r � r r r r r �py23r r
r r r
r r7 rA rC rH rP rV rX �check_urlopen_retryrG rx r5 r5 r5 r6 �<module> s 0