U
��,a�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% �)/opt/alt/python38/lib64/python3.8/gzip.pyr s$ c C s | � t�d|�� d S )Nz<L)r �structZpack)�output�valuer% r% r& �write32uE s r* c @ s8 e Zd Zd
dd�Zdd� Zddd�Zdd � Zd
d � Zd S )� _PaddedFile� c C s || _ t|�| _|| _d| _d S �Nr )�_buffer�len�_length�file�_read)�self�f�prependr% r% r& �__init__O s
z_PaddedFile.__init__c C s~ | j d kr| j�|�S | j | | jkrJ| j }| j |7 _ | j|| j � S | j }d | _ | j|d � | j�|| j | � S d S �N)r2 r1 r r0 r. )r3 �sizer r% r% r& r U s
�z_PaddedFile.readc C s> | 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<gzip r ���� �>)�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<tj�| j�}t|t�s<|�d�}|� d�rR|d d� }W n t
k
rl d}Y nX d}|rzt }| j �t |��d�� | j
}|d kr�t�� }t| j t|�� |tkr�d}n|tkr�d }nd
}| j �|� | j �d � |�r| j �|d
� d S ) N� �� zlatin-1s .gzrX r, r � � � � �)rT r r �path�basenamerA r r �encode�endswith�UnicodeEncodeError�FNAME�chrrS �timer* �int�_COMPRESS_LEVEL_BEST�_COMPRESS_LEVEL_FAST)r3 r! Zfname�flagsrV Zxflr% r% r& rU � s6
zGzipFile._write_gzip_headerc C s� | � � | jtkr&dd l}t|jd��| jd kr8td��t|t �rLt
|�}nt |�}|j }|dkr�| j�
| j�|�� | j|7 _t�|| j�| _| j|7 _|S )Nr z$write() on read-only GzipFile objectz!write() on closed GzipFile object)�_check_not_closedr rO �errno�OSError�EBADFrT r r r r/ �
memoryview�nbytesr r r8 rQ rg rh rj )r3 �datar~ Zlengthr% r% r& r s
zGzipFile.writer^ c C s2 | � � | jtkr&dd l}t|jd��| j�|�S )Nr z$read() on write-only GzipFile object)r} r rL r~ r r� r. r �r3 r8 r~ r% r% r& r s
z
GzipFile.readc C s@ | � � | jtkr&dd l}t|jd��|dk r4tj}| j� |�S )Nr z%read1() on write-only GzipFile object)
r} r rL r~ r r� r �DEFAULT_BUFFER_SIZEr. �read1r� r% r% r& r� & s
zGzipFile.read1c C s2 | � � | jtkr&dd l}t|jd��| j�|�S )Nr z$peek() on write-only GzipFile object)r} r rL r~ r r� r. �peek)r3 �nr~ r% r% r& r� 3 s
z
GzipFile.peekc C s
| j d kS r7 �rT r; r% r% r& �closed: s zGzipFile.closedc C s� | j }|d krd S d | _ zP| jtkrR|�| j�� � t|| j � t|| j
d@ � n| jt krf| j �� W 5 | j}|r�d | _|�� X d S )N� �� )
rT rH �closer rO r r �flushr* rh r8 rL r. )r3 rT rH r% r% r& r� > 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<BBIxxr
zUnknown compression methodz<Hr ro T) r� r r r' �unpackr� r] �FEXTRArv �FCOMMENT�FHCRC)r3 �magic�method�flagZ extra_lenrd r% r% r& �_read_gzip_header� s0 �
z _GzipReader._read_gzip_headerr^ c C s� |dk r| � � S |sdS | jjr>| �� 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<tdt|�t| j�f ��n|| jd@ krRtd��d}|dkrl| j�d�}qV|r|| j� |� d S )Nz<IIr
zCRC check failed %s != %sr� z!Incorrect length of data producedro r )
r' r� r� r� r rb r� r� r r5 )r3 rg Zisize�cr% r% r& r� � s
�
z_GzipReader._read_eofc s t � �� d| _d S r: )r� �_rewindr� r; r� r% r& r� s
z_GzipReader._rewind)r^ ) r= r> 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& �<module> s8 �
,* 0