Yf @ sQ d Z d d l Z e j Z Gd d d e j Z Gd d d e j Z d S)z7Internal classes used by the gzip, lzma and bz2 modules Nc @ sF e Z d Z d Z d d Z d d Z d d Z d d Z d
S)
BaseStreamz Mode-checking helper functions.c C s | j r t d d S)Nz I/O operation on closed file)closed
ValueError)self r 1/opt/alt/python35/lib64/python3.5/_compression.py_check_not_closed s z BaseStream._check_not_closedc C s | j s t j d d S)NzFile not open for reading)readableioUnsupportedOperation)r r r r _check_can_read s zBaseStream._check_can_readc C s | j s t j d d S)NzFile not open for writing)writabler
r )r r r r _check_can_write s zBaseStream._check_can_writec C s: | j s t j d | j s6 t j d 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____doc__r r r r r r r r r s
r c s e Z d Z d Z d d Z f d d Z f d d Z d d Z d
d Z d d
d Z d d Z
e j d d Z
d d Z S)DecompressReaderz5Adapts the decompressor API to a RawIOBase reader APIc C s d S)NTr )r r r r r $ s zDecompressReader.readablec K sX | | _ d | _ d | _ d | _ | | _ | | _ | j | 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 sd t | K } | j d 2 } | j t | } | | d t | <Wd QRXWd QRXt | S)NB)
memoryviewcastreadlen)r bZviewZ byte_viewdatar r r readintoB s !$zDecompressReader.readintor c C s? | d k r | j S| s& | j r* d Sd } x | j j r | j j pZ | j j t } | sd P| j | j | _ y | j j
| | } Wq | j k
r PYq XnN | j j r | j j t } | s t
d n d } | j j
| | } | r3 Pq3 W| s&d | _ | j | _ d S| j t | 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 j d d | _ d | _ | j | j | _ d S)Nr F)r seekr r r r r )r r r r _rewindr s zDecompressReader._rewindc C s | t j k r n{ | t j k r1 | j | } n\ | t j k rx | j d k rh x | j t j rg qR W| j | } n t d j | | | j k r | j
n
| | j 8} xB | d k r | j t t j | } | s P| t | 8} q W| j S)Nr z Invalid value for whence: {})
r
SEEK_SETSEEK_CURr SEEK_ENDr r( DEFAULT_BUFFER_SIZEr formatr4 minr) )r offsetwhencer+ r r r r3 x s&
zDecompressReader.seekc C s | j S)z!Return the current file position.)r )r r r r tell s zDecompressReader.tellr )r r r r r r! r# r r, r( r4 r
r5 r3 r= r r )r$ r r ! s * r )r r
r8 r0 BufferedIOBaser RawIOBaser r r r r <module> s