o
9j1 @ sB d Z ddlZddlZejZG dd dejZG dd dejZdS )z7Internal classes used by the gzip, lzma and bz2 modules Nc @ s0 e Zd ZdZdd Zdd Zdd Zdd Zd
S )
BaseStreamz Mode-checking helper functions.c C s | j rtdd S )Nz I/O operation on closed file)closed
ValueErrorself r #/usr/lib/python3.10/_compression.py_check_not_closed s z BaseStream._check_not_closedc C | s tdd S )NzFile not open for reading)readableioUnsupportedOperationr r r r _check_can_read
zBaseStream._check_can_readc C r
)NzFile not open for writing)writabler r
r r r r _check_can_write r zBaseStream._check_can_writec C s( | s td| stdd 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
zBaseStream._check_can_seekN)__name__
__module__ __qualname____doc__r r r r r r r r r s r c sr e Zd ZdZdd ZdddZ fddZd d
Zd d ZdddZ dd Z
dd Z e j
fddZdd Z ZS )DecompressReaderz5Adapts the decompressor API to a RawIOBase reader APIc C s dS )NTr r r r r r $ s zDecompressReader.readabler c K sB || _ d| _d| _d| _|| _|| _| jdi | j| _|| _d S )NFr r )_fp_eof_pos_size_decomp_factory _decomp_args
_decompressor_trailing_error)r fpdecomp_factorytrailing_error decomp_argsr r r __init__' s
zDecompressReader.__init__c s d | _ t S N)r supercloser __class__r r r) ; s
zDecompressReader.closec C s
| j S r' )r r r r r r r ? s
zDecompressReader.seekablec C s t |9}|d}| t|}||d t|< W d n1 s$w Y W d t|S W d t|S 1 s@w Y t|S )NB)
memoryviewcastreadlen)r bview byte_viewdatar r r readintoB s ( zDecompressReader.readintor c C s |dk r| S |r
| jrdS d } | jjrA| jjp | jt}|s#n<| jdi | j | _z | j
||}W n$ | j y@ Y n w | jj rR| jt}|sQt
dnd}| j
||}|r^nq|sjd| _| j| _dS | jt|7 _|S )Nr TzACompressed file ended before the end-of-stream marker was reachedr )readallr r eof unused_datar r/ BUFFER_SIZEr r
decompressr! needs_inputEOFErrorr r r0 )r sizer4 rawblockr r r r/ H sH
zDecompressReader.readc C s8 g }| tj }r|| | tj }s
d|S )Nr6 )r/ sysmaxsizeappendjoin)r chunksr4 r r r r7 q s
zDecompressReader.readallc C s0 | j d d| _d| _| jdi | j| _d S )Nr Fr )r seekr r r r r r r r r _rewind| s zDecompressReader._rewindc C s |t jkrn/|t jkr| j| }n$|t jkr.| jdk r(| t jr( | t js!| j| }ntd ||| jk r?|
n|| j8 }|dkra| t t j|}|sW | jS |t |8 }|dksH| jS )Nr z Invalid value for whence: {})
r SEEK_SETSEEK_CURr SEEK_ENDr r/ DEFAULT_BUFFER_SIZEr formatrF minr0 )r offsetwhencer4 r r r rE s,
zDecompressReader.seekc C s | j S )z!Return the current file position.)r r r r r tell s zDecompressReader.tell)r )r )r r r r r r& r) r r5 r/ r7 rF r rG rE rO
__classcell__r r r* r r ! s
) r ) r r r@ rJ r: BufferedIOBaser RawIOBaser r r r r