nd$dZddlmZddlZddlZddlmZmZddlm Z ddl m Z ddl m Z mZddlmZmZdd lmZdd lmZmZejeZd d iZGd deZGddZGddeZdS)zLazy ZIP over HTTP) annotationsN) bisect_left bisect_right)contextmanager)NamedTemporaryFile)AnyIterator) BadZipfileZipFile)Session)CONTENT_CHUNK_SIZEResponsezAccept-EncodingidentityceZdZdS)HTTPRangeRequestUnsupportedN)__name__ __module__ __qualname__Blib/python3.11/site-packages/conda_package_streaming/lazy_wheel.pyrrsDrrceZdZdZefd3d Zed4d Zed4d Zd5dZ d6dZ ed5dZ d7d8dZ d5dZ d9d:dZd;dZddZd?d#Zed@d%Zd6d&ZefdAd-ZdBd1ZdCd2ZdS)DLazyZipOverHTTPaFile-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. urlstrsessionr chunk_sizeintreturnNonec@d|_|||c|_|_|_|dt }|jdkrtdt|j d dd|_ t|_ ||j |5t|j d }t!|d r|t#|jksJ||j |z ||jD]}|j | dddn #1swxYwY|j |z g|_|j d z g|_dS) Nr)startendzrange request is not supportedz content-range/zcontent-lengthcontent)_request_count_session_url _chunk_size_stream_responser status_coderrheaders partition_lengthr_filetruncate_stayhasattrlenr(seek iter_contentwrite_left_right)selfrrrtailcontent_lengthchunks r__init__zLazyZipOverHTTP.__init__&s 5!?@@NtY'' ;%T\):)::::: IIdl^3 4 4 4**4+;<< ( (   '''' (  ( ( ( ( ( ( ( ( ( ( ( ( ( ( ("&!> ? "&,"2!3 s BE33E7:E7cdS)z!Opening mode, which is always rb.rbrr=s rmodezLazyZipOverHTTP.modeJs trc|jjS)zPath to the underlying file.)r3namerDs rrGzLazyZipOverHTTP.nameOszrboolcdS)z9Return whether random access is supported, which is True.TrrDs rseekablezLazyZipOverHTTP.seekableTtrc8|jdS)zClose the file.N)r3closerDs rrMzLazyZipOverHTTP.closeXs rc|jjS)zWhether the file is closed.)r3closedrDs rrOzLazyZipOverHTTP.closed\sz  rr'sizebytesc|}||j}}|dkr|nt||z|}td||z }|||dz |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. rr))tellr2minmax _downloadr3read)r=rP download_sizer#lengthstops rrWzLazyZipOverHTTP.readas|  T\vvvs5=+@&'I'IAtm+,, udQh'''zt$$$rcdS)z3Return whether the file is readable, which is True.TrrDs rreadablezLazyZipOverHTTP.readableprKrroffsetwhencec8|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. )r3r8)r=r]r^s rr8zLazyZipOverHTTP.seektszvv...rc4|jS)zReturn the current position.)r3rSrDs rrSzLazyZipOverHTTP.tell~sz   rN int | Nonec6|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. )r3r4)r=rPs rr4zLazyZipOverHTTP.truncatesz""4(((rcdS)z Return False.FrrDs rwritablezLazyZipOverHTTP.writablesurc8|j|SN)r3 __enter__rDs rrgzLazyZipOverHTTP.__enter__s  rexcr bool | Nonec |jj|Srf)r3__exit__)r=rhs rrkzLazyZipOverHTTP.__exit__s"tz"C((rIterator[None]c#K|} dV||dS#||wxYw)zyReturn a context manager keeping the position. At the end of the block, seek back to original position. N)rSr8)r=poss rr5zLazyZipOverHTTP._staysJ iikk  EEE IIcNNNNNDIIcNNNNs 3A cJ|jdz }ttd||jD]t}||||5 t | ddddS#t$rYnwxYw dddn #1swxYwYudS)z1Check and download until the file is a valid ZIP.r)rN)r2reversedranger-rVr5r r )r=r$r#s r _check_zipzLazyZipOverHTTP._check_zipslQeAsD,<==>>  E NN5# & & &  DMMM          "D                  s6BA:+B: BBBBB B r# int | strr$ base_headersdict[str, str]rc|}d|d||d<td|dd|d<|xjdz c_|j|j|d }||S) zReturn HTTP response to a range request from start to end. :param start: if "", request ``end` bytes from end of file.zbytes=-Rangez%szno-cachez Cache-Controlr)T)r0stream)copylogdebugr*r+getr,raise_for_status)r=r#r$rtr0responses rr.z LazyZipOverHTTP._stream_responses ##%%1E11C11 $()))#-  q =$$TY$MM!!###rleftrightIterator[tuple[int, int]]c#XK|j|||j||}}t|g|ddzx}}t|g|ddz}t ||D]\}} ||kr ||dz fV| dz}||kr||fV|g|gc|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')r;r<rTrUzip) r=r#r$rrlslicersliceijks r_mergezLazyZipOverHTTP._mergesDJ/T%Z1H&!*,---E3%&+%&&''  DAq1uuQhAAA 88S&LLL;@'C57 4: DJ 7 7 7rc|5t|j|}t|j|}|||||D]g\}}|||}||||j D]}|j |h ddddS#1swxYwYdS)z-Download bytes from start to end inclusively.N) r5rr<rr;rr.r8r9r-r3r:)r=r#r$rrrr@s rrVzLazyZipOverHTTP._downloads% ZZ\\ , ,t{E22D S11E"kk%dEBB , , s00<< %   %2243CDD,,EJ$$U++++, , , , , , , , , , , , , , , , , , , ,sB+CCC)rrrr rrrr )rr)rrH)rr )r')rPrrrQ)r)r]rr^rrr)rrrf)rPrarr)rr)rhrrri)rrl)r#rsr$rrtrurr) r#rr$rrrrrrr)r#rr$rrr )rrr__doc__r rApropertyrErGrJrMrOrWr\r8rSr4rdrgrkrr5rrHEADERSr.rrVrrrrrs=O"4"4"4"4"4HXX!!!X! % % % % %/////!!!!)))))))))   ^      JQ IIII, , , , , , ,rrceZdZdZdS) LazyCondac0d|d}|5tj|}|}t |D]\}}|j|kr|j} ||dzj}n#t$r |j}YnwxYw| || ||z t d|j|nt dddddS#1swxYwYdS)a  Conda fork specific. Prefetch the `.info` range from the remote archive. Reduces number of Range requests to 2 or 3 (1 or 2 for the directory, 1 for the file). conda_file_id: name of .conda without path or `.conda` extension zinfo-z.tar.zstr)zprefetch %s-%szno zip prefetchN) r5zipfiler infolist enumeratefilename header_offset IndexError start_dirr8rWr{r|) r= conda_file_id target_filezfrrinfor#r$s rprefetchzLazyConda.prefetchs~6m555 ZZ\\ - -&&B{{}}H$X.. - -4=K//!.E+&q1uo; &+++ l+IIe$$$IIcEk***II(* E)0, +,,,7 - - - - - - - - - - - - - - - - - -s7AD +A<;D <B D BA.D  DDN)rrrrrrrrrs#$-$-$-$-$-rr)r __future__rloggingrbisectrr contextlibrtempfilertypingrr r r requestsr requests.modelsr r getLoggerrr{r Exceptionrrrrrrrst"""""",,,,,,,,%%%%%%'''''' ''''''''88888888 g!! j )     )   D,D,D,D,D,D,D,D,N%-%-%-%-%-%-%-%-%-%-r