관리-도구
편집 파일: 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 � )