3
\� � @ s6 d dl Z e jZG dd� de j�ZG dd� de j�ZdS )� Nc @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�
BaseStreamc C s | j rtd��d S )Nz I/O operation on closed file)�closed�
ValueError)�self� r �$/usr/lib64/python3.6/_compression.py�_check_not_closed s z BaseStream._check_not_closedc C s | j � stjd��d S )NzFile not open for reading)�readable�io�UnsupportedOperation)r r r r �_check_can_read s zBaseStream._check_can_readc C s | j � stjd��d S )NzFile not open for writing)�writabler
r )r r r r �_check_can_write s zBaseStream._check_can_writec C s( | j � stjd��| j� s$tjd��d S )Nz3Seeking is only supported on files open for readingz3The underlying file object does not support seeking)r r
r �seekable)r r r r �_check_can_seek s
zBaseStream._check_can_seekN)�__name__�
__module__� __qualname__r r r r r r r r r s r c sh e Zd Zdd� Zf fdd�Z� fdd�Zdd� Zd d
� Zdd d
�Zdd� Z e
j fdd�Z dd� Z
� ZS )�DecompressReaderc C s dS )NTr )r r r r r $ s zDecompressReader.readablec K s> || _ d| _d| _d| _|| _|| _| jf | j�| _|| _d S )NFr � ���)�_fp�_eof�_pos�_size�_decomp_factory� _decomp_args�
_decompressor�_trailing_error)r �fpZdecomp_factoryZtrailing_errorZ decomp_argsr r r �__init__' s zDecompressReader.__init__c s d | _ t� j� S )N)r �super�close)r )� __class__r r r" ; s zDecompressReader.closec C s
| j j� S )N)r r )r r r r r ? s zDecompressReader.seekablec C sP t |��:}|jd��$}| jt|��}||d t|��< W d Q R X W d Q R X t|�S )N�B)�
memoryview�cast�read�len)r �bZviewZ byte_view�datar r r �readintoB s $zDecompressReader.readintor c C s� |dk r| j � S | s | jr dS d }x�| jjr�| jjp@| jjt�}|sHP | jf | j �| _y| jj
||�}W q� | j k
r� P Y q�X n4| jj r�| jjt�}|s�t
d��nd}| jj
||�}|r&P q&W |s�d| _| j| _dS | jt|�7 _|S )Nr � zACompressed file ended before the end-of-stream marker was reachedT)�readallr r �eofZ unused_datar r' � BUFFER_SIZEr r Z
decompressr Z needs_input�EOFErrorr r r( )r �sizer* Zrawblockr r r r' H s>
zDecompressReader.readc C s, | j jd� d| _d| _| jf | j�| _d S )Nr F)r �seekr r r r r )r r r r �_rewindr s zDecompressReader._rewindc C s� |t jkr nV|t jkr"| j| }n@|t jkrT| jdk rHx| jt j�rFq8W | j| }ntdj |���|| jk rv| j
� n
|| j8 }x0|dkr�| jt t j|��}|s�P |t |�8 }q�W | jS )Nr z Invalid value for whence: {})
r
�SEEK_SET�SEEK_CURr �SEEK_ENDr r' �DEFAULT_BUFFER_SIZEr �formatr3 �minr( )r �offset�whencer* r r r r2 x s&
zDecompressReader.seekc C s | j S )N)r )r r r r �tell� s zDecompressReader.tellr )r )r r r r r r" r r+ r' r3 r
r4 r2 r< �
__classcell__r r )r# r r ! s
* r )r
r7 r/ �BufferedIOBaser � RawIOBaser r r r r �<module> s