B
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 e!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-python37-pip-20.2.4-6.el8.x86_64/opt/alt/python37/lib/python3.7/site-packages/pip/_internal/network/download.py_get_http_response_size% s r c C s t | }|jtjkr |j}n|j}t|}|r>d|t|}t | rTt
d| n t
d| t
t
jkrtd}n*t | rd}n |sd}n|dkrd}nd}t| t}|s|S t||d|S )Nz{} ({})zUsing cached %szDownloading %sFTi@ )max)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 j| S )zJ
Sanitize the "filename" value from a Content-Disposition header.
)ospathbasename)filenamer r r sanitize_content_filenameY s r3 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.
r2 )cgi parse_headergetr3 )content_dispositiondefault_filename_typeparamsr2 r r r parse_content_dispositiona s
r; c C s |j }| jd}|r t||}t|d }|sPt| jdd}|rP||7 }|s~|j| jkr~tj | 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 )
r2 r r6 r; r mimetypesguess_extensionr* r/ r0 )r r' r2 r7 extr r r _get_http_response_filenamep s
rA c C s. |j ddd }| j|tdd}t| |S )N#r< r T)r stream)r* splitr6 r r )sessionr'
target_urlr r r r _http_get_download s rG c @ s e Zd Zdd ZdS )Downloadc C s || _ || _|| _d S )N)responser2 r- )selfrI r2 r- r r r __init__ s zDownload.__init__N)__name__
__module__ __qualname__rK r r r r rH s rH c @ s e Zd Zdd Zdd ZdS )
Downloaderc C s || _ || _d S )N)_session
_progress_bar)rJ rE r( r r r rK s zDownloader.__init__c
C sr yt | j|}W nD tk
rT } z&|jd k s0ttd|jj| W d d }~X Y nX t|t ||t
||| j S )Nz HTTP error %s while getting %s) rG rP r rI AssertionErrorr" critical status_coderH rA r. rQ )rJ r' r er r r __call__ s zDownloader.__call__N)rL rM rN rK rV r r r r rO s rO )-__doc__r4 r% r> r/ 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 getLoggerrL r" r r. r3 r; rA rG objectrH rO r r r r s4
,