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 ej�Zddd�Zddd�Zdd� Zd S )�BZ2File�
BZ2Compressor�BZ2Decompressor�open�compress�
decompressz%Nadeem Vawda <nadeem.vawda@gmail.com>� )r N)�RLock)r r � � c @ s� e Zd Zd(dd�Zdd� Zedd � �Zd
d � Zd d
� Zdd� Z dd� Z
d)dd�Z d+dd�Z d-dd�Z
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|dkr�d }t }t |�| _
n*|dkr�d}t }t |�| _
nt d|f ��t|tttjf�� r�t||�| _d| _|| _n.t|d��st|d��r|| _|| _ntd��| jt
k�rJtj| jttd�}tj|�| _ nd| _ d S )NFz)Use of 'buffering' argument is deprecatedr r z%compresslevel must be between 1 and 9� r �rb�w�wb�x�xb�a�abzInvalid mode: %rT�read�writez6filename 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�_mode�warnings�warn�DeprecationWarning�
ValueError�
_MODE_READ� _MODE_WRITEr � _compressor�
isinstance�str�bytes�os�PathLike�
_builtin_open�hasattr� TypeError� _compressionZDecompressReaderr �OSError�io�BufferedReader�_buffer�_pos)�self�filename�mode� buffering�
compresslevelZ mode_code�raw� 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 |t�s$t|d�s 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 |t�s$t|d�s 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 �offset�whencer7 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|kr�tj||||�S |S d S ) N�trI 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 �encoding�errors�newlineZbz_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, �append�eofr Z unused_data�join)rO ZresultsZdecomp�resr7 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,
(