U
e5d1 @ s d dddddgZ dZddlmZ dd lZdd lZdd lZdd lZdd
l m
Z
dd l m Z m
Z
dZd Zd
Ze ZG dd d ejZdddZdddZdd Zd S )BZ2File
BZ2CompressorBZ2Decompressoropencompress
decompressz%Nadeem Vawda )r N)RLock)r r c @ s e Zd ZdedfddZdd Zedd Zd d
Zd d Z d
d Z
dd Z d(ddZ d)ddZ
d*ddZdd Zd+dd Zd,d d Zd d Zd!d" Zejfd#d$Zd%d& Zd'S )-r r c C sT t | _d | _d| _t| _|tk r2tjdt dd d| krFdksPn t
d|dkrbd }t }nb|d
kr~d }t }t
|| _nF|d krd
}t }t
|| _n*|dkrd}t }t
|| _nt
d|f t|tttjfrt||| _d| _|| _n.t|dst|dr|| _|| _ntd| jt krJtj| jttd}t || _ nd| _ d S )NFzGUse of 'buffering' argument is deprecated and ignored since Python 3.0. )
stacklevelr r z%compresslevel must be between 1 and 9) r rbr )wwbr )xxbr )aabr Invalid mode: %rTreadwritez6filename must be a str, bytes, file or PathLike object)Ztrailing_errorr ) r _lock_fp_closefp _MODE_CLOSED_mode _sentinelwarningswarnDeprecationWarning
ValueError
_MODE_READ _MODE_WRITEr _compressor
isinstancestrbytesosPathLike
_builtin_openhasattr TypeError _compressionZDecompressReaderr OSErrorioBufferedReader_buffer_pos)selffilenamemode buffering
compresslevelZ mode_coderaw r; /usr/lib64/python3.8/bz2.py__init__) sT zBZ2File.__init__c C s | j | jtkr W 5 Q R d S z<| jtkr8| j n"| jtkrZ| j | j
d | _
W 5 z| jrp| j W 5 d | _d| _t| _d | _X X W 5 Q R X d S )NF) r r r r r r3 closer$ r% r r& flushr5 r; r; r< r> p s
z
BZ2File.closec C s
| j tkS N)r r r@ r; r; r< closed s zBZ2File.closedc C s | | j S rA )_check_not_closedr filenor@ r; r; r< rD s zBZ2File.filenoc C s | o| j S rA )readabler3 seekabler@ r; r; r< rF s zBZ2File.seekablec C s | | jtkS rA )rC r r$ r@ r; r; r< rE s zBZ2File.readablec C s | | jtkS rA )rC r r% r@ r; r; r< writable s zBZ2File.writabler c
C s2 | j " | | j|W 5 Q R S Q R X d S rA )r _check_can_readr3 peek)r5 nr; r; r< rI s z BZ2File.peekc
C s2 | j " | | j|W 5 Q R S Q R X d S rA )r rH r3 r r5 sizer; r; r< r s z BZ2File.readc
C s@ | j 0 | |dk r tj}| j|W 5 Q R S Q R X d S )Nr )r rH r1 DEFAULT_BUFFER_SIZEr3 read1rL r; r; r< rO s
z
BZ2File.read1c
C s2 | j " | | j|W 5 Q R S Q R X d S rA )r rH r3 readinto)r5 br; r; r< rP s zBZ2File.readintoc
C sV t |ts$t|ds td| }| j" | | j|W 5 Q R S Q R X d S N __index__zInteger argument expected) r' intr- r. rS r rH r3 readlinerL r; r; r< rU s
zBZ2File.readlinec
C sV t |ts$t|ds td| }| j" | | j|W 5 Q R S Q R X d S rR ) r' rT r- r. rS r rH r3 readlinesrL r; r; r< rV s
zBZ2File.readlinesc
C sX | j H | | j|}| j| | jt|7 _t|W 5 Q R S Q R X d S rA )r Z_check_can_writer& r r r r4 len)r5 dataZ
compressedr; r; r< r s z
BZ2File.writec
C s, | j tj| |W 5 Q R S Q R X d S rA )r r/
BaseStream
writelines)r5 seqr; r; r< rZ s zBZ2File.writelinesc
C s4 | j $ | | j||W 5 Q R S Q R X d S rA )r Z_check_can_seekr3 seek)r5 offsetwhencer; r; r< r\ s z BZ2File.seekc
C sL | j < | | jtkr0| j W 5 Q R S | jW 5 Q R S Q R X d S rA )r rC r r$ r3 tellr4 r@ r; r; r< r_ s
z BZ2File.tellN)r )rK )rK )rK )rK )__name__
__module__ __qualname__r r= r> propertyrB rD rF rE rG rI r rO rP rU rV r rZ r1 SEEK_SETr\ r_ r; r; r; r< r s$ G
r r c C s d|kr d|krPt d|f n0|d k r0t d|d k r@t d|d k rPt d|dd}t| ||d}d|krt||||S |S d S ) NtrQ r z0Argument 'encoding' not supported in binary modez.Argument 'errors' not supported in binary modez/Argument 'newline' not supported in binary moder )r9 )r# replacer r1
TextIOWrapper)r6 r7 r9 encodingerrorsnewlineZbz_modeZ binary_filer; r; r< r ! s c C s t |}|| | S rA )r r r? )rX r9 compr; r; r< r J s c C sh g }| r^t }z|| }W n tk
r< |r6Y q^n Y nX || |jsVtd|j} qd|S )NzACompressed data ended before the end-of-stream marker was reached )r r r0 appendeofr# Z unused_datajoin)rX ZresultsZdecompresr; r; r< r U s
)r r NNN)r )__all__
__author__builtinsr r, r1 r* r r/ Z threadingr Z_bz2r r r r$ r% objectr rY r r r r; r; r; r< s4
)