3
Re @ s* d Z ddlZddlZddlZddlZddlmZ ddlmZ ddl m
Z
ddl m Z ddl
mZ ddlmZmZmZ dd lmZmZmZ dd
lmZ erdd lmZmZ dd lm Z dd
l m Z ddl m Z ej!e"Z#dd Z$dd Z%dd Z&dd Z'dd Z(dd Z)G dd d e*Z+G d d d e*Z,dS ) z)Download files with progress indicators.
N)CONTENT_CHUNK_SIZE)DownloadProgressProvider)NetworkConnectionError)PyPI)
is_from_cache)HEADERSraise_for_statusresponse_chunks) format_sizeredact_auth_from_urlsplitext)MYPY_CHECK_RUNNING)IterableOptional)Response)Link)
PipSessionc
C s. yt | jd S tttfk
r( d S X d S )Nzcontent-length)intheaders
ValueErrorKeyError TypeError)resp r /builddir/build/BUILDROOT/alt-python36-pip-20.2.4-5.el8.x86_64/opt/alt/python36/lib/python3.6/site-packages/pip/_internal/network/download.py_get_http_response_size% s r c C s t | }|jtjkr |j}n|j}t|}|r>dj|t|}t | rTt
j d| n t
j d| t
j t
jkrtd}n*t | rd}n |sd}n|d krd}nd}t| t}|s|S t||d|S )
Nz{} ({})zUsing cached %szDownloading %sFT( i )maxi@ )r netlocr file_storage_domainshow_urlurl_without_fragmentr formatr
r loggerinfogetEffectiveLevelloggingINFOr r r )r link progress_bar total_lengthurl
logged_url
show_progresschunksr r r _prepare_download- s2
r/ c C s t jj| S )zJ
Sanitize the "filename" value from a Content-Disposition header.
)ospathbasename)filenamer r r sanitize_content_filenameY s r4 c C s, t j| \}}|jd}|r$t|}|p*|S )z
Parse the "filename" value from a Content-Disposition header, and
return the default filename if the result is empty.
r3 )cgi parse_headergetr4 )content_dispositiondefault_filename_typeparamsr3 r r r parse_content_dispositiona s
r< c C s |j }| jjd}|r t||}t|d }|sPtj| jjdd}|rP||7 }| r|j| jkrtj j| jd }|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 )
r3 r r7 r< r mimetypesguess_extensionr+ r0 r1 )r r( r3 r8 extr r r _get_http_response_filenamep s
rB c C s. |j jddd }| j|tdd}t| |S )N#r= r T)r stream)r+ splitr7 r r )sessionr(
target_urlr r r r _http_get_download s rH c @ s e Zd Zdd ZdS )Downloadc C s || _ || _|| _d S )N)responser3 r. )selfrJ r3 r. r r r __init__ s zDownload.__init__N)__name__
__module__ __qualname__rL r r r r rI s rI c @ s e Zd Zdd Zdd ZdS )
Downloaderc C s || _ || _d S )N)_session
_progress_bar)rK rF r) r r r rL s zDownloader.__init__c C sr yt | j|}W nD tk
rT } z(|jd k s0ttjd|jj| W Y d d }~X nX t|t ||t
||| j S )Nz HTTP error %s while getting %s) rH rQ r rJ AssertionErrorr# critical status_coderI rB r/ rR )rK r( r er r r __call__ s zDownloader.__call__N)rM rN rO rL rW r r r r rP s rP )-__doc__r5 r& r? r0 Zpip._vendor.requests.modelsr pip._internal.cli.progress_barsr pip._internal.exceptionsr pip._internal.models.indexr pip._internal.network.cacher pip._internal.network.utilsr r r pip._internal.utils.miscr
r r pip._internal.utils.typingr
typingr r r pip._internal.models.linkr pip._internal.network.sessionr getLoggerrM r# r r/ r4 r< rB rH objectrI rP r r r r s4
,