a

?�f�@s*
dZd
dl
Z
d
dlZd
dlZd
dlZd
dlmZ
d
dlmZ
d
dl	m
Z

d
dlmZ
d
dl
mZ
d
dlmZmZmZ
d
d	lmZmZmZ
d
d
lmZ
er�d
dlmZmZ
d
dlmZ
d
d
lmZ
d
dlm Z 
e�!e"�
Z#dd�Z$dd�Z%dd�Z&dd�Z'dd�Z(dd�Z)Gdd�de*�Z+Gdd�de*�Z,dS)z)Download files with progress indicators.
�N)
�CONTENT_CHUNK_SIZE)
�DownloadProgressProvider)
�NetworkConnectionError)
�PyPI)
�
is_from_cache)�HEADERS�raise_for_status�response_chunks)�format_size�redact_auth_from_url�splitext)
�MYPY_CHECK_RUNNING)�Iterable�Optional)
�Response)
�Link)
�
PipSessionc


Cs0zt|j
d
�
WStttfy*


YdS0dS)Nzcontent-length)�int�headers�
ValueError�KeyError�	TypeError)
�resp�r�B/usr/lib/python3.9/site-packages/pip/_internal/network/download.py�_get_http_response_size%s


rcCs�t|�
}|
j
tjkr|
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�|St||d�|�
S)Nz{} ({})zUsing cached %szDownloading %sFTi@�)
�max)r�netlocrZfile_storage_domainZshow_urlZurl_without_fragmentr�formatr
r�logger�infoZgetEffectiveLevel�logging�INFOr	rr)r�link�progress_barZtotal_length�urlZ
logged_urlZ
show_progress�chunksrrr�_prepare_download-s6












��r'c

Cstj
�|�
S)
zJ
    Sanitize the "filename" value from a Content-Disposition header.
    )�os�path�basename)
�filenamerrr�sanitize_content_filenameYsr,cCs,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,)�content_dispositionZdefault_filenameZ_type�paramsr+rrr�parse_content_dispositionas




r1cCs�|
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+rr.r1r�	mimetypesZguess_extensionr%r(r))rr#r+r/Zextrrr�_get_http_response_filenameps 






�





r5cCs.|
j�
d
d�d}|j|tdd�}t|�

|S)N�
#r2r
T)r�stream)r%�splitr.rr)�sessionr#Z
target_urlrrrr�_http_get_download�s


r:c@seZ
dZd
d�ZdS)�DownloadcCs|
|_||_
||_dS�
N)�responser+r&)�selfr=r+r&rrr�__init__�s

zDownload.__init__N)�__name__�
__module__�__qualname__r?rrrrr;�s
r;c@seZ
dZd
d�Zdd�ZdS)�
DownloadercCs|
|_||_
dSr<)�_session�
_progress_bar)r>r9r$rrrr?�s
zDownloader.__init__c
Cstzt|j
|
�}WnFtyV
}
z.|jdu
s.J�
t�d
|jj|
�
�WYd}~n
d}~00t|t||
�t	||
|j
��S)NzHTTP error %s while getting %s)r:rDrr=rZcriticalZstatus_coder;r5r'rE)r>r#r�
errr�__call__�s





�


�zDownloader.__call__N)r@rArBr?rGrrrrrC�s
	rC)-�__doc__r-r!r4r(Zpip._vendor.requests.modelsrZpip._internal.cli.progress_barsrZpip._internal.exceptionsrZpip._internal.models.indexrZpip._internal.network.cacherZpip._internal.network.utilsrrr	Zpip._internal.utils.miscr
rrZpip._internal.utils.typingr
�typingrrrZpip._internal.models.linkrZpip._internal.network.sessionrZ	getLoggerr@rrr'r,r1r5r:�objectr;rCrrrr�<module>
s4









,