3
\� � @ 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 �tj| �t| �fS )Nr )�AssertionError�bz2�compress�len)�input�errors� r �+/usr/lib64/python3.6/encodings/bz2_codec.py�
bz2_encode s r c C s |dks t �tj| �t| �fS )Nr )r r �
decompressr )r 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 )N)r
)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 �|| _tj� | _d S )Nr )r r r �
BZ2Compressor� compressobj)r r r r r
�__init__ s zIncrementalEncoder.__init__Fc C s. |r | j j|�}|| j j� S | j j|�S d S )N)r r �flush)r r �final�cr r r
r # s zIncrementalEncoder.encodec C s t j� | _d S )N)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 �|| _tj� | _d S )Nr )r r r �BZ2Decompressor�
decompressobj)r r r r r
r . s zIncrementalDecoder.__init__Fc C s& y | j j|�S tk
r dS X d S )N� )r r �EOFError)r r r r r r
r 3 s zIncrementalDecoder.decodec C s t j� | _d S )N)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 )� StreamReaderN)r r r r# r$ 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 r. )r )r ) �__doc__r, r r r
r r r r" r% r. r r r r
� s