�
c��jG. � � � g d �Z dZddlmZ ddlZddlZddlZddlm Z m
Z
dZ dZ dZ
G d� d ej � Z d
d
�Zdd �Zd � Zy))�BZ2File�
BZ2Compressor�BZ2Decompressor�open�compress�
decompressz%Nadeem Vawda � )r N)r r � � c � � e Zd Z ddd�d�Zd� Zed� � Zd� Zd� Zd� Z d � Z
dd
�Z dd �Z dd �Z
d
� Zdd�Zdd�Zd� Zd� Zej( fd�Zd� Zy)r � ��
compresslevelc � � d | _ d| _ t | _ d|cxk rdk st d� � t d� �|dv r d}t
}ne|dv rd}t }t |� | _ nH|d v rd
}t }t |� | _ n+|d v rd }t }t |� | _ nt d
|��� �t |t t t j f� r t ||� | _ d| _ || _ n2t |d� s t |d� r|| _ || _ n t! d� �| j t
k( rEt# j$ | j t& t( �� }t+ j, |� | _ y d| _ y )NFr r z%compresslevel must be between 1 and 9)� �r�rbr )�w�wbr )�x�xbr )�a�abr �Invalid mode: T�read�writez6filename must be a str, bytes, file or PathLike object)�trailing_errorr )�_fp�_closefp� _MODE_CLOSED�_mode�
ValueError�
_MODE_READ� _MODE_WRITEr � _compressor�
isinstance�str�bytes�os�PathLike�
_builtin_open�hasattr� TypeError� _compression�DecompressReaderr �OSError�io�BufferedReader�_buffer�_pos)�self�filename�moder � mode_code�raws � /usr/lib64/python3.12/bz2.py�__init__zBZ2File.__init__% sc � � �" ��� ��
�!��
��]�'�a�'� � D�E� E� (� � D�E� E� �?� "��D�"�I�
�[�
��D�#�I� ,�]� ;�D� �
�[�
��D�#�I� ,�]� ;�D� �
�[�
��D�#�I� ,�]� ;�D� � �4� 9�:� :� �h��e�R�[�[� 9� :�$�X�t�4�D�H� �D�M�"�D�J�
�X�v�
&�'�(�G�*D� �D�H�"�D�J�� T�U� U� �:�:�� #� �/�/���� ��9�C� �,�,�S�1�D�L��D�I� c �� � | j t k( ry | j t k( r| j j � nM| j t
k( r:| j j | j j � � d | _ | j r| j j � d | _ d| _
t | _ d | _ y # d | _ d| _
t | _ d | _ w xY w# | j r| j j � d | _ d| _
t | _ d | _ w # d | _ d| _
t | _ d | _ w xY wxY w)NF) r r r" r2 �closer# r r r$ �flushr �r4 s r9 r= z
BZ2File.closea s � � �
�:�:� � %� � $��z�z�Z�'�� � �"�"�$����{�*������t� /� /� 5� 5� 7�8�#'�� �
$��=�=��H�H�N�N�$� ��� %��
� )��
� #�� �� ��� %��
� )��
� #�� ��
$��=�=��H�H�N�N�$� ��� %��
� )��
� #�� �� ��� %��
� )��
� #�� �s0 �A;C? �&C �"C<�?E-�&E�'!E-�"E*�*E-c �* � | j t k( S �N)r r r? s r9 �closedzBZ2File.closedy s � �*��z�z�\�)�)r; c �X � | j � | j j � S rA )�_check_not_closedr �filenor? s r9 rE zBZ2File.fileno~ s" � �A� � � � ��x�x��� � r; c �\ � | j � xr | j j � S rA )�readabler2 �seekabler? s r9 rH zBZ2File.seekable� s# � �7��}�}��:�4�<�<�#8�#8�#:�:r; c �J � | j � | j t k( S rA )rD r r" r? s r9 rG zBZ2File.readable� s � �=� � � � ��z�z�Z�'�'r; c �J � | j � | j t k( S rA )rD r r# r? s r9 �writablezBZ2File.writable� s � �=� � � � ��z�z�[�(�(r; c �Z � | j � | j j |� S rA )�_check_can_readr2 �peek)r4 �ns r9 rN z BZ2File.peek� s, � � �
� � � � �|�|� � ��#�#r; c �Z � | j � | j j |� S rA )rM r2 r �r4 �sizes r9 r z BZ2File.read� s* � � �
� � � ��|�|� � ��&�&r; c � � | j � |dk rt j }| j j |� S )Nr )rM r0 �DEFAULT_BUFFER_SIZEr2 �read1rQ s r9 rU z
BZ2File.read1� s< � � �
� � � � �!�8��)�)�D��|�|�!�!�$�'�'r; c �Z � | j � | j j |� S rA )rM r2 �readinto)r4 �bs r9 rW zBZ2File.readinto� s* � � �
� � � ��|�|�$�$�Q�'�'r; c �� � t |t � s't |d� s t d� �|j � }| j � | j j |� S �N� __index__zInteger argument expected)r% �intr+ r, r[ rM r2 �readlinerQ s r9 r] zBZ2File.readline� sV � � � �$��$��4��-� � ;�<�<��>�>�#�D� � � � ��|�|�$�$�T�*�*r; c �� � t |t � s't |d� s t d� �|j � }| j � | j j |� S rZ )r% r\ r+ r, r[ rM r2 � readlinesrQ s r9 r_ zBZ2File.readlines� sV � � � �$��$��4��-� � ;�<�<��>�>�#�D� � � � ��|�|�%�%�d�+�+r; c �0 � | j � t |t t f� r t |� }nt |� }|j }| j j |� }| j j |� | xj |z
c_ |S rA ) �_check_can_writer% r' � bytearray�len�
memoryview�nbytesr$ r r r r3 )r4 �data�length�
compresseds r9 r z
BZ2File.write� s{ � � �
� � � � �d�U�I� .� /���Y�F� �d�#�D��[�[�F��%�%�.�.�t�4�
� �����z�"� � � �V�� ��
r; c �D � t j j | |� S rA )r- �
BaseStream�
writelines)r4 �seqs r9 rk zBZ2File.writelines� s" � � � �&�&�1�1�$��<� )rf r �comps r9 r r 9 s- � �� � � '�D� �=�=�� ���� � -�-r; c �� � g }| rSt � } |j | � }|j |� |j s t d� �|j } | r�Sdj |� S # t $ r |rY � � w xY w)NzACompressed data ended before the end-of-stream marker was reachedr; )r r r/ �append�eofr! � unused_data�join)rf �results�decomp�ress r9 r r D s� � �� �G�
� �"�� ��#�#�D�)�C� ���s���z�z� � @� A�
A��!�!�� � �8�8�G� � �� � ���� �s �A* �* A9�7A9)r r NNN)r )�__all__�
__author__�builtinsr r* r0 r( r- �_bz2r r r r" r# rj r r r rz r; r9 �r� se ��� -�� 5�
� *� � � � /� � ��
�� �r�l�%�%� r�j -.�-1�'�T.� r;