관리-도구
편집 파일: download.cpython-38.pyc
U &?�f� � @ sP d Z ddlZddlZddlZddlZddlmZmZm Z ddl 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)�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 s2 zt | jd �W S tttfk r, Y d S X d S )Nzcontent-length)�int�headers� ValueError�KeyError� TypeError)r � r �L/opt/hc_python/lib/python3.8/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|�}|rB|� dt|�� d�}t| �rXt � d|� n t � d|� t � � t j krxd}n*t| �r�d}n |s�d}n|dkr�d}nd}t| �}|s�|S t||d�}||�S ) Nz (�)zUsing cached %szDownloading %sFTi )�bar_type�size)r �netlocr �file_storage_domain�show_url�url_without_fragmentr r r �logger�info�getEffectiveLevel�logging�INFOr r ) r r r � total_length�url� logged_url� show_progress�chunks�rendererr r r �_prepare_download s0 r0 )�filenamer c C s t j�| �S )zJ Sanitize the "filename" value from a Content-Disposition header. )�os�path�basename)r1 r r r �sanitize_content_filenameI s r5 )�content_disposition�default_filenamer c C s4 t j�� }| |d< |�d�}|r,tt|��}|p2|S )z� Parse the "filename" value from a Content-Disposition header, and return the default filename if the result is empty. � content-typer1 )�email�message�Message� get_paramr5 �str)r6 r7 �mr1 r r r �parse_content_dispositionP 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� r8 � ) r1 r �getr? r � mimetypes�guess_extensionr+ r2 r3 )r r r1 r6 �extr r r �_get_http_response_filename_ s rF )�sessionr r c C s. |j �dd�d }| j|tdd�}t|� |S )N�#r@ r T)r �stream)r+ �splitrB r r )rG r � target_urlr r r r �_http_get_downloadt s rL c @ s8 e Zd Zeedd�dd�Zeeeeef d�dd�ZdS )� DownloaderN�rG r r c C s || _ || _d S �N��_session� _progress_bar��selfrG r r r r �__init__| s zDownloader.__init__)r �locationr c C s� zt | j|�}W nD tk rT } z&|jdk s0t�t�d|jj|� � W 5 d}~X Y nX t||�}t j � ||�}t ||| j �}t|d��}|D ]} |�| � q�W 5 Q R X |j�dd�} || fS )z.Download the file given by link into location.N� HTTP error %s while getting %s�wb� Content-TyperA �rL rQ r �response�AssertionErrorr% �critical� status_coderF r2 r3 �joinr0 rR �open�writer rB ) rT r rV r �er1 �filepathr. � content_file�chunk� content_typer r r �__call__� s$ � zDownloader.__call__) �__name__� __module__� __qualname__r r= rU r r rg r r r r rM { s � rM c @ sH e Zd Zeedd�dd�Zee eeeeeeef f d�dd�Z dS )�BatchDownloaderNrN c C s || _ || _d S rO rP rS r r r rU � s zBatchDownloader.__init__)�linksrV r c c s� |D ]�}zt | j|�}W nD tk r\ } z&|jdk s8t�t�d|jj|� � W 5 d}~X Y nX t||�}t j � ||�}t ||| j �}t|d��} |D ]} | �| � q�W 5 Q R X |j�dd�} ||| ffV qdS )z0Download the files given by links into location.NrW rX rY rA rZ ) rT rl rV r r rb r1 rc r. rd re rf r r r rg � s& � zBatchDownloader.__call__) rh ri rj r r= rU r r r rg r r r r rk � s � �rk ).�__doc__� email.messager9 r( rC r2 �typingr r r Zpip._vendor.requests.modelsr � 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 � getLoggerrh r% r r r= �bytesr0 r5 r? rF rL rM rk r r r r �<module> s6 � *