관리-도구
편집 파일: lazy_wheel.cpython-38.pyc
U &?�f� � @ 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 ed�dd�Z%G dd� d�Z&dS )zLazy 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 Zd ZdS )r N)�__name__� __module__� __qualname__� r r �N/opt/hc_python/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 dS )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 Zd ZdZefeeedd�dd�Ze ed�dd��Z e ed�d d ��Z e d�d d �Z dd�d d�Ze e d�dd��Zd4eed�dd�Ze d�dd�Zd5eeed�dd�Zed�dd �Zd6ee ed�d d �Ze d�d d �Zd d�d!d"�Zedd#�d$d%�Zee d& d�d'd(��Z dd�d)d*�Z e feee eef e!d+�d,d-�Z"eeeee e#eef ddf d.�d/d0�Z$eedd1�d2d3�Z%dS )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 |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! r& r+ r r r �__init__1 s zLazyZipOverHTTP.__init__)r" c C s dS )z!Opening mode, which is always rb.�rbr �r9 r r r �modeA s zLazyZipOverHTTP.modec C s | j jS )z Path to the underlying file.)r3 r r< r r r r F s zLazyZipOverHTTP.namec C s dS )z9Return whether random access is supported, which is True.Tr r< r r r �seekableK s zLazyZipOverHTTP.seekablec C s | j �� dS )zClose the file.N)r3 �closer< r r r r? O s zLazyZipOverHTTP.closec C s | j jS )zWhether the file is closed.)r3 �closedr< r r r r@ S s zLazyZipOverHTTP.closed���)�sizer" 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 � )�maxr0 �tellr2 �min� _downloadr3 �read)r9 rB � download_size�start�length�stopr r r rH X s zLazyZipOverHTTP.readc C s dS )z3Return whether the file is readable, which is True.Tr r< r r r �readablef s zLazyZipOverHTTP.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 rN rO r r r rP j s zLazyZipOverHTTP.seekc C s | j �� S )z Return the current position.)r3 rE r< r r r rE t s zLazyZipOverHTTP.tellc 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. )r3 r4 )r9 rB r r r r4 x s zLazyZipOverHTTP.truncatec C s dS )z Return False.Fr r< r r r �writable� s zLazyZipOverHTTP.writablec C s | j �� | S �N)r3 � __enter__r< r r r rS � s zLazyZipOverHTTP.__enter__)�excr" c G s | j j|� d S rR )r3 �__exit__)r9 rT r r r rU � s zLazyZipOverHTTP.__exit__)NNNc c s$ | � � }z dV W 5 | �|� X dS )zyReturn a context manager keeping the position. At the end of the block, seek back to original position. N)rE rP )r9 �posr r r �_stay� s zLazyZipOverHTTP._stayc C sv | j d }ttd|| j��D ]T}| �||� | �� �6 z t| � W n tk rV Y nX W 5 Q R � qrW 5 Q R X q dS )z1Check and download until the file is a valid ZIP.rC r N)r2 �reversed�ranger0 rG rW r r )r9 �endrJ r r r r8 � s zLazyZipOverHTTP._check_zip)rJ rZ � base_headersr" c C s8 |� � }d|� d|� �|d< d|d< | jj| j|dd�S )z:Return HTTP response to a range request from start to end.zbytes=�-�Rangezno-cachez Cache-ControlT)r' �stream)�copyr. r7 r/ )r9 rJ rZ r[ r' r r r �_stream_response� s z LazyZipOverHTTP._stream_response)rJ rZ �left�rightr" 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 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 rJ rZ ra rb �lslice�rslice�i�j�kr r r �_merge� s zLazyZipOverHTTP._merge)rJ rZ r" c C s� | � � �t t| j|�}t| j|�}| �||||�D ]D\}}| �||�}|�� | �|� t || j �D ]}| j � |� qdq2W 5 Q R X dS )z-Download bytes from start to end inclusively.N) rW r r6 r r5 ri r` r rP r r0 r3 �write)r9 rJ rZ ra rb �response�chunkr r r rG � s zLazyZipOverHTTP._download)rA )r )N)&r r r �__doc__r �strr r1 r: �propertyr= r �boolr> r? r@ r) rH rM rP rE r r4 rQ rS r rU r r rW r8 r r r r` r ri rG r r r r r# ( sR � � � � � r# N)'rm �__all__�bisectr r � contextlibr �tempfiler �typingr r r r r r �zipfiler r Zpip._vendor.packaging.utilsr Zpip._vendor.requests.modelsr r �pip._internal.metadatar r r � pip._internal.network.sessionr �pip._internal.network.utilsr r r � Exceptionr rn r r# r r r r �<module> s