U
    cUh                     @   s   d Z ddgZddlmZmZ ddlmZ ddlmZ ddl	m
Z
mZmZmZmZmZ ddlmZmZ dd	lmZ dd
lmZmZ ddlmZmZmZ ddlmZ ddlmZm Z m!Z! G dd de"Z#e$e$eedddZ%G dd dZ&dS )zLazy ZIP over HTTPHTTPRangeRequestUnsupporteddist_from_wheel_url    )bisect_leftbisect_right)contextmanager)NamedTemporaryFile)AnyDict	GeneratorListOptionalTuple)
BadZipFileZipFile)canonicalize_name)CONTENT_CHUNK_SIZEResponse)BaseDistributionMemoryWheelget_wheel_distribution)
PipSession)HEADERSraise_for_statusresponse_chunksc                   @   s   e Zd ZdS )r   N)__name__
__module____qualname__ r   r   `/opt/bot/vods_filmes/script/venv/lib/python3.8/site-packages/pip/_internal/network/lazy_wheel.pyr      s   )nameurlsessionreturnc              
   C   s<   t ||(}t|j|}t|t| W  5 Q R  S Q R X dS )a  Return a distribution object from the given wheel URL.

    This uses HTTP range requests to only fetch the portion 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!   zfwheelr   r   r   r      s    c                   @   s|  e Zd ZdZefeeeddddZe	edddZ
e	edd	d
ZedddZddddZe	edddZd4eedddZedddZd5eeedddZedddZd6ee edddZeddd Zd dd!d"Zedd#d$d%Zeed& dd'd(Zddd)d*Zefeee eef e!d+d,d-Z"eeeeee#eef ddf d.d/d0Z$eedd1d2d3Z%dS )7r#   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 |td}t| |jdks$t|||  | _| _| _t|j	d | _
t | _| | j
 g | _g | _d|j	ddkrtd|   d S )N)headers   zContent-LengthbyteszAccept-Rangesnonezrange request is not supported)headr   r   status_codeAssertionError_session_url_chunk_sizeintr'   _lengthr   _filetruncate_left_rightgetr   
_check_zip)selfr    r!   r&   r+   r   r   r   __init__1   s    zLazyZipOverHTTP.__init__)r"   c                 C   s   dS )z!Opening mode, which is always rb.rbr   r9   r   r   r   modeA   s    zLazyZipOverHTTP.modec                 C   s   | j jS )zPath to the underlying file.)r3   r   r<   r   r   r   r   F   s    zLazyZipOverHTTP.namec                 C   s   dS )z9Return whether random access is supported, which is True.Tr   r<   r   r   r   seekableK   s    zLazyZipOverHTTP.seekablec                 C   s   | j   dS )zClose the file.N)r3   closer<   r   r   r   r?   O   s    zLazyZipOverHTTP.closec                 C   s   | j jS )zWhether the file is closed.)r3   closedr<   r   r   r   r@   S   s    zLazyZipOverHTTP.closed)sizer"   c                 C   s`   t || j}|  | j }}|dk 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      )maxr0   tellr2   min	_downloadr3   read)r9   rB   Zdownload_sizestartlengthstopr   r   r   rH   X   s    zLazyZipOverHTTP.readc                 C   s   dS )z3Return whether the file is readable, which is True.Tr   r<   r   r   r   readablef   s    zLazyZipOverHTTP.readabler   )offsetwhencer"   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.
        )r3   seek)r9   rM   rN   r   r   r   rO   j   s    zLazyZipOverHTTP.seekc                 C   s
   | j  S )zReturn the current position.)r3   rE   r<   r   r   r   rE   t   s    zLazyZipOverHTTP.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.
        )r3   r4   )r9   rB   r   r   r   r4   x   s    zLazyZipOverHTTP.truncatec                 C   s   dS )zReturn False.Fr   r<   r   r   r   writable   s    zLazyZipOverHTTP.writablec                 C   s   | j   | S N)r3   	__enter__r<   r   r   r   rR      s    
zLazyZipOverHTTP.__enter__)excr"   c                 G   s   | j j|  d S rQ   )r3   __exit__)r9   rS   r   r   r   rT      s    zLazyZipOverHTTP.__exit__)NNNc              	   c   s$   |   }z
dV  W 5 | | X dS )zyReturn a context manager keeping the position.

        At the end of the block, seek back to original position.
        N)rE   rO   )r9   posr   r   r   _stay   s    
zLazyZipOverHTTP._stayc              
   C   sv   | j d }ttd|| jD ]T}| || |  6 zt|  W n tk
rV   Y nX W 5 Q R   qrW 5 Q R X qdS )z1Check and download until the file is a valid ZIP.rC   r   N)r2   reversedranger0   rG   rV   r   r   )r9   endrI   r   r   r   r8      s    

zLazyZipOverHTTP._check_zip)rI   rY   base_headersr"   c                 C   s8   |  }d| d| |d< d|d< | jj| j|ddS )z:Return HTTP response to a range request from start to end.zbytes=-Rangezno-cachezCache-ControlT)r'   stream)copyr.   r7   r/   )r9   rI   rY   rZ   r'   r   r   r   _stream_response   s    z LazyZipOverHTTP._stream_response)rI   rY   leftrightr"   c           
      c   s   | j || | j||  }}t|g|dd   }}t|g|dd  }t||D ]&\}}	||krv||d fV  |	d }qX||kr||fV  |g|g | j ||< | j||< dS )a/  Return a generator 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
        NrC   rA   )r5   r6   rF   rD   zip)
r9   rI   rY   r`   ra   ZlsliceZrsliceijkr   r   r   _merge   s    

zLazyZipOverHTTP._merge)rI   rY   r"   c              	   C   s   |   t t| j|}t| j|}| ||||D ]D\}}| ||}|  | | t	|| j
D ]}| j| qdq2W 5 Q R X dS )z-Download bytes from start to end inclusively.N)rV   r   r6   r   r5   rf   r_   r   rO   r   r0   r3   write)r9   rI   rY   r`   ra   responsechunkr   r   r   rG      s    

zLazyZipOverHTTP._download)rA   )r   )N)&r   r   r   __doc__r   strr   r1   r:   propertyr=   r   boolr>   r?   r@   r)   rH   rL   rO   rE   r   r4   rP   rR   r   rT   r   r
   rV   r8   r   r	   r   r_   r   rf   rG   r   r   r   r   r#   (   sR   	  

  
   r#   N)'rj   __all__bisectr   r   
contextlibr   tempfiler   typingr   r	   r
   r   r   r   zipfiler   r   pip._vendor.packaging.utilsr   Zpip._vendor.requests.modelsr   r   pip._internal.metadatar   r   r   pip._internal.network.sessionr   pip._internal.network.utilsr   r   r   	Exceptionr   rk   r   r#   r   r   r   r   <module>   s    