3
��2anO � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZddddgZd\Z Z
Z Z Z
d\ZZd d
d�Zdd� ZG dd� d�ZG dd� dej�ZG dd� dej�Zd dd�Zdd� Zdd� Zedkr�e� dS ) � N�GzipFile�open�compress�
decompress� � � � � �rb� c C s� d|kr d|krPt d|f ��n0|d k r0t d��|d k r@t d��|d k rPt d��|jdd�}t| tttjf�r|t| ||�}n,t| d�s�t| d �r�td ||| �}nt d
��d|kr�t
j ||||�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/python36/lib64/python3.6/gzip.pyr s$ c C s | j tjd|�� d S )Nz<L)r �struct�pack)�output�valuer# r# r$ �write32u@ 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__J s
z_PaddedFile.__init__c C s~ | j d kr| jj|�S | j | | jkrJ| j }| j |7 _ | j|| j � S | j }d | _ | j|d � | jj|| j | � S d S )N)r0 r/ r r. r, )r1 �sizer r# r# r$ r P s
z_PaddedFile.readc C s> | j d kr|| _n| j t|�8 _ d S t| j�| _d| _ d S )Nr )r0 r, r- r. )r1 r3 r# r# r$ r3 ] s
z_PaddedFile.prependc C s d | _ d | _| jj|�S )N)r0 r, r/ �seek)r1 Zoffr# r# r$ r6 f s z_PaddedFile.seekc C s dS )NTr# )r1 r# r# r$ �seekablek s z_PaddedFile.seekableN)r+ )r+ )�__name__�
__module__� __qualname__r4 r r3 r6 r7 r# r# r# r$ r* E s
r* c @ s� e Zd ZdZd,dd�Zedd� �Zedd� �Zd d
� Zd d � Z d
d� Z
dd� Z d.dd�Z d0dd�Z
dd� Zedd� �Zdd� Zejfd d �Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zejfd(d)�Zd2d*d+�Z dS )3r Nr c C s2 |r"d|ksd|kr"t dj|���|r6d|kr6|d7 }|d krTtj||pJd� }| _|d kr|t|dd�}t|ttf�s�d}n
t j
|�}|d kr�t|dd�}|j d �r�t | _
t|�}tj|�| _|| _nN|j d��rt| _
| j|� tj|tjtj tjd
�| _|| _nt dj|���|| _ | j
tk�r.| j � d S )Nr
�UzInvalid mode: {!r}r r �namer r �r�w�a�xr )r> r? r@ ) r �format�builtinsr � myfileobj�getattrr r r r �fspath�
startswith�READr � _GzipReaderr �BufferedReaderr, r< �WRITE� _init_write�zlibZ compressobjZDEFLATED� MAX_WBITSZ
DEF_MEM_LEVELr � _write_mtime�fileobj�_write_gzip_header)r1 r r r rO �mtime�rawr# r# r$ r4 { s> #
zGzipFile.__init__c C sB dd l }|jdtd� | jtkr<| jdd � dkr<| jd S | jS )Nr zuse the name attributer � z.gz���)�warnings�warn�DeprecationWarningr rJ r< )r1 rU r# r# r$ r � s
zGzipFile.filenamec C s
| j jjS )N)r, rR � _last_mtime)r1 r# r# r$ rQ � s zGzipFile.mtimec C s. t | j�}d|dd� d tt| �� d S )Nz<gzip r � �>���)�reprrO �hex�id)r1 �sr# r# r$ �__repr__� s
zGzipFile.__repr__c C s. || _ tjd�| _d| _g | _d| _d| _d S )Nr+ r )r< rL �crc32�crcr5 Zwritebuf�bufsize�offset)r1 r r# r# r$ rK � s zGzipFile._init_writec C s� | j jd� | j jd� y<tjj| j�}t|t�s<|jd�}|j d�rR|d d � }W n t
k
rl d}Y nX d}|rzt }| j jt |�jd�� | j
}|d kr�tj� }t| j t|�� | j jd� | j jd � |r�| j j|d
� d S ) Ns �� zlatin-1s .gzrS r+ r � � �� rT )rO r r �path�basenamer< r r �encode�endswith�UnicodeEncodeError�FNAME�chrrN �timer) �int)r1 Zfname�flagsrQ r# r# r$ rP � s,
zGzipFile._write_gzip_headerc C s� | j � | jtkr&dd l}t|jd��| jd kr8td��t|t �rLt
|�}nt |�}|j }|dkr�| jj
| jj|�� | j|7 _tj|| j�| _| j|7 _|S )Nr z$write() on read-only GzipFile objectz!write() on closed GzipFile object)�_check_not_closedr rJ �errno�OSError�EBADFrO r r r r- �
memoryview�nbytesr r r5 rL ra rb rd )r1 �datart Zlengthr# r# r$ r � s
zGzipFile.writer c C s2 | j � | jtkr&dd l}t|jd��| jj|�S )Nr z$read() on write-only GzipFile object)rs r rG rt ru rv r, r )r1 r5 rt r# r# r$ r s
z
GzipFile.readc C s@ | j � | jtkr&dd l}t|jd��|dk r4tj}| jj |�S )Nr z%read1() on write-only GzipFile object)
rs r rG rt ru rv r �DEFAULT_BUFFER_SIZEr, �read1)r1 r5 rt r# r# r$ r{ s
zGzipFile.read1c C s2 | j � | jtkr&dd l}t|jd��| jj|�S )Nr z$peek() on write-only GzipFile object)rs r rG rt ru rv r, �peek)r1 �nrt r# r# r$ r| # s
z
GzipFile.peekc C s
| j d kS )N)rO )r1 r# r# r$ �closed* s zGzipFile.closedc C s� | j }|d krd S d | _ zP| jtkrR|j| jj� � t|| j� t|| jd@ � n| jt krf| j
j � W d | j }|r�d | _ |j � X d S )Nl �� )
rO r rJ r r �flushr) rb r5 rG r, �closerC )r1 rO rC r# r# r$ r� . s
zGzipFile.closec C s4 | j � | jtkr0| jj| jj|�� | jj� d S )N)rs r rJ rO r r r )r1 Z zlib_moder# r# r$ r A s
zGzipFile.flushc C s
| j j� S )N)rO �fileno)r1 r# r# r$ r� H s zGzipFile.filenoc C s" | j tkrtd��| jjd� d S )NzCan't rewind in write moder )r rG ru r, r6 )r1 r# r# r$ �rewindP s
zGzipFile.rewindc C s
| j tkS )N)r rG )r1 r# r# r$ �readableW s zGzipFile.readablec C s
| j tkS )N)r rJ )r1 r# r# r$ �writableZ s zGzipFile.writablec C s dS )NTr# )r1 r# r# r$ r7 ] s zGzipFile.seekablec C s� | j tkr�|tjkr2|tjkr*| j| }ntd��|| jk rDtd��|| j }dd }x t|d �D ]}| j |� qdW | j d|d � n | j t
kr�| j � | j j
||�S | jS )NzSeek from end not supportedzNegative seek in write moderh i )r rJ r �SEEK_SET�SEEK_CURrd r ru �ranger rG rs r, r6 )r1 rd �whence�count�chunk�ir# r# r$ r6 ` s
z
GzipFile.seekc C s | j � | jj|�S )N)rs r, �readline)r1 r5 r# r# r$ r� t s zGzipFile.readline)NNr NNr[ )r[ r[ )r[ r[ )r[ ) r8 r9 r: rC r4 �propertyr rQ r` rK rP r r r{ r| r~ r� rL Z Z_SYNC_FLUSHr r� r� r� r� r7 r r� r6 r� r# r# r# r$ r n s,
H
c sZ e Zd Z� fdd�Zdd� Zdd� Zdd� Zdd
d �Zd d
� Zdd� Z � fdd�Z
� Z S )rH c s, t � jt|�tjtj d� d| _d | _d S )N)ZwbitsT)�superr4 r* rL Z
decompressobjrM � _new_memberrX )r1 �fp)� __class__r# r$ r4 z s z_GzipReader.__init__c C s t jd�| _d| _d S )Nr+ r )rL ra �_crc� _stream_size)r1 r# r# r$ �
_init_read� s z_GzipReader._init_readc C sJ | j j|�}x8t|�|k rD| j j|t|� �}|s:td��||7 }qW |S )NzACompressed file ended before the end-of-stream marker was reached)�_fpr r- �EOFError)r1 r} ry r r# r# r$ � _read_exact� s z_GzipReader._read_exactc C s� | j jd�}|dkrdS |dkr,td| ��tjd| jd��\}}| _|dkrVtd��|t@ r|tjd | jd��\}| j|� |t@ r�x | j jd
�}| s�|d kr�P q�W |t @ r�x | j jd
�}| s�|d kr�P q�W |t
@ r�| jd� d S )
Nr r+ Fs �zNot a gzipped file (%r)z<BBIxxr zUnknown compression methodz<Hr rh T) r� r ru r% �unpackr� rX �FEXTRArn �FCOMMENT�FHCRC)r1 �magic�method�flagZ extra_lenr_ r# r# r$ �_read_gzip_header� s0
z _GzipReader._read_gzip_headerr c C s� |dk r| j � S |sdS x�| jjr@| j� d| _| jf | j�| _| jrh| j� | j� sb| j | _
dS d| _| j j t
j�}| jj||�}| jjdkr�| j j| jj� n | jjdkr�| j j| jj� |dkr�P |dkrtd��qW | j|� | j t|�7 _ |S )Nr r+ TFzACompressed file ended before the end-of-stream marker was reached)�readallZ
_decompressor�eof� _read_eofr� Z_decomp_factoryZ _decomp_argsr� r� Z_posZ_sizer� r r rz r Zunconsumed_tailr3 Z unused_datar� �_add_read_datar- )r1 r5 �bufZ
uncompressr# r# r$ r � s:
z_GzipReader.readc C s$ t j|| j�| _| jt|� | _d S )N)rL ra r� r� r- )r1 ry r# r# r$ r� � s z_GzipReader._add_read_datac C s� t jd| jd��\}}|| jkr<tdt|�t| j�f ��n|| jd@ krRtd��d}x|dkrn| jjd�}qXW |r�| jj |� d S )Nz<IIr zCRC check failed %s != %sl �� z!Incorrect length of data producedrh r )
r% r� r� r� ru r] r� r� r r3 )r1 ra Zisize�cr# r# r$ r� � s
z_GzipReader._read_eofc s t � j� d| _d S )NT)r� �_rewindr� )r1 )r� r# r$ r� s
z_GzipReader._rewindr[ )r[ ) r8 r9 r: r4 r� r� r� r r� r� r� �
__classcell__r# r# )r� r$ rH y s !
3rH c C s4 t j� }t|d|d��}|j| � W d Q R X |j� S )N�wb)rO r r )r �BytesIOr r �getvalue)ry r r� r2 r# r# r$ r s c C s$ t tj| �d��
}|j� S Q R X d S )N)rO )r r r� r )ry r2 r# r# r$ r s c C s> t jdd � } | o | d dk}|r.| dd � } | s8dg} � x�| D ]�}|r�|dkrltddt jjd�}t jj}q�|d
d � d kr�td
t|�� q@t|d�}t j|d d� d �}n>|dkr�t jj}tdd t jjd�}nt j|d�}t|d d �}x |j
d �}|� s�P |j |� q�W |t jjk �r"|j � |t jjk r@|j � q@W d S )Nr r z-d�-r r )r r rO rS z.gzz filename doesn't end in .gz:r� i rT rT )
�sys�argvr �stdin�buffer�stdout�printr\ r rB r r r� )�argsr �argr2 �gr� r# r# r$ �_test s<
r� �__main__)r r r r r
)r r )r r NNN)r )r% r� rp r rL rB r Z _compression�__all__ZFTEXTr� r� rn r� rG rJ r r) r* Z
BaseStreamr ZDecompressReaderrH r r r� r8 r# r# r# r$ �<module> s*
+)
&