a
    "h                     @   s6   d dl Z e jZG dd de jZG dd de jZdS )    Nc                   @   s,   e Zd Zdd Zdd Zdd Zdd Zd	S )

BaseStreamc                 C   s   | j rtdd S )NzI/O operation on closed file)closed
ValueErrorself r   4/opt/python-3.9.21/usr/lib/python3.9/_compression.py_check_not_closed   s    zBaseStream._check_not_closedc                 C   s   |   stdd S )NzFile not open for reading)readableioUnsupportedOperationr   r   r   r   _check_can_read   s    zBaseStream._check_can_readc                 C   s   |   stdd S )NzFile not open for writing)writabler   r   r   r   r   r   _check_can_write   s    zBaseStream._check_can_writec                 C   s(   |   std|  s$tdd S )Nz3Seeking is only supported on files open for readingz3The underlying file object does not support seeking)r
   r   r   seekabler   r   r   r   _check_can_seek   s    
zBaseStream._check_can_seekN)__name__
__module____qualname__r	   r   r   r   r   r   r   r   r   	   s   r   c                       sf   e Zd Zdd ZdddZ fddZdd	 Zd
d ZdddZdd Z	e
jfddZdd Z  ZS )DecompressReaderc                 C   s   dS )NTr   r   r   r   r   r
   $   s    zDecompressReader.readabler   c                 K   sB   || _ d| _d| _d| _|| _|| _| jf i | j| _|| _d S )NFr   )_fp_eof_pos_size_decomp_factory_decomp_args_decompressor_trailing_error)r   fpdecomp_factorytrailing_errordecomp_argsr   r   r   __init__'   s    zDecompressReader.__init__c                    s   d | _ t  S N)r   supercloser   	__class__r   r   r&   ;   s    zDecompressReader.closec                 C   s
   | j  S r$   )r   r   r   r   r   r   r   ?   s    zDecompressReader.seekablec              	   C   sx   t |X}|d.}| t|}||d t|< W d    n1 sH0    Y  W d    n1 sf0    Y  t|S )NB)
memoryviewcastreadlen)r   bview	byte_viewdatar   r   r   readintoB   s    LzDecompressReader.readintor   c                 C   s   |dk r|   S |r| jrdS d }| jjr| jjp<| jt}|sDq| jf i | j	| _z| j
||}W q | jy   Y qY q0 n4| jjr| jt}|stdnd}| j
||}|r"qq"|sd| _| j| _dS |  jt|7  _|S )Nr       zACompressed file ended before the end-of-stream marker was reachedT)readallr   r   eofunused_datar   r,   BUFFER_SIZEr   r   
decompressr   needs_inputEOFErrorr   r   r-   )r   sizer1   rawblockr   r   r   r,   H   s@    


zDecompressReader.readc                 C   s0   | j d d| _d| _| jf i | j| _d S )Nr   F)r   seekr   r   r   r   r   r   r   r   r   _rewindr   s    zDecompressReader._rewindc                 C   s   |t jkrnR|t jkr"| j| }n<|t jkrP| jdk rD| t jrDq6| j| }ntd	||| jk rr| 
  n
|| j8 }|dkr| tt j|}|sq|t|8 }q|| jS )Nr   zInvalid value for whence: {})r   SEEK_SETSEEK_CURr   SEEK_ENDr   r,   DEFAULT_BUFFER_SIZEr   formatr>   minr-   )r   offsetwhencer1   r   r   r   r=   x   s&    






zDecompressReader.seekc                 C   s   | j S r$   )r   r   r   r   r   tell   s    zDecompressReader.tell)r   )r   )r   r   r   r
   r#   r&   r   r2   r,   r>   r   r?   r=   rG   __classcell__r   r   r'   r   r   !   s   

*r   )r   rB   r7   BufferedIOBaser   	RawIOBaser   r   r   r   r   <module>   s   