U
ʗRe @ sT d Z ddlZddlZddlZddlZddlmZmZm Z ddl
m Z m Z ddl
mZ ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZ dd lmZmZm Z dd l m Z m Z m Z e!e"Z#e ee$ d
ddZ%e ee&ee' dddZ(e&e&dddZ)e&e&e&dddZ*e ee&dddZ+eee d d d Z,G d d d Z-G d!d" d"Z.dS )#z)Download files with progress indicators.
N)IterableOptionalTuple)CONTENT_CHUNK_SIZEResponse) get_download_progress_renderer)NetworkConnectionError)PyPI)Link)
is_from_cache)
PipSession)HEADERSraise_for_statusresponse_chunks) format_sizeredact_auth_from_urlsplitext)respreturnc
C s2 zt | jd W S tttfk
r, Y d S X d S )Nzcontent-length)intheaders
ValueErrorKeyError TypeError)r r /builddir/build/BUILDROOT/alt-python38-pip-22.2.1-2.el8.x86_64/opt/alt/python38/lib/python3.8/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 | rd}n |sd}n|dkrd}nd}t| t}|s|S t||d}||S )Nz{} ({})zUsing cached %szDownloading %sFTi@ )bar_typesize)r netlocr file_storage_domainshow_urlurl_without_fragmentr formatr r loggerinfogetEffectiveLevelloggingINFOr r r ) r r r total_lengthurl
logged_url
show_progresschunksrendererr r r _prepare_download s0
r1 )filenamer c C s t j| S )zJ
Sanitize the "filename" value from a Content-Disposition header.
)ospathbasename)r2 r r r sanitize_content_filenameH s r6 )content_dispositiondefault_filenamer c C s4 t j }| |d< |d}|r,tt|}|p2|S )z
Parse the "filename" value from a Content-Disposition header, and
return the default filename if the result is empty.
content-typer2 )emailmessageMessage get_paramr6 str)r7 r8 mr2 r r r parse_content_dispositionO s
r@ )r r 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 r9 )
r2 r getr@ r mimetypesguess_extensionr, r3 r4 )r r r2 r7 extr r r _get_http_response_filename^ s
rG )sessionr r c C s. |j ddd }| j|tdd}t| |S )N#rA r T)r stream)r, splitrC r
r )rH r
target_urlr r r r _http_get_downloads s rM c @ s8 e Zd ZeeddddZeeeeef dddZdS )
DownloaderNrH r r c C s || _ || _d S N_session
_progress_barselfrH r r r r __init__{ s zDownloader.__init__)r locationr c
C s zt | j|}W nD tk
rT } z&|jdk s0ttd|jj| W 5 d}~X Y nX t||}t j
||}t ||| j
}t|d}|D ]} || qW 5 Q R X |jdd}
||
fS )z.Download the file given by link into location.N HTTP error %s while getting %swb Content-TyperB rM rR r responseAssertionErrorr& critical status_coderG r3 r4 joinr1 rS openwriter rC ) rU r rW r er2 filepathr/ content_filechunk content_typer r r __call__ s$
zDownloader.__call__) __name__
__module__ __qualname__r r> rV r
r rh r r r r rN z s
rN c @ sH e Zd ZeeddddZee eeeeeeef f dddZ dS )BatchDownloaderNrO c C s || _ || _d S rP rQ rT r r r rV s zBatchDownloader.__init__)linksrW r c c s |D ]}zt | j|}W nD tk
r\ } z&|jdk s8ttd|jj| W 5 d}~X Y nX t||}t j
||}t ||| j
}t|d} |D ]}
| |
qW 5 Q R X |jdd} ||| ffV qdS )z0Download the files given by links into location.NrX rY rZ rB r[ ) rU rm rW r r rc r2 rd r/ re rf rg r r r rh s&
zBatchDownloader.__call__)
ri rj rk r r> rV r r
r rh r r r r rl s rl )/__doc__
email.messager: r) rD r3 typingr r r Zpip._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 getLoggerri r& r r r> bytesr1 r6 r@ rG rM rN rl r r r r s6
*