| Current File : //lib/python3.9/site-packages/pip/_internal/network/__pycache__/download.cpython-39.pyc |
a
[��f� � @ sT 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"e ee# d
�dd�Z$e ee%ee& d�dd�Z'e%e%d�dd�Z(e%e%e%d�dd�Z)e ee%d�dd�Z*eee d �d 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)�DownloadProgressProvider)�NetworkConnectionError)�PyPI)�Link)�
is_from_cache)�
PipSession)�HEADERS�raise_for_status�response_chunks)� format_size�redact_auth_from_url�splitext)�resp�returnc
C s0 zt | jd �W S tttfy* Y d S 0 d S )Nzcontent-length)�int�headers�
ValueError�KeyError� TypeError)r � r �B/usr/lib/python3.9/site-packages/pip/_internal/network/download.py�_get_http_response_size s r )r �link� progress_barr c C s� t | �}|jtjkr |j}n|j}t|�}|r>d�|t|��}t | �rTt
� d|� n t
� d|� t
� � t
jkrtd}n*t | �r�d}n |s�d}n|dkr�d}nd}t| t�}|s�|S t||d�|�S )Nz{} ({})zUsing cached %szDownloading %sFTi@� )�max)r �netlocr Zfile_storage_domainZshow_urlZurl_without_fragmentr �formatr r �logger�infoZgetEffectiveLevel�logging�INFOr r r )r r r Z total_length�urlZ
logged_urlZ
show_progress�chunksr r r �_prepare_download s.
r( )�filenamer c C s t j�| �S )zJ
Sanitize the "filename" value from a Content-Disposition header.
)�os�path�basename)r) r r r �sanitize_content_filenameG s r- )�content_disposition�default_filenamer c C s, t �| �\}}|�d�}|r$t|�}|p*|S )z�
Parse the "filename" value from a Content-Disposition header, and
return the default filename if the result is empty.
r) )�cgiZ parse_header�getr- )r. r/ Z_type�paramsr) r r r �parse_content_dispositionN s
r3 )r r r c C s� |j }| j�d�}|r t||�}t|�d }|sPt�| j�dd��}|rP||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� )
r) r r1 r3 r � mimetypesZguess_extensionr& r* r+ )r r r) r. �extr r r �_get_http_response_filename\ s
r8 )�sessionr r c C s. |j �dd�d }| j|tdd�}t|� |S )N�#r4 r T)r �stream)r& �splitr1 r
r )r9 r Z
target_urlr r r r �_http_get_downloadq s r= c @ s8 e Zd Zeedd�dd�Zeeeeef d�dd�ZdS )�
DownloaderN�r9 r r c C s || _ || _d S �N��_session�
_progress_bar��selfr9 r r r r �__init__y s zDownloader.__init__)r �locationr c
C s� zt | j|�}W nF tyV } z.|jdus.J �t�d|jj|� � W Y d}~n
d}~0 0 t||�}tj �
||�}t ||| j �}t
|d��$}|D ]} |�| � q�W d � n1 s�0 Y |j�dd�}
||
fS )z.Download the file given by link into location.N� HTTP error %s while getting %s�wb� Content-Typer5 �r= rB r Zresponser" ZcriticalZ status_coder8 r* r+ �joinr( rC �open�writer r1 ) rE r rG r �er) �filepathr' � content_file�chunk� content_typer r r �__call__� s
�
*zDownloader.__call__) �__name__�
__module__� __qualname__r �strrF r
r rT r r r r r> x s
� r> c @ sH e Zd Zeedd�dd�Zee eeeeeeef f d�dd�Z dS )�BatchDownloaderNr? c C s || _ || _d S r@ rA rD r r r rF � s zBatchDownloader.__init__)�linksrG r c c s� |D ]�}zt | j|�}W nF ty^ } z.|jdus6J �t�d|jj|� � W Y d}~n
d}~0 0 t||�}tj �
||�}t ||| j �}t
|d��$} |D ]}
| �|
� q�W d � n1 s�0 Y |j�dd�} ||| ffV qdS )z0Download the files given by links into location.NrH rI rJ r5 rK ) rE rZ rG r r rO r) rP r' rQ rR rS r r r rT � s& �
*zBatchDownloader.__call__)
rU rV rW r rX rF r r
r rT r r r r rY � s � �rY ).�__doc__r0 r$ r6 r* �typingr r r Zpip._vendor.requests.modelsr r Z pip._internal.cli.progress_barsr Zpip._internal.exceptionsr Zpip._internal.models.indexr Zpip._internal.models.linkr
Zpip._internal.network.cacher Z pip._internal.network.sessionr Zpip._internal.network.utilsr
r r Zpip._internal.utils.miscr r r Z getLoggerrU r" r r rX �bytesr( r- r3 r8 r= r> rY r r r r �<module> s6
� )