U
e5d�S � @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZdddddgZd\Z Z
Z Z Z
d\ZZd Zd
Zd Zd edddfd
d�Zdd� ZG dd� d�ZG dd� de�ZG dd� dej�ZG dd� dej�Zefdd�dd�Z dd� Z dd� Z e dkr�e � dS ) � N� BadGzipFile�GzipFile�open�compress�
decompress)� � � � � )r r r � � �rbc C s� d|kr d|krPt d|f ��n0|d k r0t d��|d k r@t d��|d k rPt d��|�dd�}t| tttjf�r|t| ||�}n,t| d�s�t| d �r�td ||| �}nt d
��d|kr�t
� ||||�S |S d S ) N�t�bzInvalid mode: %rz0Argument 'encoding' not supported in binary modez.Argument 'errors' not supported in binary modez/Argument 'newline' not supported in binary mode� �read�writez1filename must be a str or bytes object, or a file) �
ValueError�replace�
isinstance�str�bytes�os�PathLiker �hasattr� TypeError�io�
TextIOWrapper)�filename�mode�
compresslevel�encoding�errors�newlineZgz_modeZ binary_file� r% � /usr/lib64/python3.8/gzip.pyr s$ c C s | � t�d|�� d S )Nz | j d kr|| _n| j t|�8 _ d S t| j�| _d| _ d S r- )r2 r. r/ r0 )r3 r5 r% r% r& r5 b s
z_PaddedFile.prependc C s d | _ d | _| j�|�S r7 )r2 r. r1 �seek)r3 Zoffr% r% r& r9 k s z_PaddedFile.seekc C s dS �NTr% �r3 r% r% r& �seekablep s z_PaddedFile.seekableN)r, )r, )�__name__�
__module__� __qualname__r6 r r5 r9 r< r% r% r% r& r+ J s
r+ c @ s e Zd ZdS )r N)r= r> r? r% r% r% r& r t s c @ s� e Zd ZdZddeddfdd�Zedd� �Zedd� �Zdd � Z d
d � Z
d d
� Z dd� Z d+dd�Z
d,dd�Zdd� Zedd� �Zdd� Zejfdd �Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zejfd'd(�Z d-d)d*�Z dS ).r Nc C s4 |r"d|ksd|kr"t d�|���|r6d|kr6|d7 }|d krTt�||pJd� }| _|d kr|t|dd�}t|ttf�s�d}n
t �
|�}|d kr�t|dd�}|� d �r�t | _
t|�}t�|�| _|| _nN|� d
��rt| _
| �|� t�|tjtj tjd �| _|| _nt d�|���|| _ | j
tk�r0| � |� d S ) Nr �UzInvalid mode: {!r}r r �namer r �r)�w�a�xr ) r �format�builtinsr � myfileobj�getattrr r r r �fspath�
startswith�READr � _GzipReaderr �BufferedReaderr. rA �WRITE� _init_write�zlibZ compressobjZDEFLATED� MAX_WBITSZ
DEF_MEM_LEVELr � _write_mtime�fileobj�_write_gzip_header)r3 r r r! rT �mtime�rawr% r% r& r6 � s@ #
� zGzipFile.__init__c C sB dd l }|�dtd� | jtkr<| jdd � dkr<| jd S | jS )Nr zuse the name attributer ������.gz)�warnings�warn�DeprecationWarningr rO rA )r3 rZ r% r% r& r � s
zGzipFile.filenamec C s
| j jjS r7 )r. rW � _last_mtimer; r% r% r& rV � s zGzipFile.mtimec C s. t | j�}d|dd� d tt| �� d S )Nz)�reprrT �hex�id)r3 �sr% r% r& �__repr__� s
zGzipFile.__repr__c C s. || _ t�d�| _d| _g | _d| _d| _d S �Nr, r )rA rQ �crc32�crcr8 Zwritebuf�bufsize�offset)r3 r r% r% r& rP � s zGzipFile._init_writec C s | j �d� | j �d� z s
zGzipFile.closec C s4 | � � | jtkr0| j�| j�|�� | j�� d S r7 )r} r rO rT r r r� )r3 Z zlib_moder% r% r& r� Q s
zGzipFile.flushc C s
| j �� S r7 )rT �filenor; r% r% r& r� X s zGzipFile.filenoc C s" | j tkrtd��| j�d� d S )NzCan't rewind in write moder )r rL r r. r9 r; r% r% r& �rewind` s
zGzipFile.rewindc C s
| j tkS r7 )r rL r; r% r% r& �readableg s zGzipFile.readablec C s
| j tkS r7 )r rO r; r% r% r& �writablej s zGzipFile.writablec C s dS r: r% r; r% r% r& r< m s zGzipFile.seekablec C s� | j tkr�|tjkr2|tjkr*| j| }ntd��|| jk rDtd��|| j }d}t|d �D ]}| � |� q^| � d|d � n | j t
kr�| � � | j �
||�S | jS )NzSeek from end not supportedzNegative seek in write modes � ro )r rO r �SEEK_SET�SEEK_CURrj r r �ranger rL r} r. r9 )r3 rj �whence�count�chunk�ir% r% r& r9 p s
z
GzipFile.seekc C s | � � | j�|�S r7 )r} r. �readline)r3 r8 r% r% r& r� � s zGzipFile.readline)r^ )r^ )r^ ) r= r> r? rH rz r6 �propertyr rV re rP rU r r r� r� r� r� rQ Z Z_SYNC_FLUSHr� r� r� r� r� r< r r� r9 r� r% r% r% r& r x s8 �
I
c sZ e Zd Z� fdd�Zdd� Zdd� Zdd� Zdd
d �Zd d
� Zdd� Z � fdd�Z
� Z S )rM c s, t � jt|�tjtj d� d| _d | _d S )N)ZwbitsT)�superr6 r+ rQ Z
decompressobjrR � _new_memberr] )r3 �fp�� __class__r% r& r6 � s
�z_GzipReader.__init__c C s t �d�| _d| _d S rf )rQ rg �_crc� _stream_sizer; r% r% r& �
_init_read� s z_GzipReader._init_readc C sF | j �|�}t|�|k rB| j �|t|� �}|s8td��||7 }q |S )N�ACompressed file ended before the end-of-stream marker was reached)�_fpr r/ �EOFError)r3 r� r� r r% r% r& � _read_exact� s
z_GzipReader._read_exactc C s� | j �d�}|dkrdS |dkr,td| ��t�d| �d��\}}| _|dkrVtd��|t@ r|t�d | �d��\}| �|� |t@ r�| j �d
�}|r�|d kr�q�q�|t @ r�| j �d
�}|r�|d kr�q�q�|t
@ r�| �d� d S )
Nr r, Frk zNot a gzipped file (%r)z| �� d| _| jf | j�| _| jrf| �� | �� s`| j | _
dS d| _| j � t
j�}| j�||�}| jjdkr�| j �| jj� n | jjdkr�| j �| jj� |dkr�q�|dkrtd��q| �|� | j t|�7 _ |S )Nr r, TFr� )�readallZ
_decompressor�eof� _read_eofr� Z_decomp_factoryZ _decomp_argsr� r� Z_posZ_sizer� r r r� r Zunconsumed_tailr5 Z unused_datar� �_add_read_datar/ )r3 r8 �bufZ
uncompressr% r% r& r � s: �
z_GzipReader.readc C s$ t �|| j�| _| jt|� | _d S r7 )rQ rg r� r� r/ )r3 r� r% r% r& r� � s z_GzipReader._add_read_datac C s� t �d| �d��\}}|| jkr r? r6 r� r� r� r r� r� r� �
__classcell__r% r% r� r& rM � s !
3rM )rV c C s6 t �� }t|d||d��}|�| � W 5 Q R X |�� S )N�wb)rT r r! rV )r �BytesIOr r �getvalue)r� r! rV r� r4 r% r% r& r s c
C s0 t t�| �d��}|�� W 5 Q R � S Q R X d S )Nr� )r r r� r )r� r4 r% r% r& r s c C s� ddl m} | dd�}|�� }|jdddd� |jd dd
d� |jd d dd
d� |jdddgdd� |�� }t}|jr|t}n
|jr�t }|j
D ]�}|j r�|dkr�t ddt
jjd�}t
jj}n>|dd � dkr�t
�d|��� t|d�}t�|d d� d�}nB|dk�r"t
jj}t ddt
jj|d�}nt�|d�}t|d d�}|�d�}|�sP�q^|�|� �q<|t
jjk �rt|�� |t
jjk r�|�� q�d S ) Nr )�ArgumentParserzeA simple command line interface for the gzip module: act like gzip, but do not delete the input file.)Z descriptionz--fast�
store_truezcompress faster)�action�helpz--bestzcompress betterz-dz --decompressz act like gunzip instead of gzip�args�*�-r1 )�nargs�default�metavarr r )r r rT rX rY z filename doesn't end in .gz: r� )r r rT r! r� )�argparser� Z add_mutually_exclusive_group� add_argument�
parse_args�_COMPRESS_LEVEL_TRADEOFFZfastr{ Zbestrz r� r r �sys�stdin�buffer�stdout�exitr rG r r r� ) r� �parser�groupr� r! �argr4 �gr� r% r% r& �main' sR �
�
�
r� �__main__) r' r� rx r rQ rG r Z _compression�__all__ZFTEXTr� r� rv r� rL rO r{ r� rz r r* r+ r r Z
BaseStreamr ZDecompressReaderrM r r r� r= r% r% r% r& � s8 �
,* 0