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 ej�Zddd�Zddd�Zdd� Zd S )�BZ2File�
BZ2Compressor�BZ2Decompressor�open�compress�
decompressz%Nadeem Vawda � )r N)�RLock)r r � � c @ s� e Zd Zdedfdd�Zdd� Zedd� �Zd d
� Zd d � Z d
d� Z
dd� Z d(dd�Z d)dd�Z
d*dd�Zdd� 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 kr�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�}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 )�w�wbr )�x�xbr )�a�abr �Invalid mode: %rT�read�writez6filename must be a str, bytes, file or PathLike object)Ztrailing_errorr ) r �_lock�_fp�_closefp� _MODE_CLOSED�_mode� _sentinel�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� 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& �flush�r5 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.peek�����c
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 |t�s$t|d�s 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 |t�s$t|d�s 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 �offset�whencer; 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|kr�t�||||�S |S d S ) N�trQ 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 �encoding�errors�newlineZbz_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 �append�eofr# Z unused_data�join)rX ZresultsZdecomp�resr; 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 � �
)