U
e5dS @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZdddddgZd\Z Z
Z Z Z
d\ZZd Zd
Zd Zd edddfd
dZdd ZG dd dZG dd deZG dd dejZG dd dejZefddddZ dd Z dd Z e dkre dS ) N BadGzipFileGzipFileopencompress
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| tttjfr|t| ||}n,t| dst| d rtd ||| }nt d
d|krt
||||S |S d S ) NtbzInvalid mode: %rz0Argument 'encoding' not supported in binary modez.Argument 'errors' not supported in binary modez/Argument 'newline' not supported in binary mode readwritez1filename must be a str or bytes object, or a file)
ValueErrorreplace
isinstancestrbytesosPathLiker hasattr TypeErrorio
TextIOWrapper)filenamemode
compresslevelencodingerrorsnewlineZgz_modeZ binary_file r% /usr/lib64/python3.8/gzip.pyr s$ c C s | td| 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 ZdZddeddfddZedd Zedd Zdd Z d
d Z
d d
Z dd Z d+ddZ
d,ddZdd 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|ttfsd}n
t
|}|d krt|dd}| d rt | _
t|}t|| _|| _nN| d
rt| _
| | t|tjtj tjd | _|| _nt d||| _ | j
tkr0| | d S ) Nr UzInvalid mode: {!r}r r namer r r)waxr ) r formatbuiltinsr myfileobjgetattrr r r r fspath
startswithREADr _GzipReaderr BufferedReaderr. rA WRITE _init_writezlibZ compressobjZDEFLATED MAX_WBITSZ
DEF_MEM_LEVELr _write_mtimefileobj_write_gzip_header)r3 r r r! rT mtimerawr% r% r& r6 s@ #
zGzipFile.__init__c C sB dd l }|dtd | jtkr<| jdd dkr<| jd S | jS )Nr zuse the name attributer .gz)warningswarnDeprecationWarningr 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 hexid)r3 sr% r% r& __repr__ s
zGzipFile.__repr__c C s. || _ td| _d| _g | _d| _d| _d S Nr, r )rA rQ crc32crcr8 Zwritebufbufsizeoffset)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| jd 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_SETSEEK_CURrj r r ranger rL r} r. r9 )r3 rj whencecountchunkir% 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 fddZdd Zdd Zdd Zdd
d Zd d
Zdd Z fddZ
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 )NACompressed 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| td| d\}}| _|dkrVtd|t@ r|td | d\}| | |t@ r| j d
}|r|d krqq|t @ r| j d
}|r|d krqq|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 |dkrq|dkrtdq| | | j t|7 _ |S )Nr r, TFr )readallZ
_decompressoreof _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 )Nwb)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
|jrt }|j
D ]}|j r|dkrt ddt
jjd}t
jj}n>|dd dkrt
d| t|d}t|d d d}nB|dkr"t
jj}t ddt
jj|d}nt|d}t|d d}|d}|sPq^|| q<|t
jjk rt| |t
jjk r| qd 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)actionhelpz--bestzcompress betterz-dz --decompressz act like gunzip instead of gzipargs*-r1 )nargsdefaultmetavarr 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 sysstdinbufferstdoutexitr rG r r r ) r parsergroupr 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