a
    If                     @   s   d Z ddlZddlZdddZdddZG dd	 d	ejZG d
d dejZG dd dejZG dd deejZG dd deej	Z	dd Z
dS )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 (mal@lemburg.com).
    Nstrictc                 C   s   |dksJ t | t| fS Nr   )bz2compressleninputerrors r
   $lib/python3.9/encodings/bz2_codec.py
bz2_encode   s    r   c                 C   s   |dksJ t | t| fS r   )r   
decompressr   r   r
   r
   r   
bz2_decode   s    r   c                   @   s    e Zd ZdddZdddZdS )	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 Zd Zd
ddZdddZdd Zd	S )IncrementalEncoderr   c                 C   s    |dksJ || _ t | _d S r   )r	   r   BZ2Compressorcompressobjr   r	   r
   r
   r   __init__   s    zIncrementalEncoder.__init__Fc                 C   s.   |r| j |}|| j   S | j |S d S r   )r   r   flush)r   r   finalcr
   r
   r   r   #   s    zIncrementalEncoder.encodec                 C   s   t  | _d S r   )r   r   r   r   r
   r
   r   reset*   s    zIncrementalEncoder.resetN)r   )F)r   r   r   r   r   r!   r
   r
   r
   r   r      s   

r   c                   @   s(   e Zd Zd
ddZdddZdd Zd	S )IncrementalDecoderr   c                 C   s    |dksJ || _ t | _d S r   )r	   r   BZ2Decompressordecompressobjr   r
   r
   r   r   .   s    zIncrementalDecoder.__init__Fc                 C   s(   z| j |W S  ty"   Y dS 0 d S )N )r$   r   EOFError)r   r   r   r
   r
   r   r   3   s    zIncrementalDecoder.decodec                 C   s   t  | _d S r   )r   r#   r$   r    r
   r
   r   r!   9   s    zIncrementalDecoder.resetN)r   )F)r   r   r   r   r   r!   r
   r
   r
   r   r"   -   s   

r"   c                   @   s   e Zd ZeZdS )StreamWriterNr   r   r   bytescharbuffertyper
   r
   r
   r   r'   <   s   r'   c                   @   s   e Zd ZeZdS )StreamReaderNr(   r
   r
   r
   r   r+   ?   s   r+   c                
   C   s   t jdttttttdd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    r4   )r   )r   )__doc__r2   r   r   r   r   r   r"   r'   r+   r4   r
   r
   r
   r   <module>   s   	

