Edit File: lazy_wheel.cpython-37.pyc
B �j�c� � @ s� d Z ddgZddlmZmZ ddlmZ ddlmZ ddl m Z m Z ddl m Z dd lmZ dd lmZmZmZ dd lmZ dd lmZ er�dd lmZmZmZm Z m Z m Z ddl m Z ddl m!Z! ddl"m#Z# G dd� de$�Z%dd� Z&G dd� de'�Z(dS )zLazy ZIP over HTTP�HTTPRangeRequestUnsupported�dist_from_wheel_url� )� bisect_left� bisect_right)�contextmanager)�NamedTemporaryFile)� BadZipfile�ZipFile)�CONTENT_CHUNK_SIZE)�range)�HEADERS�raise_for_status�response_chunks)�MYPY_CHECK_RUNNING)�$pkg_resources_distribution_for_wheel)�Any�Dict�Iterator�List�Optional�Tuple)� Distribution)�Response)� PipSessionc @ s e Zd ZdS )r N)�__name__� __module__� __qualname__� r r ��/builddir/build/BUILDROOT/alt-python37-pip-20.2.4-5.el8.x86_64/opt/alt/python37/lib/python3.7/site-packages/pip/_internal/network/lazy_wheel.pyr s c C s, t ||��}t|�}t|| |j�S Q R X dS )a% Return a pkg_resources.Distribution 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 �name)r �url�session�wheel�zip_filer r r r " s c @ s� e Zd ZdZefdd�Zedd� �Zedd� �Zdd � Z d d � Z ed d � �Z d+dd�Z dd� Z d,dd�Zdd� Zd-dd�Zdd � Zd d � Zd d � Zed!d"� �Zd#d$� Zefd%d&�Zd'd(� Zd)d*� ZdS ).r 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. c C s� |j |td�}t|� |jdks$t�||| | _| _| _t|j d �| _ t � | _ | � | j � g | _g | _d|j �dd�kr�td��| �� d S )N)�headers�� zContent-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" � chunk_sizer) r r r �__init__= s zLazyZipOverHTTP.__init__c C s dS )z!Opening mode, which is always rb.�rbr )r7 r r r �modeL s zLazyZipOverHTTP.modec C s | j jS )z Path to the underlying file.)r1 r )r7 r r r r R s zLazyZipOverHTTP.namec C s dS )z9Return whether random access is supported, which is True.Tr )r7 r r r �seekableX s zLazyZipOverHTTP.seekablec C s | j �� dS )zClose the file.N)r1 �close)r7 r r r r= ] s zLazyZipOverHTTP.closec C s | j jS )zWhether the file is closed.)r1 �closed)r7 r r r r> b s zLazyZipOverHTTP.closed���c C s` t || j�}| �� | j }}|dk r(|n t|| |�}t d|| �}| �||d � | j�|�S )z�Read 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. �tellr0 �min� _downloadr1 �read)r7 �sizeZ download_size�start�length�stopr r r rE h s zLazyZipOverHTTP.readc C s dS )z3Return whether the file is readable, which is True.Tr )r7 r r r �readablew s zLazyZipOverHTTP.readabler 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. )r1 �seek)r7 �offset�whencer r r rK | s zLazyZipOverHTTP.seekc C s | j �� S )z Return the current possition.)r1 rB )r7 r r r rB � s zLazyZipOverHTTP.tellNc C s | j �|�S )z�Resize 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. )r1 r2 )r7 rF r r r r2 � s zLazyZipOverHTTP.truncatec C s dS )z Return False.Fr )r7 r r r �writable� s zLazyZipOverHTTP.writablec C s | j �� | S )N)r1 � __enter__)r7 r r r rO � s zLazyZipOverHTTP.__enter__c G s | j j|� S )N)r1 �__exit__)r7 �excr r r rP � s zLazyZipOverHTTP.__exit__c c s$ | � � }z dV W d| �|� X dS )zyReturn a context manager keeping the position. At the end of the block, seek back to original position. N)rB rK )r7 �posr r r �_stay� s zLazyZipOverHTTP._stayc C sn | j d }x^ttd|| j��D ]H}| �||� | �� �* y t| � W n tk rX Y nX P W dQ R X q W dS )z1Check and download until the file is a valid ZIP.r@ r N)r0 �reversedr r. rD rS r r )r7 �endrG r r r r6 � s zLazyZipOverHTTP._check_zipc C s4 |� � }d�||�|d< d|d< | jj| j|dd�S )z:Return HTTP response to a range request from start to end.z bytes={}-{}ZRangezno-cachez Cache-ControlT)r% �stream)�copy�formatr, r5 r- )r7 rG rU Z base_headersr% r r r �_stream_response� s z LazyZipOverHTTP._stream_responsec c s� | j ||� | j||� }}t|g|dd� � }}t|g|dd� �}x4t||�D ]&\}} ||krx||d fV | d }qZW ||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? )r3 r4 rC rA �zip) r7 rG rU �left�rightZlsliceZrslice�i�j�kr r r �_merge� s zLazyZipOverHTTP._mergec C s� | � � �| t| j|�}t| j|�}x\| �||||�D ]H\}}| �||�}|�� | �|� x t || j �D ]}| j � |� qhW q4W W dQ R X dS )z-Download bytes from start to end inclusively.N) rS r r4 r r3 r` rY r rK r r. r1 �write)r7 rG rU r[ r\ �response�chunkr r r rD � s zLazyZipOverHTTP._download)r? )r )N)r r r �__doc__r r9 �propertyr; r r< r= r> rE rJ rK rB r2 rN rO rP r rS r6 r rY r` rD r r r r r 4 s( r N))rd �__all__�bisectr r � contextlibr �tempfiler �zipfiler r Zpip._vendor.requests.modelsr �pip._vendor.six.movesr �pip._internal.network.utilsr r r �pip._internal.utils.typingr �pip._internal.utils.wheelr �typingr r r r r r Zpip._vendor.pkg_resourcesr r � pip._internal.network.sessionr � Exceptionr r �objectr r r r r �<module> s$