U
e5d� � @ s� d Z ddlZddlZddd�Zddd�ZG dd � d ej�ZG d
d � d ej�ZG d d
� d
ej�ZG dd� deej�ZG dd� deej �Z dd� Z
dS )a Python 'bz2_codec' Codec - bz2 compression encoding.
This codec de/encodes from bytes to bytes and is therefore usable with
bytes.transform() and bytes.untransform().
Adapted by Raymond Hettinger from zlib_codec.py which was written
by Marc-Andre Lemburg ([email protected]).
� N�strictc C s |dks t �t�| �t| �fS �Nr )�AssertionError�bz2�compress�len��input�errors� r �+/usr/lib64/python3.8/encodings/bz2_codec.py�
bz2_encode s r
c C s |dks t �t�| �t| �fS r )r r �
decompressr r r r r �
bz2_decode s r c @ s e Zd Zddd�Zddd�ZdS ) �Codecr c C s
t ||�S �N)r
��selfr r
r r r �encode s z Codec.encodec C s
t ||�S r )r r r r r �decode s z Codec.decodeN)r )r )�__name__�
__module__� __qualname__r r r r r r r s
r c @ s( e Zd Zd
dd�Zd dd�Zdd� Zd S ) �IncrementalEncoderr c C s |dks t �|| _t�� | _d S r )r r
r �
BZ2Compressor� compressobj�r r
r r r �__init__ s zIncrementalEncoder.__init__Fc C s. |r | j �|�}|| j �� S | j �|�S d S r )r r �flush)r r �final�cr r r r # s zIncrementalEncoder.encodec C s t �� | _d S r )r r r �r r r r �reset* s zIncrementalEncoder.resetN)r )F)r r r r r r" r r r r r s
r c @ s( e Zd Zd
dd�Zd dd�Zdd� Zd S ) �IncrementalDecoderr c C s |dks t �|| _t�� | _d S r )r r
r �BZ2Decompressor�
decompressobjr r r r r . s zIncrementalDecoder.__init__Fc C s* z| j �|�W S tk
r$ Y dS X d S )N� )r% r �EOFError)r r r r r r r 3 s zIncrementalDecoder.decodec C s t �� | _d S r )r r$ r% r! r r r r" 9 s zIncrementalDecoder.resetN)r )F)r r r r r r" r r r r r# - s
r# c @ s e Zd ZeZdS )� StreamWriterN�r r r �bytes�charbuffertyper r r r r( < s r( c @ s e Zd ZeZdS )� StreamReaderNr) r r r r r, ? s r, c
C s t jdttttttdd�S )Nr F)�namer r �incrementalencoder�incrementaldecoder� streamwriter� streamreader�_is_text_encoding)�codecs� CodecInfor
r r r# r( r, r r r r � getregentryD s �r5 )r )r ) �__doc__r3 r r
r r r r# r( r, r5 r r r r � s