3
\ @ s6 d dl Z e jZG dd de jZG dd de jZdS ) Nc @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
BaseStreamc C s | j rtdd 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 stjdd S )NzFile not open for reading)readableioUnsupportedOperation)r r r r _check_can_read s zBaseStream._check_can_readc C s | j stjdd 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$tjdd 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 fddZ fddZdd Zd d
Zdd d
Zdd Z e
j fddZ 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 superclose)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 )NB)
memoryviewcastreadlen)r bZviewZ byte_viewdatar 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 qX n4| jj r| jjt}|st
dnd}| jj
||}|r&P q&W |sd| _| 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_inputEOFErrorr 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 jrFq8W | j| }ntdj ||| jk rv| j
n
|| j8 }x0|dkr| jt t j|}|sP |t |8 }qW | jS )Nr z Invalid value for whence: {})
r
SEEK_SETSEEK_CURr SEEK_ENDr r' DEFAULT_BUFFER_SIZEr formatr3 minr( )r offsetwhencer* 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 s