3
\0
@ s d dddddgZ dZddlmZ dd lZdd lZdd lZdd lZydd
l m
Z
W n e k
rp dd
l m
Z
Y nX dd l
mZmZ dZd Zd
ZG dd d ejZdddZdddZdd Zd S )BZ2File
BZ2CompressorBZ2Decompressoropencompress
decompressz%Nadeem Vawda <nadeem.vawda@gmail.com> )r N)RLock)r r c @ s e Zd Zd(ddZdd Zedd Zd
d Zd d
Zdd Z dd Z
d)ddZ d+ddZ d-ddZ
dd Zd/d d Zd1d d Zd d! Zd"d# Zejfd$d%Zd&d' ZdS )2r rN c C sT t | _d | _d| _t| _|d k r.tjdt d| ko@dkn sNt d|dkr`d}t
}nb|dkr|d
}t }t || _
nF|dkrd }t }t || _
n*|dkrd}t }t || _
nt d|f t|tttjf rt||| _d| _|| _n.t|dst|dr|| _|| _ntd| jt
krJtj| jttd}tj|| _ nd| _ d S )NFz)Use of 'buffering' argument is deprecatedr r z%compresslevel must be between 1 and 9 r rbwwbxxbaabzInvalid mode: %rTreadwritez6filename must be a str, bytes, file or PathLike object)Ztrailing_errorr )r
r r )r r )r r )r r ) r _lock_fp_closefp _MODE_CLOSED_modewarningswarnDeprecationWarning
ValueError
_MODE_READ _MODE_WRITEr _compressor
isinstancestrbytesosPathLike
_builtin_openhasattr TypeError _compressionZDecompressReaderr OSErrorioBufferedReader_buffer_pos)selffilenamemode buffering
compresslevelZ mode_coderaw r7 /usr/lib64/python3.6/bz2.py__init__+ sL
zBZ2File.__init__c C s | j | jtkrd S z<| jtkr.| jj n"| jtkrP| jj| j j
d | _ W d z| j rf| jj W d d | _d| _ t| _d | _X X W d Q R X d S )NF) r r r r r/ closer! r r r" flushr )r1 r7 r7 r8 r: p s
z
BZ2File.closec C s
| j tkS )N)r r )r1 r7 r7 r8 closed s zBZ2File.closedc C s | j | jj S )N)_check_not_closedr fileno)r1 r7 r7 r8 r> s zBZ2File.filenoc C s | j o| jj S )N)readabler/ seekable)r1 r7 r7 r8 r@ s zBZ2File.seekablec C s | j | jtkS )N)r= r r )r1 r7 r7 r8 r? s zBZ2File.readablec C s | j | jtkS )N)r= r r! )r1 r7 r7 r8 writable s zBZ2File.writabler c C s& | j | j | jj|S Q R X d S )N)r _check_can_readr/ peek)r1 nr7 r7 r8 rC s z BZ2File.peekr c C s& | j | j | jj|S Q R X d S )N)r rB r/ r )r1 sizer7 r7 r8 r s z BZ2File.readc C s4 | j $ | j |dk r tj}| jj|S Q R X d S )Nr )r rB r- DEFAULT_BUFFER_SIZEr/ read1)r1 rE r7 r7 r8 rG s
z
BZ2File.read1c C s& | j | j | jj|S Q R X d S )N)r rB r/ readinto)r1 br7 r7 r8 rH s zBZ2File.readintoc C sJ t |ts$t|ds td|j }| j | j | jj|S Q R X d S )N __index__zInteger argument expected) r# intr) r* rJ r rB r/ readline)r1 rE r7 r7 r8 rL s
zBZ2File.readlinec C sJ t |ts$t|ds td|j }| j | j | jj|S Q R X d S )NrJ zInteger argument expected) r# rK r) r* rJ r rB r/ readlines)r1 rE r7 r7 r8 rM s
zBZ2File.readlinesc C sL | j < | j | jj|}| jj| | jt|7 _t|S Q R X d S )N)r Z_check_can_writer" r r r r0 len)r1 dataZ
compressedr7 r7 r8 r s z
BZ2File.writec
C s | j tjj| |S Q R X d S )N)r r+
BaseStream
writelines)r1 seqr7 r7 r8 rQ s zBZ2File.writelinesc
C s( | j | j | jj||S Q R X d S )N)r Z_check_can_seekr/ seek)r1 offsetwhencer7 r7 r8 rS s z BZ2File.seekc C s4 | j $ | j | jtkr$| jj S | jS Q R X d S )N)r r= r r r/ tellr0 )r1 r7 r7 r8 rV s
z BZ2File.tell)r Nr )r )rW rW )rW rW )rW rW )rW )__name__
__module__ __qualname__r9 r: propertyr< r> r@ r? rA rC r rG rH rL rM r rQ r- SEEK_SETrS rV r7 r7 r7 r8 r s"
E
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|jdd}t| ||d}d|krtj||||S |S d S ) NtrI zInvalid mode: %rz0Argument 'encoding' not supported in binary modez.Argument 'errors' not supported in binary modez/Argument 'newline' not supported in binary moder
)r5 )r replacer r-
TextIOWrapper)r2 r3 r5 encodingerrorsnewlineZbz_modeZ binary_filer7 r7 r8 r ! s c C s t |}|j| |j S )N)r r r; )rO r5 compr7 r7 r8 r J s c C sj g }xZ| r^t }y|j| }W n tk
r< |r6P n Y nX |j| |jsVtd|j} qW dj|S )NzACompressed data ended before the end-of-stream marker was reached )r r r, appendeofr Z unused_datajoin)rO ZresultsZdecompresr7 r7 r8 r U s
)r r NNN)r )__all__
__author__builtinsr r( r- r& r r+ Z threadingr ImportErrorZdummy_threadingZ_bz2r r r r r! rP r r r r7 r7 r7 r8 <module> s,
(