File: //usr/lib/python3/dist-packages/pip/_internal/network/__pycache__/download.cpython-310.pyc
o
@%Ne� � @ sj d Z ddlZddlZddlZddlZddlmZmZmZ ddl m
Z
m Z ddl m
Z
ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZ dd lmZmZmZ 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 �@/usr/lib/python3/dist-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 s, t �| �\}}|�d�}|rt|�}|p|S )z�
Parse the "filename" value from a Content-Disposition header, and
return the default filename if the result is empty.
r2 )�cgi� parse_header�getr6 )r7 r8 �_type�paramsr2 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� z content-type� )
r2 r r; r> r � mimetypes�guess_extensionr, r3 r4 )r r r2 r7 �extr r r �_get_http_response_filename] s
rD �sessionc C s. |j �dd�d }| j|tdd�}t|� |S )N�#r? r T)r �stream)r, �splitr; r
r )rE r �
target_urlr r r r �_http_get_downloadr s rJ c @ s@ e Zd Zdededdfdd�Zdededeeef fd d
�ZdS ) �
DownloaderrE r r Nc C � || _ || _d S �N��_session�
_progress_bar��selfrE r r r r �__init__z �
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-Typer@ �rJ rO r �responser& �critical� status_coderD r3 r4 �joinr1 rP �open�writer r; ) rR r rU r �er2 �filepathr/ � content_file�chunk� content_typer r r �__call__� s(
���
� �zDownloader.__call__) �__name__�
__module__� __qualname__r �strrS r
r re r r r r rK y s ��
�"rK c @ sP e Zd Zdededdfdd�Zdee dedeeeeeef f fd d
�Z dS ) �BatchDownloaderrE r r Nc C rL rM rN rQ r r r rS � rT zBatchDownloader.__init__�linksrU 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.NrV rW rX r@ rY ) rR rk rU r r r` r2 ra r/ rb rc rd r r r re � s2 ����
� ��zBatchDownloader.__call__)
rf rg rh r ri rS r r
r re r r r r rj � s ��
����rj ).�__doc__r9 r) rA r3 �typingr r r �pip._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 � getLoggerrf r& r r ri �bytesr1 r6 r> rD rJ rK rj r r r r �<module> s@
���
�*