File: //usr/local/lib/python3.10/site-packages/pip/_internal/network/__pycache__/download.cpython-310.pyc
o
��i� � @ sj d Z ddlZddlZddlZddlZddlmZmZm Z ddl
m Z m Z ddl
mZ ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZ dd lmZmZm Z dd l m Z m Z m Z e�!e"�Z#d
e dee$ fdd�Z%d
e dede&dee' fdd�Z(de&de&fdd�Z)de&de&de&fdd�Z*d
e dede&fd d �Z+d edede fd d �Z,G d!d"� d"�Z-G d#d$� d$�Z.dS )%z)Download files with progress indicators.
� N)�Iterable�Optional�Tuple)�CONTENT_CHUNK_SIZE�Response)� get_download_progress_renderer)�NetworkConnectionError)�PyPI)�Link)�
is_from_cache)�
PipSession)�HEADERS�raise_for_status�response_chunks)� format_size�redact_auth_from_url�splitext�resp�returnc
C s, zt | jd �W S tttfy Y d S w )Nzcontent-length)�int�headers�
ValueError�KeyError� TypeError)r � r �I/usr/local/lib/python3.10/site-packages/pip/_internal/network/download.py�_get_http_response_size s
�r �link� progress_barc C s� t | �}|jtjkr|j}n|j}t|�}|r d�|t|��}t | �r*t
� d|� nt
� d|� t
� � t
jkr:d}nt | �rAd}n|sFd}n |dkrMd}nd}t| t�}|sX|S t||d�}||�S )Nz{} ({})zUsing cached %szDownloading %sFTi@� )�bar_type�size)r �netlocr �file_storage_domain�show_url�url_without_fragmentr �formatr r �logger�info�getEffectiveLevel�logging�INFOr r r ) r r r � total_length�url�
logged_url�
show_progress�chunks�rendererr r r �_prepare_download s0
r1 �filenamec C s t j�| �S )zJ
Sanitize the "filename" value from a Content-Disposition header.
)�os�path�basename)r2 r r r �sanitize_content_filenameH s r6 �content_disposition�default_filenamec C s4 t j�� }| |d< |�d�}|rtt|��}|p|S )z�
Parse the "filename" value from a Content-Disposition header, and
return the default filename if the result is empty.
� content-typer2 )�email�message�Message� get_paramr6 �str)r7 r8 �mr2 r r r �parse_content_dispositionO s
r@ c C s� |j }| j�d�}|rt||�}t|�d }|s(t�| j�dd��}|r(||7 }|s?|j| jkr?tj �| j�d }|r?||7 }|S )zoGet an ideal filename from the given HTTP response, falling back to
the link filename if not provided.
zcontent-disposition� r9 � )
r2 r �getr@ r � mimetypes�guess_extensionr, r3 r4 )r r r2 r7 �extr r r �_get_http_response_filename^ s
rG �sessionc C s. |j �dd�d }| j|tdd�}t|� |S )N�#rA r T)r �stream)r, �splitrC r
r )rH r �
target_urlr r r r �_http_get_downloads s rM c @ s@ e Zd Zdededdfdd�Zdededeeef fd d
�ZdS ) �
DownloaderrH r r Nc C � || _ || _d S �N��_session�
_progress_bar��selfrH r r r r �__init__{ �
zDownloader.__init__r �locationc
C s� zt | j|�}W n ty% } z|jdusJ �t�d|jj|� � d}~ww t||�}tj �
||�}t ||| j �}t
|d��}|D ]} |�| � qAW d � n1 sSw Y |j�dd�}
||
fS )z.Download the file given by link into location.N� HTTP error %s while getting %s�wb� Content-TyperB �rM rR r �responser& �critical� status_coderG r3 r4 �joinr1 rS �open�writer rC ) rU r rX r �er2 �filepathr/ � content_file�chunk� content_typer r r �__call__� s(
���
� �zDownloader.__call__) �__name__�
__module__� __qualname__r r> rV r
r rh r r r r rN z s ��
�"rN c @ sP e Zd Zdededdfdd�Zdee dedeeeeeef f fd d
�Z dS ) �BatchDownloaderrH r r Nc C rO rP rQ rT r r r rV � rW zBatchDownloader.__init__�linksrX c c s� � |D ]h}zt | j|�}W n ty* } z|jdus J �t�d|jj|� � d}~ww t||�}tj �
||�}t ||| j �}t
|d��} |D ]}
| �|
� qFW d � n1 sXw Y |j�dd�} ||| ffV qdS )z0Download the files given by links into location.NrY rZ r[ rB r\ ) rU rm rX r r rc r2 rd r/ re rf rg r r r rh � s2 ����
� ��zBatchDownloader.__call__)
ri rj rk r r> rV r r
r rh r r r r rl � s ��
����rl )/�__doc__�
email.messager: r) rD r3 �typingr r r Zpip._vendor.requests.modelsr r � pip._internal.cli.progress_barsr �pip._internal.exceptionsr �pip._internal.models.indexr �pip._internal.models.linkr
�pip._internal.network.cacher � pip._internal.network.sessionr �pip._internal.network.utilsr
r r �pip._internal.utils.miscr r r � getLoggerri r& r r r> �bytesr1 r6 r@ rG rM rN rl r r r r �<module> s@
���
�*