a
lj @ s d Z ddgZddlmZmZ ddlmZ ddlmZ ddl m
Z
m Z m Z m
Z
mZmZ ddlmZmZ dd lmZ dd
lmZmZ dd lmZmZmZ dd l m Z dd
l m Z m Z m!Z! G dd de"Z#e$e$e edddZ%G dd dZ&dS )zLazy ZIP over HTTPHTTPRangeRequestUnsupporteddist_from_wheel_url ) bisect_left bisect_right)contextmanager)NamedTemporaryFile)AnyDictIteratorListOptionalTuple)
BadZipfileZipFile)canonicalize_name)CONTENT_CHUNK_SIZEResponse)BaseDistribution MemoryWheelget_wheel_distribution)
PipSession)HEADERSraise_for_statusresponse_chunksc @ s e Zd ZdS )r N)__name__
__module__ __qualname__ r r D/usr/lib/python3.9/site-packages/pip/_internal/network/lazy_wheel.pyr s )nameurlsessionreturnc C sH t ||*}t|j|}t|t| W d S 1 s:0 Y dS )a Return a distribution object from the given wheel URL.
This uses HTTP range requests to only fetch the potion of the wheel
containing metadata, just enough for the object to be constructed.
If such requests are not supported, HTTPRangeRequestUnsupported
is raised.
N)LazyZipOverHTTPr r r r )r r r! zfZwheelr r r r s c @ sz e Zd ZdZefeeeddddZe edddZ
e edd d
Z e dd d Z
ddd
dZe e dddZd3eedddZe dddZd4eeedddZeddd Zd5ee edd d Ze dd d Zd dd!d"Zeee d#d$d%Zee d dd&d'Z ddd(d)Z e feee eef e!d*d+d,Z"eeeee e#eef d-d.d/Z$eedd0d1d2Z%dS )6r# a File-like object mapped to a ZIP file over HTTP.
This uses HTTP range requests to lazily fetch the file's content,
which is supposed to be fed to ZipFile. If such requests are not
supported by the server, raise HTTPRangeRequestUnsupported
during initialization.
N)r r!
chunk_sizer" c C s |j |td}t| |jdks$J ||| | _| _| _t|jd | _ t
| _ | | j g | _
g | _d|jddvrtd| d S )N)headers zContent-Lengthbytesz
Accept-RangesZnonez range request is not supported)headr r Z status_code_session_url _chunk_sizeintr& _lengthr _filetruncate_left_rightgetr
_check_zip)selfr r! r% r) r r r __init__1 s zLazyZipOverHTTP.__init__)r" c C s dS )z!Opening mode, which is always rb.rbr r5 r r r modeA s zLazyZipOverHTTP.modec C s | j jS )z Path to the underlying file.)r/ r r8 r r r r F s zLazyZipOverHTTP.namec C s dS )z9Return whether random access is supported, which is True.Tr r8 r r r seekableK s zLazyZipOverHTTP.seekablec C s | j dS )zClose the file.N)r/ closer8 r r r r; O s zLazyZipOverHTTP.closec C s | j jS )zWhether the file is closed.)r/ closedr8 r r r r< S s zLazyZipOverHTTP.closed)sizer" c C s` t || j}| | j }}|dk r(|n t|| |}t d|| }| ||d | j|S )zRead up to size bytes from the object and return them.
As a convenience, if size is unspecified or -1,
all bytes until EOF are returned. Fewer than
size bytes may be returned if EOF is reached.
r )maxr, tellr. min _downloadr/ read)r5 r> Z
download_sizestartlengthstopr r r rD X s zLazyZipOverHTTP.readc C s dS )z3Return whether the file is readable, which is True.Tr r8 r r r readablef s zLazyZipOverHTTP.readabler )offsetwhencer" c C s | j ||S )a- Change stream position and return the new absolute position.
Seek to offset relative position indicated by whence:
* 0: Start of stream (the default). pos should be >= 0;
* 1: Current position - pos may be negative;
* 2: End of stream - pos usually negative.
)r/ seek)r5 rI rJ r r r rK j s zLazyZipOverHTTP.seekc C s
| j S )z Return the current position.)r/ rA r8 r r r rA t s zLazyZipOverHTTP.tellc C s | j |S )zResize the stream to the given size in bytes.
If size is unspecified resize to the current position.
The current stream position isn't changed.
Return the new file size.
)r/ r0 )r5 r> r r r r0 x s zLazyZipOverHTTP.truncatec C s dS )z
Return False.Fr r8 r r r writable s zLazyZipOverHTTP.writablec C s | j | S N)r/ __enter__r8 r r r rN s
zLazyZipOverHTTP.__enter__)excr" c G s | j j| S rM )r/ __exit__)r5 rO r r r rP s zLazyZipOverHTTP.__exit__c c s. | }zdV W | | n | | 0 dS )zyReturn a context manager keeping the position.
At the end of the block, seek back to original position.
N)rA rK )r5 posr r r _stay s zLazyZipOverHTTP._stayc
C s | j d }ttd|| jD ]h}| || | @ z t| W n tyT Y n0 W d qW d q 1 sz0 Y q dS )z1Check and download until the file is a valid ZIP.r? r N)r. reversedranger, rC rR r r )r5 endrE r r r r4 s
zLazyZipOverHTTP._check_zip)rE rU base_headersr" c C s8 | }d| d| |d< d|d< | jj| j|ddS )z:Return HTTP response to a range request from start to end.zbytes=-ZRangezno-cachez
Cache-ControlT)r& stream)copyr* r3 r+ )r5 rE rU rV r& r r r _stream_response s z LazyZipOverHTTP._stream_response)rE rU leftrightr" c
c s | j || | j|| }}t|g|dd }}t|g|dd }t||D ]&\}} ||krv||d fV | d }qX||kr||fV |g|g | j ||< | j||< dS )a/ Return an iterator of intervals to be fetched.
Args:
start (int): Start of needed interval
end (int): End of needed interval
left (int): Index of first overlapping downloaded data
right (int): Index after last overlapping downloaded data
Nr? r= )r1 r2 rB r@ zip)
r5 rE rU r[ r\ ZlsliceZrsliceijkr r r _merge s
zLazyZipOverHTTP._merge)rE rU r" c C s | ~ t| j|}t| j|}| ||||D ]D\}}| ||}| | | t || j
D ]}| j | qdq2W d n1 s0 Y dS )z-Download bytes from start to end inclusively.N)
rR r r2 r r1 ra rZ r rK r r, r/ write)r5 rE rU r[ r\ Zresponsechunkr r r rC s
zLazyZipOverHTTP._download)r= )r )N)&r r r __doc__r strr r- r6 propertyr9 r boolr: r; r< r( rD rH rK rA r r0 rL rN r rP r r
rR r4 r r r rZ r
ra rC r r r r r# ( sD
r# N)'rd __all__Zbisectr r
contextlibr Ztempfiler typingr r r
r r r
zipfiler r Zpip._vendor.packaging.utilsr Zpip._vendor.requests.modelsr r Zpip._internal.metadatar r r Z pip._internal.network.sessionr Zpip._internal.network.utilsr r r Exceptionr re r r# r r r r s