o
ai @ sj 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"d
e dee# fddZ$d
e dede%dee& fddZ'de%de%fddZ(de%de%de%fddZ)d
e dede%fd d Z*d edede fd d Z+G d!d" d"Z,G d#d$ d$Z-dS )%z)Download files with progress indicators.
N)IterableOptionalTuple)CONTENT_CHUNK_SIZEResponse)DownloadProgressProvider)NetworkConnectionError)PyPI)Link)
is_from_cache)
PipSession)HEADERSraise_for_statusresponse_chunks) format_sizeredact_auth_from_urlsplitextrespreturnc
C s, zt | jd W S tttfy Y d S w )Nzcontent-length)intheaders
ValueErrorKeyError TypeError)r r /builddir/build/BUILDROOT/alt-python310-pip-21.3.1-5.el8.x86_64/opt/alt/python310/lib/python3.10/site-packages/pip/_internal/network/download.py_get_http_response_size s
r link progress_barc C s t | }|jtjkr|j}n|j}t|}|r d|t|}t | r*t
d| nt
d| t
t
jkr:d}nt | rAd}n|sFd}n |dkrMd}nd}t| t}|sX|S t||d|S )Nz{} ({})zUsing cached %szDownloading %sFTi@ )max)r netlocr file_storage_domainshow_urlurl_without_fragmentr formatr r loggerinfogetEffectiveLevelloggingINFOr r r )r r r total_lengthurl
logged_url
show_progresschunksr r r _prepare_download s.
r/ filenamec C s t j| S )zJ
Sanitize the "filename" value from a Content-Disposition header.
)ospathbasename)r0 r r r sanitize_content_filenameG s r4 content_dispositiondefault_filenamec C s, t | \}}|d}|rt|}|p|S )z
Parse the "filename" value from a Content-Disposition header, and
return the default filename if the result is empty.
r0 )cgi parse_headergetr4 )r5 r6 _typeparamsr0 r r r parse_content_dispositionN s
r< c C s |j }| jd}|rt||}t|d }|s(t| jdd}|r(||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 )
r0 r r9 r< r mimetypesguess_extensionr+ r1 r2 )r r r0 r5 extr r r _get_http_response_filename\ s
rB sessionc C s. |j ddd }| j|tdd}t| |S )N#r= r T)r stream)r+ splitr9 r
r )rC r
target_urlr r r r _http_get_downloadq s rH c @ s@ e Zd ZdededdfddZdededeeef fd d
ZdS )
DownloaderrC r r Nc C || _ || _d S N_session
_progress_barselfrC r r r r __init__y
zDownloader.__init__r locationc
C s zt | j|}W n ty% } z|jdusJ td|jj| d}~ww t||}tj
||}t ||| j }t
|d}|D ]} || qAW d n1 sSw Y |jdd}
||
fS )z.Download the file given by link into location.N HTTP error %s while getting %swb Content-Typer> rH rM r responser% critical status_coderB r1 r2 joinr/ rN openwriter r9 ) rP r rS r er0 filepathr. content_filechunk content_typer r r __call__ s(
zDownloader.__call__) __name__
__module__ __qualname__r strrQ r
r rc r r r r rI x s
"rI c @ sP e Zd ZdededdfddZdee dedeeeeeef f fd d
Z dS ) BatchDownloaderrC r r Nc C rJ rK rL rO r r r rQ rR zBatchDownloader.__init__linksrS c c s |D ]h}zt | j|}W n ty* } z|jdus J td|jj| d}~ww t||}tj
||}t ||| j }t
|d} |D ]}
| |
qFW d n1 sXw Y |jdd} ||| ffV qdS )z0Download the files given by links into location.NrT rU rV r> rW ) rP ri rS r r r^ r0 r_ r. r` ra rb r r r rc s2
zBatchDownloader.__call__)
rd re rf r rg rQ r r
r rc r r r r rh s
rh ).__doc__r7 r( r? r1 typingr r r pip._vendor.requests.modelsr r 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 getLoggerrd r% r r rg bytesr/ r4 r< rB rH rI rh r r r r s@
)