a
R�f�0 � @ s� g d �Z dZddlmZ ddlZddlZddlZddlm Z ddl
m Z m Z dZ
dZdZG d d
� d
ej�Zdd
d�Zddd�Zdd� ZdS ))�BZ2File�
BZ2Compressor�BZ2Decompressor�open�compress�
decompressz%Nadeem Vawda <nadeem.vawda@gmail.com>� )r N)�RLock)r r � � c @ s� e Zd Zd)dd�dd�Zdd� Zedd � �Zd
d � Zd d
� Zdd� Z dd� Z
d*dd�Z d+dd�Z d,dd�Z
dd� Zd-d d �Zd.d d �Zd d!� Zd"d#� Zejfd$d%�Zd&d'� Zd(S )/r �r� ��
compresslevelc C s8 t � | _d | _d| _t| _d| kr.dks8n td��|dv rJd}t}nb|dv rfd}t}t |�| _
nF|d v r�d
}t}t |�| _
n*|d v r�d }t}t |�| _
ntd
|f ��t |t t
tjf�r�t||�| _d| _|| _n*t|d�s�t|d�r�|| _|| _ntd��| jtk�r.tj| jttd�}t�|�| _nd| _d S )NFr 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�
ValueError�
_MODE_READ� _MODE_WRITEr � _compressor�
isinstance�str�bytes�os�PathLike�
_builtin_open�hasattr� TypeError� _compressionZDecompressReaderr �OSError�io�BufferedReader�_buffer�_pos)�self�filename�moder Z mode_code�raw� r5 �(/opt/alt/python39/lib64/python3.9/bz2.py�__init__& sH �zBZ2File.__init__c C s | j �� | jtkr"W d � d S z�| jtkr:| j�� n"| jtkr\| j�| j �
� � d | _ W z,| j rp| j�� W d | _d| _ t| _d | _q�d | _d| _ t| _d | _0 nJz,| j r�| j�� W d | _d| _ t| _d | _nd | _d| _ t| _d | _0 0 W d � n1 �s0 Y d S )NF) r r r r r/ �closer! r r r" �flushr �r1 r5 r5 r6 r8 e s>
�
� �z
BZ2File.closec C s
| j tkS �N)r r r: r5 r5 r6 �closed~ s zBZ2File.closedc C s | � � | j�� S r; )�_check_not_closedr �filenor: r5 r5 r6 r> � s zBZ2File.filenoc C s | � � o| j�� S r; )�readabler/ �seekabler: r5 r5 r6 r@ � s zBZ2File.seekablec C s | � � | jtkS r; )r= r r r: r5 r5 r6 r? � s zBZ2File.readablec C s | � � | jtkS r; )r= r r! r: r5 r5 r6 �writable� s zBZ2File.writabler c C s> | j �$ | �� | j�|�W d � S 1 s00 Y d S r; )r �_check_can_readr/ �peek)r1 �nr5 r5 r6 rC � s z BZ2File.peek���c C s> | j �$ | �� | j�|�W d � S 1 s00 Y d S r; )r rB r/ r �r1 �sizer5 r5 r6 r � s z BZ2File.readc C sL | j �2 | �� |dk r tj}| j�|�W d � S 1 s>0 Y d S )Nr )r rB r- �DEFAULT_BUFFER_SIZEr/ �read1rF r5 r5 r6 rI � s
z
BZ2File.read1c C s> | j �$ | �� | j�|�W d � S 1 s00 Y d S r; )r rB r/ �readinto)r1 �br5 r5 r6 rJ � s zBZ2File.readintoc C sb t |t�s$t|d�s td��|�� }| j�$ | �� | j�|�W d � S 1 sT0 Y d S �N� __index__zInteger argument expected) r# �intr) r* rM r rB r/ �readlinerF r5 r5 r6 rO � s
zBZ2File.readlinec C sb t |t�s$t|d�s td��|�� }| j�$ | �� | j�|�W d � S 1 sT0 Y d S rL ) r# rN r) r* rM r rB r/ � readlinesrF r5 r5 r6 rP � s
zBZ2File.readlinesc C s� | j �h | �� t|ttf�r(t|�}nt|�}|j}| j� |�}| j
� |� | j |7 _ |W d � S 1 st0 Y d S r; )
r Z_check_can_writer# r% � bytearray�len�
memoryview�nbytesr" r r r r0 )r1 �dataZlengthZ
compressedr5 r5 r6 r � s
z
BZ2File.writec C s8 | j � tj�| |�W d � S 1 s*0 Y d S r; )r r+ �
BaseStream�
writelines)r1 �seqr5 r5 r6 rW � s zBZ2File.writelinesc C s@ | j �& | �� | j�||�W d � S 1 s20 Y d S r; )r Z_check_can_seekr/ �seek)r1 �offset�whencer5 r5 r6 rY s z BZ2File.seekc C sZ | j �@ | �� | jtkr2| j�� W d � S | jW d � S 1 sL0 Y d S r; )r r= r r r/ �tellr0 r: r5 r5 r6 r\ s
z BZ2File.tellN)r )r )rE )rE )rE )rE )�__name__�
__module__� __qualname__r7 r8 �propertyr< r> r@ r? rA rC r rI rJ rO rP r rW r- �SEEK_SETrY r\ r5 r5 r5 r6 r s$ ?
r r r c C s� d|v r d|v rPt d|f ��n0|d ur0t d��|d ur@t d��|d urPt d��|�dd�}t| ||d�}d|v r�t�||||�S |S d S ) N�trK r z0Argument 'encoding' not supported in binary modez.Argument 'errors' not supported in binary modez/Argument 'newline' not supported in binary moder r
)r �replacer r- �
TextIOWrapper)r2 r3 r �encoding�errors�newlineZbz_modeZ binary_filer5 r5 r6 r s r c C s t |�}|�| �|�� S r; )r r r9 )rU r �compr5 r5 r6 r G s r c C sf g }| r\t � }z|�| �}W n ty: |r4Y q\n� Y n0 |�|� |jsTtd��|j} qd�|�S )NzACompressed data ended before the end-of-stream marker was reached� )r r r, �append�eofr Z unused_data�join)rU ZresultsZdecomp�resr5 r5 r6 r R s
r )r r NNN)r )�__all__�
__author__�builtinsr r( r- r& r+ Z threadingr Z_bz2r r r r r! rV r r r r5 r5 r5 r6 �<module> s$ �
)