관리-도구
편집 파일: download.cpython-310.pyc
o ƚRe� � @ 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)�DownloadProgressProvider)�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 ��/builddir/build/BUILDROOT/alt-python310-pip-21.3.1-3.el8.x86_64/opt/alt/python310/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@� )�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/ �filenamec 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_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. r0 )�cgi� parse_header�getr4 )r5 r6 �_type�paramsr0 r r r �parse_content_dispositionN 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� ) r0 r r9 r< r � mimetypes�guess_extensionr+ r1 r2 )r r r0 r5 �extr r r �_get_http_response_filename\ s rB �sessionc 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 @ s@ e Zd Zdededdfdd�Zdededeeef fd d �ZdS ) � DownloaderrC r r Nc C � || _ || _d S �N��_session� _progress_bar��selfrC r r r r �__init__y � 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> �rH rM r �responser% �critical� status_coderB r1 r2 �joinr/ rN �open�writer r9 ) rP r rS r �er0 �filepathr. � content_file�chunk� content_typer r r �__call__� s( ��� � �zDownloader.__call__) �__name__� __module__� __qualname__r �strrQ r r rc r r r r rI x s �� �"rI c @ sP e Zd Zdededdfdd�Zdee dedeeeeeef f fd d �Z dS ) �BatchDownloaderrC r r Nc C rJ rK rL rO r r r rQ � rR zBatchDownloader.__init__�linksrS 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.NrT rU rV r> rW ) rP ri rS r r r^ r0 r_ r. r` ra rb r r r rc � s2 ���� � ��zBatchDownloader.__call__) rd re rf r rg rQ r r r rc r r r r rh � s �� ����rh ).�__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 � getLoggerrd r% r r rg �bytesr/ r4 r< rB rH rI rh r r r r �<module> s@ ��� �)