관리-도구
편집 파일: download.cpython-39.pyc
a ��Re� � @ 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 ��/builddir/build/BUILDROOT/alt-python39-pip-21.3.1-2.el8.x86_64/opt/alt/python39/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 �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�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)r0 r r r �sanitize_content_filenameG s r4 )�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. r0 )�cgi� parse_header�getr4 )r5 r6 �_type�paramsr0 r r r �parse_content_dispositionN s r< )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� ) r0 r r9 r< r � mimetypes�guess_extensionr+ r1 r2 )r r r0 r5 �extr r r �_get_http_response_filename\ s rB )�sessionr r c C s. |j �dd�d }| j|tdd�}t|� |S )N�#r= r T)r �stream)r+ �splitr9 r r )rC r � target_urlr r r r �_http_get_downloadq s rH c @ s8 e Zd Zeedd�dd�Zeeeeef d�dd�ZdS )� DownloaderN�rC r r c C s || _ || _d S �N��_session� _progress_bar��selfrC 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-Typer> �rH rM r �responser% �critical� status_coderB r1 r2 �joinr/ rN �open�writer r9 ) rP r rR r �er0 �filepathr. � content_file�chunk� content_typer r r �__call__� s � *zDownloader.__call__) �__name__� __module__� __qualname__r �strrQ r r rb r r r r rI x s � rI c @ sH e Zd Zeedd�dd�Zee eeeeeeef f d�dd�Z dS )�BatchDownloaderNrJ c C s || _ || _d S rK rL rO r r r rQ � s zBatchDownloader.__init__)�linksrR 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.NrS rT rU r> rV ) rP rh rR r r r] r0 r^ r. r_ r` ra r r r rb � s& � *zBatchDownloader.__call__) rc rd re r rf rQ r r r rb r r r r rg � s � �rg ).�__doc__r7 r( r? r1 �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 � getLoggerrc r% r r rf �bytesr/ r4 r< rB rH rI rg r r r r �<module> s6 � )