3
\nO � @ 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# � /usr/lib64/python3.6/gzip.pyr s$ c C s | j tjd|�� d S )Nz<L)r �structZpack)�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)r/ r. r r- r+ )r0 �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 )r/ r+ r, r- )r0 r2 r# r# r$ r2 ] s
z_PaddedFile.prependc C s d | _ d | _| jj|�S )N)r/ r+ r. �seek)r0 Zoffr# r# r$ r5 f s z_PaddedFile.seekc C s dS )NTr# )r0 r# r# r$ �seekablek s z_PaddedFile.seekableN)r* )r* )�__name__�
__module__� __qualname__r3 r r2 r5 r6 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)r0 r r r rN �mtime�rawr# r# r$ r3 { 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 rI r; )r0 rT r# r# r$ r � s
zGzipFile.filenamec C s
| j jjS )N)r+ rQ � _last_mtime)r0 r# r# r$ rP � s zGzipFile.mtimec C s. t | j�}d|dd� d tt| �� d S )Nz<gzip r � �>���)�reprrN �hex�id)r0 �sr# r# r$ �__repr__� s
zGzipFile.__repr__c C s. || _ tjd�| _d| _g | _d| _d| _d S )Nr* r )r; rK �crc32�crcr4 Zwritebuf�bufsize�offset)r0 r r# r# r$ rJ � 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 .gzrR r* r � � �� rS )rN r r �path�basenamer; r r �encode�endswith�UnicodeEncodeError�FNAME�chrrM �timer( �int)r0 Zfname�flagsrP r# r# r$ rO � 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 rI �errno�OSError�EBADFrN r r r r, �
memoryview�nbytesr r r4 rK r` ra rc )r0 �datars 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)rr r rF rs rt ru r+ r )r0 r4 rs 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)
rr r rF rs rt ru r �DEFAULT_BUFFER_SIZEr+ �read1)r0 r4 rs r# r# r$ rz s
zGzipFile.read1c C s2 | j � | jtkr&dd l}t|jd��| jj|�S )Nr z$peek() on write-only GzipFile object)rr r rF rs rt ru r+ �peek)r0 �nrs r# r# r$ r{ # s
z
GzipFile.peekc C s
| j d kS )N)rN )r0 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 �� )
rN r rI r r �flushr( ra r4 rF r+ �closerB )r0 rN rB r# r# r$ r . s
zGzipFile.closec C s4 | j � | jtkr0| jj| jj|�� | jj� d S )N)rr r rI rN r r r~ )r0 Z zlib_moder# r# r$ r~ A s
zGzipFile.flushc C s
| j j� S )N)rN �fileno)r0 r# r# r$ r� H s zGzipFile.filenoc C s" | j tkrtd��| jjd� d S )NzCan't rewind in write moder )r rF rt r+ r5 )r0 r# r# r$ �rewindP s
zGzipFile.rewindc C s
| j tkS )N)r rF )r0 r# r# r$ �readableW s zGzipFile.readablec C s
| j tkS )N)r rI )r0 r# r# r$ �writableZ s zGzipFile.writablec C s dS )NTr# )r0 r# r# r$ r6 ] 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 moderg i )r rI r �SEEK_SET�SEEK_CURrc r rt �ranger rF rr r+ r5 )r0 rc �whence�count�chunk�ir# r# r$ r5 ` s
z
GzipFile.seekc C s | j � | jj|�S )N)rr r+ �readline)r0 r4 r# r# r$ r� t s zGzipFile.readline)NNr NNrZ )rZ rZ )rZ rZ )rZ ) r7 r8 r9 rB r3 �propertyr rP r_ rJ rO r r rz r{ r} r rK Z Z_SYNC_FLUSHr~ r� r� r� r� r6 r r� r5 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 )rG c s, t � jt|�tjtj d� d| _d | _d S )N)ZwbitsT)�superr3 r) rK Z
decompressobjrL � _new_memberrW )r0 �fp)� __class__r# r$ r3 z s z_GzipReader.__init__c C s t jd�| _d| _d S )Nr* r )rK r` �_crc� _stream_size)r0 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)r0 r| rx 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 rg T) r� r rt r% �unpackr� rW �FEXTRArm �FCOMMENT�FHCRC)r0 �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 ry r Zunconsumed_tailr2 Z unused_datar� �_add_read_datar, )r0 r4 �bufZ
uncompressr# r# r$ r � s:
z_GzipReader.readc C s$ t j|| j�| _| jt|� | _d S )N)rK r` r� r� r, )r0 rx 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 producedrg r )
r% r� r� r� rt r\ r� r� r r2 )r0 r` Zisize�cr# r# r$ r� � s
z_GzipReader._read_eofc s t � j� d| _d S )NT)r� �_rewindr� )r0 )r� r# r$ r� s
z_GzipReader._rewindrZ )rZ ) r7 r8 r9 r3 r� r� r� r r� r� r� �
__classcell__r# r# )r� r$ rG y s !
3rG c C s4 t j� }t|d|d��}|j| � W d Q R X |j� S )N�wb)rN r r )r �BytesIOr r �getvalue)rx r r� r1 r# r# r$ r s c C s$ t tj| �d��
}|j� S Q R X d S )N)rN )r r r� r )rx r1 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 rN rR z.gzz filename doesn't end in .gz:r� i rS rS )
�sys�argvr �stdin�buffer�stdout�printr[ r rA r r r )�argsr �argr1 �gr� r# r# r$ �_test s<
r� �__main__)r r r r r
)r r )r r NNN)r )r% r� ro r rK rA r Z _compression�__all__ZFTEXTr� r� rm r� rF rI r r( r) Z
BaseStreamr ZDecompressReaderrG r r r� r7 r# r# r# r$ �<module> s*
+)
&