3
]i48 2 @ s d Z ddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z y ddl Z W n e
k
r| ddlZ Y nX yddlZejZW n e
k
r dZe jZY nX y ddlZW n e
k
r dZY nX y ddlZW n e
k
r dZY nX ddddddd d
d d d
dg ZG dd deZG dd deZe ZZdfZdhZdjZdZ dZ dZ dZ dZ dZ!dZ"dZ#dZ$dZ%d Z&e
j'e%Z(dZ)dZ*d Z+d Z,d Z-d Z.d!Z/d"Z0dZ1d#Z2d$Z3d%Z4e
j'e3Z5dZ6dZ7d Z8d Z9d Z:d Z;d!Z<d"Z=dZ>d#Z?d&Z@d'ZAdZBd(ZCdZDd)ZEdZFd*ZGd+ZHd,ZId-ZJe
j'eIZKdZLdZMd ZNd ZOd ZPd ZQd!ZRd"ZSdZTd#ZUd&ZVd'ZWd.ZXd/ZYe
j'eXZZd0Z[d1Z\e
j'e[Z]dZ^dZ_d Z`d Zad Zbd Zcd!Zdd"ZedZfd#Zgd2Zhe
jid3Zjd4d5 Zkd6d7 Zld8d
Zmd9d: Znd;d< ZoG d=d d epZqG d>d? d?ZrG d@dA dAZsG dBdC dCZtdDdEdFdFdFdFdGdHdIdJdGdKdLdMdNdOdPdQZudRdS ZvdTdU ZwdVdW ZxG dXdY dYZyG dZd[ d[ZzG d\d] d]ej{Z|G d^d_ d_ej{Z}G d`d d Z~G dad
d
e~ZdkdbdcZeddkre dS )lzP
Read and write ZIP files.
XXX references to utf-8 need further investigation.
N
BadZipFile
BadZipfileerror
ZIP_STORED ZIP_DEFLATED ZIP_BZIP2ZIP_LZMA
is_zipfileZipInfoZipFile PyZipFile LargeZipFilec @ s e Zd ZdS )r N)__name__
__module__ __qualname__ r r /usr/lib64/python3.6/zipfile.pyr + s c @ s e Zd ZdZdS )r
zu
Raised when writing a zipfile, the zipfile requires ZIP64 extensions
and those extensions are disabled.
N)r r r __doc__r r r r r
/ s - . ? s <4s4H2LHs PK z<4s4B4HL2L5H2Ls PK
z <4s2B4HL2L2Hs PKz<4sLQLs PKz
<4sQ2H2L4Qs PKiPKz<HHc
C s t j}d}g }d }}xf|d t| kr||| ||d \}}|d | } ||krv||krn|j| || | }d}| }qW |s| S dj|S )NFr r T )_EXTRA_FIELD_STRUCTunpacklenappendjoin)
extraZxidsr- ZmodifiedbufferstartiZxidZxlenjr r r _strip_extra s r6 c C s* yt | rdS W n tk
r$ Y nX dS )NTF) _EndRecDataOSError)fpr r r _check_zipfile s r: c C sV d}y8t | dr t| d}n t| d}t|}W dQ R X W n tk
rP Y nX |S )zQuickly see if a file is a ZIP file by checking the magic number.
The filename argument may be a file or file-like object too.
Fread)r9 rbN)hasattrr: openr8 )filenameresultr9 r r r r s
c C s y| j |t d W n tk
r( |S X | jt}t|tkrD|S tjt|\}}}}|tkrd|S |dkst|dkr|t d| j |t t
d | jt
}t|t
kr|S tjt |\
}}} }
} } }
}}}|t kr|S ||t
< | |t< | |t< |
|t< ||t< ||t< ||t< |S )zM
Read the ZIP64 end-of-archive records and use that to update endrec
r r r z3zipfiles that span multiple disks are not supported)seeksizeEndCentDir64Locatorr8 r; r. structr- structEndArchive64LocatorstringEndArchive64Locatorr sizeEndCentDir64structEndArchive64stringEndArchive64_ECD_SIGNATURE_ECD_DISK_NUMBER_ECD_DISK_START_ECD_ENTRIES_THIS_DISK_ECD_ENTRIES_TOTAL _ECD_SIZE _ECD_OFFSET)fpinoffsetendrecdataZsigZdisknoZreloffZdisksZszcreate_versionZ read_versionZdisk_numZdisk_dirZdircountZ dircount2ZdirsizeZ diroffsetr r r
_EndRecData64 s6
rU c C sf | j dd | j }y| j t d W n tk
r: dS X | j }t|tkr|dd tkr|d d dkrtjt |}t
|}|j d |j |t t | t |S t
|d
t d}| j |d | j }|jt}|dkrb|||t }t|tkrdS t
tjt |}|t }||t |t | }|j | |j || t | || | |S dS ) zReturn data from the "End of Central Directory" record, or None.
The data is a list of the nine items in the ZIP "End of central dir"
record followed by a tenth item, the file seek offset of this record.r r Nr s r+ r r i )rA tellsizeEndCentDirr8 r; r. stringEndArchiverC r- structEndArchivelistr/ rU maxrfind_ECD_COMMENT_SIZE) rP ZfilesizerS rR ZmaxCommentStartr3 ZrecDataZ commentSizecommentr r r r7 s>
r7 c @ sV e Zd ZdZd)Zd+ddZd d Zd,d d Zd!d" Zd#d$ Z e
d-d%d&Z d'd( Z d S ).r
z>Class with attributes describing each file in the ZIP archive.
orig_filenamer? date_time
compress_typer_ r1
create_systemrT extract_versionreserved flag_bitsvolume
internal_attr
external_attr
header_offsetCRC
compress_size file_size _raw_time _end_offsetNoName r r c C s || _ |jtd}|dkr(|d| }tjdkrJtj|krJ|jtjd}|| _|| _|d dk rjtdt | _
d| _ d| _ t
jdkrd| _nd| _t| _t| _d| _d| _d| _d| _d| _d | _d S )Nr /i z+ZIP does not support timestamps before 1980r+ Zwin32r )r` findchrossepreplacer? ra
ValueErrorr rb r_ r1 sysplatformrc DEFAULT_VERSIONrT rd re rf rg rh ri ro )selfr? ra Z null_byter r r __init__X s0
zZipInfo.__init__c C s d| j j| jf g}| jtkr8|jdtj| j| j | jd? }| jd@ }|rd|jdt j
| |rv|jd| | j }| s| j r|jd| j | s| j
r| jtks| j | j
kr|jd| j
|jd d
j|S ) Nz<%s filename=%rz compress_type=%sr i z filemode=%rz external_attr=%#xz
file_size=%rz compress_size=%r> ) __class__r r? rb r r/ compressor_namesgetri statfilemodeis_dirrm rl r0 )r| r@ hiloisdirr r r __repr__ s(
zZipInfo.__repr__Nc C s| | j }|d d d> |d d> B |d B }|d d> |d d> B |d d B }| jd
@ rfd } }}n| j}| j}| j}| j}d} |d kr|tkp|tk}|rd }
|tj|
dtj |
d || }|tks|tkr|st
d
d}d}t } | j t
krtt| } n| j tkrtt| } t| | j| _t| | j| _| j \} } tjtt| j| j| | j |||||t| t|
}
|
| | S )z-Return the per-file header as a bytes object.r i r$ r r! r r r& r r Nz<HHQQz'Filesize would require ZIP64 extensionsl )ra rf rk rl rm r1 ZIP64_LIMITrC packcalcsizer
ZIP64_VERSIONrb r r\
BZIP2_VERSIONr LZMA_VERSIONrd rT _encodeFilenameFlagsstructFileHeaderstringFileHeaderre r. )r| zip64dtdosdatedostimerk rl rm r1 min_versionfmtr? rf headerr r r
FileHeader sF $$
zZipInfo.FileHeaderc C s> y| j jd| jfS tk
r8 | j jd| jdB fS X d S )Nasciizutf-8i )r? encoderf UnicodeEncodeError)r| r r r r s z ZipInfo._encodeFilenameFlagsc C s( | j }tj}xt|dkr"|d|d d \}}|dkr|dkrZ|d|dd }nV|dkrv|d|dd }n:|d
kr|d |dd }n |d
krf }ntd||f d
}| jdkr|| | _|d7 }| jdkr|| | _|d7 }| jdkr| j}|| | _|d7 }||d d }qW d S )Nr z<HHr z<QQQ r z<QQr r z<Qr r z"Corrupt extra field %04x (size=%d) )r r )r1 rC r- r. r rm rl rj )r| r1 r- tplnZcountsidxoldr r r _decodeExtra s4
zZipInfo._decodeExtrac C s t |tjrtj|}tj|}tj|j}tj|j }|dd }|dkrP|}tj
j tj
j |d }x$|d tj
tjfkr|dd }qjW |r|d7 }| ||}|jd@ d> |_|rd|_| jdO _n|j|_|S )a_ Construct an appropriate ZipInfo for a file on the filesystem.
filename should be the path to a file or directory on the filesystem.
arcname is the name which it will have within the archive (by default,
this will be the same as filename, but without a drive letter and with
leading path separators removed).
r r" Nr rr i r )
isinstanceru PathLikefspathr S_ISDIRst_modetime localtimest_mtimepathnormpath
splitdriverv altsepri rm st_size)clsr? arcnamestr mtimera zinfor r r from_file s(
zZipInfo.from_filec C s | j d dkS )z2Return True if this archive member is a directory.r rr )r? )r| r r r r s zZipInfo.is_dir)r` r? ra rb r_ r1 rc rT rd re rf rg rh ri rj rk rl rm rn ro rq r r r r r )rp r )N)N)
r r r r __slots__r} r r r r classmethodr r r r r r r
> s:
+
.$ !c @ s< e Zd ZdZdd ZdZdd Zdd Zd d
Zd d Z dS )
_ZipDecryptera Class to handle decryption of files stored within a ZIP archive.
ZIP supports a password-based form of encryption. Even though known
plaintext attacks have been found against it, it is still useful
to be able to get data out of such a file.
Usage:
zd = _ZipDecrypter(mypwd)
plain_char = zd(cypher_char)
plain_text = map(zd, cypher_text)
c C sh d} dgd }xTt dD ]H}|}x6t dD ]*}|d@ rH|d? d@ | A }q*|d? d@ }q*W |||< qW |S )zGenerate a CRC-32 table.
ZIP encryption uses the CRC32 one-byte primitive for scrambling some
internal keys. We noticed that a direct implementation is faster than
relying on binascii.crc32().
l q[ r r r i)range)Zpolytabler4 crcr5 r r r _GenerateCRCTable% s
z _ZipDecrypter._GenerateCRCTableNc C s |d? d@ | j ||A d@ A S )z(Compute the CRC32 primitive on one byte.r i )crctable)r| Zchr r r r _crc329 s z_ZipDecrypter._crc32c C sB t jd krt j t _d| _d| _d| _x|D ]}| j| q,W d S )NixV4igE#ixV4)r r r key0key1key2 _UpdateKeys)r| pwdpr r r r} = s
z_ZipDecrypter.__init__c C sX | j || j| _| j| jd@ d@ | _| jd d d@ | _| j | jd? d@ | j| _d S )Nr l ir r )r r r r )r| cr r r r F s z_ZipDecrypter._UpdateKeysc C s0 | j dB }|||dA d? d@ A }| j| |S )zDecrypt a single character.r r r r )r r )r| r kr r r __call__L s
z_ZipDecrypter.__call__)
r r r r r r r r} r r r r r r r s r c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
LZMACompressorc C s
d | _ d S )N)_comp)r| r r r r} W s zLZMACompressor.__init__c C sF t jdt ji}t jt jt jt j|gd| _tjdddt || S )Nid)filtersz<BBHr$ r )
lzmaZ_encode_filter_properties FILTER_LZMA1r
FORMAT_RAW_decode_filter_propertiesr rC r r. )r| Zpropsr r r _initZ s zLZMACompressor._initc C s* | j d kr | j | j j| S | j j|S )N)r r compress)r| rS r r r r a s
zLZMACompressor.compressc C s& | j d kr | j | j j S | j j S )N)r r flush)r| r r r r f s
zLZMACompressor.flushN)r r r r} r r r r r r r r U s r c @ s e Zd Zdd Zdd ZdS )LZMADecompressorc C s d | _ d| _d| _d S )Nr+ F)_decomp _unconsumedeof)r| r r r r} n s zLZMADecompressor.__init__c C s | j d kr| j|7 _t| jdkr*dS tjd| jdd \}t| jd| krXdS tjtjtjtj | jdd| gd| _ | jd| d }| `| j j
|}| j j | _ |S )Nr r+ z<Hr )r ) r r r. rC r- r r r r r
decompressr )r| rS Zpsizer@ r r r r s s
zLZMADecompressor.decompressN)r r r r} r r r r r r l s r ZstoreZshrinkreduceZimplodetokenizeZdeflateZ deflate64Zbzip2r ZterseZlz77ZwavpackZppmd)r r r r r r! r" r# r r$ r% r r r* a b c C sX | t kr
nJ| tkr tsTtdn4| tkr6tsTtdn | tkrLtsTtdntdd S )Nz.Compression requires the (missing) zlib modulez-Compression requires the (missing) bz2 modulez.Compression requires the (missing) lzma modulez(That compression method is not supported) r r zlib RuntimeErrorr bz2r r NotImplementedError) compressionr r r _check_compression s r c C s@ | t krtjtjtjdS | tkr*tj S | tkr8t S d S d S )Nr( i)
r r Z compressobjZZ_DEFAULT_COMPRESSIONZDEFLATEDr r Z
BZ2Compressorr r )rb r r r _get_compressor s
r c C sn | t kr d S | tkr tjdS | tkr.tj S | tkr<t S t j
| }|r\t d| |f nt d| f d S )Nr( zcompression type %d (%s)zcompression type %di) r r r Z
decompressobjr r ZBZ2Decompressorr r r r r )rb Zdescrr r r _get_decompressor s
r c @ s0 e Zd Zdd Zd ddZdddZd d
Zd S ) _SharedFilec C s2 || _ || _|| _|| _|| _|j| _|j| _d S )N)_file_pos_close_lock_writingseekablerW )r| fileposcloselockwritingr r r r} s z_SharedFile.__init__r c C sB | j 2 | j rtd| jj| j | jj | _| jS Q R X d S )Nz}Can't reposition in the ZIP file while there is an open writing handle on it. Close the writing handle before trying to read.)r r rx r rA r rW )r| rQ whencer r r rA s z_SharedFile.seekr c C sL | j < | j rtd| jj| j | jj|}| jj | _|S Q R X d S )NzyCan't read from the ZIP file while there is an open writing handle on it. Close the writing handle before trying to read.)r r rx r rA r r; rW )r| nrS r r r r; s z_SharedFile.readc C s$ | j d k r | j }d | _ | j| d S )N)r r )r| fileobjr r r r s
z_SharedFile.closeN)r r )r )r r r r} rA r; r r r r r r s
r c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
Zd S ) _Tellablec C s || _ d| _d S )Nr )r9 rQ )r| r9 r r r r} s z_Tellable.__init__c C s | j j|}| j|7 _|S )N)r9 writerQ )r| rS r r r r r s z_Tellable.writec C s | j S )N)rQ )r| r r r rW s z_Tellable.tellc C s | j j d S )N)r9 r )r| r r r r s z_Tellable.flushc C s | j j d S )N)r9 r )r| r r r r s z_Tellable.closeN)r r r r} r rW r r r r r r r s
r c s e Zd ZdZd&ZdZd'Zd(dd Zd
d Zd*d d
Z d+ddZ
dd Z d-ddZ dd Z
dd Zdd Zdd Z fd d Zd d Zd.d!d"Zd#d$ Z ZS )/
ZipExtFilezZFile-like object for reading an archive member.
Is returned by ZipFile.open().
r r i r NFc C s || _ || _|| _|j| _|j| _|j| _t | j| _
d| _ d| _ d| _
d | _| jd k rf| jd8 _|| _|j| _t|dr|j| _td| _nd | _d| _y4|j r|j | _|j| _|j| _ | j| _ d| _W n t k
r Y nX d S )NFr+ r r rk T) _fileobj
_decrypter_close_fileobjrb _compress_typerl _compress_leftrm _leftr
_decompressor_eof _readbuffer_offsetnewlinesmoder? namer= rk
_expected_crccrc32 _running_crc _seekabler rW _orig_compress_start_orig_compress_size_orig_file_size_orig_start_crcAttributeError)r| r r
zipinfoZ decrypterZ
close_fileobjr r r r} s:
zZipExtFile.__init__c C sv d| j j| j jf g}| jsX|jd| j| jf | jtkrb|jdt j
| j| j n
|jd |jd dj |S )Nz<%s.%sz name=%r mode=%rz compress_type=%sz [closed]r~ r ) r r r closedr/ r r
r r r r r0 )r| r@ r r r r ? s
zZipExtFile.__repr__c C sL |dk r>| j jd| jd }|dkr>| j | j| }|| _|S tjj| |S )zrRead and return a line from the stream.
If limit is specified, at most limit bytes will be read.
r
r )r rs r ioBufferedIOBasereadline)r| limitr4 liner r r r M s zZipExtFile.readlinec C sr |t | j| j kr\| j|}t || jkrJ|| j| jd | _d| _n| jt |8 _| j| j| jd S )z6Returns buffered bytes without advancing the position.Nr i )r. r r r; )r| r chunkr r r peek] s
zZipExtFile.peekc C s dS )NTr )r| r r r readablej s zZipExtFile.readablec C s |dks|dk rL| j | jd }d| _ d| _x| jsF|| j| j7 }q.W |S || j }|t| j k r~| j | j| }|| _|S |t| j }| j | jd }d| _ d| _x^|dko| j r| j|}|t|k r|| _ || _||d| 7 }P ||7 }|t|8 }qW |S )zRead and return up to n bytes.
If the argument is omitted, None, or negative, data is read and returned until EOF is reached..
Nr r+ )r r r _read1MAX_Nr. )r| r bufendrS r r r r; m s4
zZipExtFile.readc C s@ | j d krd S t|| j| _| jr<| j| j kr<td| j d S )NzBad CRC-32 for file %r)r r
r r r r )r| newdatar r r _update_crc s
zZipExtFile._update_crcc C s |dks|dk rV| j | jd }d| _ d| _x$| jsP| j| j}|r.||7 }P q.W |S || j }|t| j k r| j | j| }|| _|S |t| j }| j | jd }d| _ d| _|dkr xN| js
| j|}|t|k r|| _ || _||d| 7 }P |r||7 }P qW |S )z7Read up to n bytes with at most one read() system call.Nr r+ )r r r r r! r. )r| r r" rS r# r r r read1 s>
zZipExtFile.read1c C s" | j s|dkrdS | jtkrH| jj}|t|krR|| j|t| 7 }n
| j|}| jtkrj| jdk| _ nx| jtkrt || j
}| jj ||}| jj p| jdko| jj | _ | j r|| jj
7 }n | jj |}| jj p| jdk| _ |d | j }| jt|8 _| jdkrd| _ | j| |S )Nr r+ T)r r r r Zunconsumed_tailr. _read2r r r\
MIN_READ_SIZEr r r r r% )r| r rS r r r r s2
zZipExtFile._read1c C sj | j dkrdS t|| j}t|| j }| jj|}| j t|8 _ |sLt| jd k rft t
| j|}|S )Nr r+ ) r r\ r( minr r; r. EOFErrorr bytesmap)r| r rS r r r r' s
zZipExtFile._read2c
s& z| j r| jj W d t j X d S )N)r r r super)r| )r r r r s zZipExtFile.closec C s | j S )N)r )r| r r r r s zZipExtFile.seekabler c C s, | j stjd| j }|dkr&|}n.|dkr8|| }n |dkrL| j| }ntd|| jkrd| j}|dk rpd}|| }|| j }|dkr|t| jk r|| _d}nR|dk r| j j
| j | j | _
| j| _| j| _d| _d| _tj| j| _d| _|}x,|dkr"t| j|}| j| ||8 }qW | j S )Nz!underlying stream is not seekabler r r zCwhence must be os.SEEK_SET (0), os.SEEK_CUR (1), or os.SEEK_END (2)r+ F)r r UnsupportedOperationrW r rx r r. r r rA r r r r r r Zzipfiler r r r r)
MAX_SEEK_READr; )r| rQ r Zcurr_posZnew_posZ read_offsetZ buff_offsetZread_lenr r r rA sD
zZipExtFile.seekc C s0 | j stjd| j| j t| j | j }|S )Nz!underlying stream is not seekable)r r r. r r r. r r )r| Zfileposr r r rW * s
zZipExtFile.tell i @i )NFr )r )r r )r )r )r r r r r! r( r/ r} r r r r r; r% r& r r' r r rA rW
__classcell__r r )r r r s&
*
!
%$
/r c s@ e Zd Zdd Zedd Zdd Zdd Z fd d
Z Z S )
_ZipWriteFilec C s4 || _ || _|| _t|j| _d| _d| _d| _d S )Nr ) _zinfo_zip64_zipfiler rb _compressor
_file_size_compress_size_crc)r| zfr r r r r r} 2 s z_ZipWriteFile.__init__c C s | j jS )N)r5 r9 )r| r r r r ; s z_ZipWriteFile._fileobjc C s dS )NTr )r| r r r writable? s z_ZipWriteFile.writablec C sf | j rtdt|}| j|7 _t|| j| _| jrV| jj|}| jt|7 _| j j
| |S )Nz I/O operation on closed file.) r rx r. r7 r
r9 r6 r r8 r r )r| rS nbytesr r r r B s z_ZipWriteFile.writec sX | j r
d S t j | jrN| jj }| jt|7 _| jj| | j| j _
n
| j | j _
| j | j _
| j | j _| j jd@ r| jrdnd}| jjtj|t| j j
| j j
| j j | jj | j_nn| js| j tkrtd| jtkrtd| jj | j_| jj| j j | jj| j j| j | jj| jj d| j_ | jj j | j | j | jj | j j < d S )Nr z<LLQQz<LLLLz+File size unexpectedly exceeded ZIP64 limitz1Compressed size unexpectedly exceeded ZIP64 limitF)!r r- r r6 r r8 r. r r r3 rl r7 r9 rk rm rf r4 rC r
_DD_SIGNATURErW r5 start_dirr r rA rj r r filelistr/
NameToInfor? )r| r" r )r r r r N s8
z_ZipWriteFile.close)
r r r r} propertyr r; r r r1 r r )r r r2 1 s
r2 c @ s e Zd ZdZdZdZdedfddZdd Zd d
Z d d Z
d
d Z dd Z dd Z
d<ddZdd Zdd Zdd Zedd Zejd d Zd=d d Zd>d d!d"d#Zd?d$d%Zd@d&d'ZdAd(d)Zed*d+ Zd,d- Z d.d/ Z dBd0d1Z dCd2d3Z d4d5 Z d6d7 Z!d8d9 Z"d:d; Z#dS )Dr a Class with methods to open, read, write, close, list zip files.
z = ZipFile(file, mode="r", compression=ZIP_STORED, allowZip64=True)
file: Either the path to the file, or a file-like object.
If it is a path, the file will be opened and closed by ZipFile.
mode: The mode can be either read 'r', write 'w', exclusive create 'x',
or append 'a'.
compression: ZIP_STORED (no compression), ZIP_DEFLATED (requires zlib),
ZIP_BZIP2 (requires bz2) or ZIP_LZMA (requires lzma).
allowZip64: if True ZipFile will create files with ZIP64 extensions when
needed, otherwise it will raise an exception when this would
be necessary.
NrTc C sF |dkrt dt| || _d| _d| _i | _g | _|| _|| _d| _ d | _
t |t j
rdt j|}t |trd| _|| _d
d d d
d ddd}|| }x^ytj||| _W n( tk
r ||kr|| }w Y nX P qW nd| _|| _t|dd| _d| _tj | _d| _d| _ y|dkr*| j n|dkrd| _y| jj | _ W n2 t tfk
r| t!| j| _d| _ d| _Y n6X y| jj"| j W n t tfk
r d| _Y nX nf|dkry| j | jj"| j W n6 t#k
r | jj"dd d| _| jj | _ Y nX nt dW n$ | j}d| _| j$| Y nX dS )z]Open the ZIP file with mode read 'r', write 'w', exclusive create 'x',
or append 'a'.rB wxaz+ZipFile requires mode 'r', 'w', 'x', or 'a'Fr Nr+ r< zw+bzx+bzr+bwbZxb)rB rC rD rE zr+bzw+bzx+br r Tr z"Mode must be 'r', 'w', 'x', or 'a')rB rC rD rE )rC rD )%rx r _allowZip64
_didModifydebugr@ r? r r
r _commentr ru r r str _filePassedr? r r> r9 r8 getattr _fileRefCnt threadingRLockr r r _RealGetContentsrW r> r r rA r _fpclose)r| r r
r
allowZip64ZmodeDictr r9 r r r r} s
zZipFile.__init__c C s | S )Nr )r| r r r __enter__ s zZipFile.__enter__c C s | j d S )N)r )r| typevalue tracebackr r r __exit__ s zZipFile.__exit__c C s d| j j| j jf g}| jd k rd| jr8|jd| j n| jd k rR|jd| j |jd| j n
|jd |jd dj|S )Nz<%s.%sz file=%rz filename=%rz mode=%rz [closed]r~ r ) r r r r9 rL r/ r? r
r0 )r| r@ r r r r s
zZipFile.__repr__c C s | j }y t|}W n tk
r. tdY nX |s<td| jdkrNt| |t }|t }|t | _ |t
| | }|t t kr|t
t 8 }| jdkr|| }td||| || | _|j| jd |j|}tj|}d}x||k r|jt} t| tkrtdtjt| } | t tkr0td| jdkrDt| |j| t }
| d } | d @ rp|
j d
}
n
|
j d }
t |
} |j| t | _ |j| t | _!| t" | _#| dd \ | _$| _%| _&| _'| _(| _)}
}| _*| _+| _,| j&t-krt.d
| j&d | dd \| _/| _0| _1|
| _2|d? d |d? d@ |d@ |
d? |
d? d@ |
d@ d f| _3| j4 | j#| | _#| j5j6| | | j7| j8< |t | t | t | t }| jdkrtd| qW | j}x*t9| j5dd ddD ]}||_:|j#}qW dS ) z/Read in the table of contents for the ZIP file.zFile is not a zip filer r zgiven, inferred, offsetr zTruncated central directoryz&Bad magic number for central directoryr! i zutf-8cp437r zzip file version %.1fr% r( r* r$ i r r& r totalc S s | j S )N)rj )r r r r <lambda>I s z*ZipFile._RealGetContents.<locals>.<lambda>T)keyreverseN);r9 r7 r8 r rI printrN rO _ECD_COMMENTrJ
_ECD_LOCATIONrI rH rF rB r> rA r; r BytesIOsizeCentralDirr. rC r- structCentralDir
_CD_SIGNATUREstringCentralDir_CD_FILENAME_LENGTHdecoder
_CD_EXTRA_FIELD_LENGTHr1 _CD_COMMENT_LENGTHr_ _CD_LOCAL_HEADER_OFFSETrj rT rc rd re rf rb rk rl rm MAX_EXTRACT_VERSIONr rg rh ri rn ra r r? r/ r@ r? sortedro )r| r9 rR Zsize_cdZ offset_cdconcatZinferredrS rZ centdirr? flagsrD tdZ
end_offsetr r r r rQ s|
4
zZipFile._RealGetContentsc C s dd | j D S )z+Return a list of file names in the archive.c S s g | ]
}|j qS r )r? ).0rS r r r
<listcomp>P s z$ZipFile.namelist.<locals>.<listcomp>)r? )r| r r r namelistN s zZipFile.namelistc C s | j S )zJReturn a list of class ZipInfo instances for files in the
archive.)r? )r| r r r infolistR s zZipFile.infolistc C sP t dd
|d x:| jD ]0}d|jdd }t d |j||jf |d qW dS ) z+Print a table of contents for the zip file.z%-46s %19s %12s File Name Modified Size)r z%d-%02d-%02d %02d:%02d:%02dNr" z
%-46s %s %12d)rv rw rx )r^ r? ra r? rm )r| r r Zdater r r printdirW s zZipFile.printdirc C s^ d}xT| j D ]J}y.| j|jd}x|j|r0q$W W dQ R X W q tk
rT |jS X q W dS )z%Read all the files and check the CRC.r r rB Ni )r? r> r? r; r )r| Z
chunk_sizer fr r r testzip` s zZipFile.testzipc C s$ | j j|}|dkr td| |S )z,Return the instance of ZipInfo given 'name'.Nz(There is no item named %r in the archive)r@ r KeyError)r| r infor r r getinfom s
zZipFile.getinfoc C s8 |r"t |t r"tdt|j |r.|| _nd| _dS )z)Set default password for encrypted files.zpwd: expected bytes, got %sN)r r+ TypeErrorrU r r )r| r r r r setpasswordv s
zZipFile.setpasswordc C s | j S )z.The comment text associated with the ZIP file.)rJ )r| r r r r_ s zZipFile.commentc C s^ t |ts tdt|j t|tkrNdd l}|jdt dd |d t }|| _ d| _
d S )Nz comment: expected bytes, got %sr z3Archive comment is too long; truncating to %d bytesr )
stacklevelT) r r+ r rU r r. ZIP_MAX_COMMENTwarningswarnrJ rH )r| r_ r r r r r_ s
c C s" | j |d|
}|j S Q R X dS )zReturn file bytes for name.rB N)r> r; )r| r r r9 r r r r; s z ZipFile.readF) force_zip64c s |d"krt d|r2t|t r2tdt|j |rF|dkrFt d jsTt dt|trd|}n$|dkr~t|} j|_ n
j
|}|dkr j ||dS j rt d j
d 7 _
t j|j j j fd
d }y|jt}t|tk rtd tjt|}|t tkr td
|j|t }|t rF|j|t |j d@ rZt d|j d@ rnt d|j d@ r|j d} n
|j d} | |j krtd|j |f |j!dk r|j" |j# |j!krtd|j d|j d @ }
d} |
r|s j$}|st%d| t&|} |jd} t't(| | dd }
|j d @ rX|j)d ? d @ }n|j*d ? d @ }|
d |krt%d | t+|||| d!S |j, Y nX dS )#au Return file-like object for 'name'.
name is a string for the file name within the ZIP file, or a ZipInfo
object.
mode should be 'r' to read a file already in the ZIP file, or 'w' to
write to a file newly added to the archive.
pwd is the password to decrypt files (only used for reading).
When writing, if the file size is not known in advance but may exceed
2 GiB, pass force_zip64 to use the ZIP64 format, which can handle large
files. If the size is known in advance, it is best to pass a ZipInfo
instance for name, with zinfo.file_size set.
rB rC z open() requires mode "r" or "w"zpwd: expected bytes, got %sz'pwd is only supported for reading filesz2Attempt to use ZIP archive that was already closed)r zyCan't read from the ZIP file while there is an open writing handle on it. Close the writing handle before trying to read.r c s j S )N)r r )r| r r r[ s z ZipFile.open.<locals>.<lambda>zTruncated file headerz Bad magic number for file header z$compressed patched data (flag bit 5)@ z strong encryption (flag bit 6)i zutf-8rY z/File name in directory %r and header %r differ.NzOverlapped entries: z (possible zip bomb)z6File %r is encrypted, password required for extractionr r r r r r& zBad password for file %rT> rC rB )-rx r r+ r rU r r9 r
r rb r~ _open_to_writer rN r rj rR r r; sizeFileHeaderr. r rC r- r
_FH_SIGNATUREr _FH_FILENAME_LENGTH_FH_EXTRA_FIELD_LENGTHrf r rg r` ro rW rl r r r r[ r, rn rk r r )r| r r
r r r Zzef_fileZfheaderfnameZ fname_strZ is_encryptedZzdr hZ
check_byter )r| r r> s
z ZipFile.openc C s |r| j rtd| jr"tdt|ds2d|_d|_d|_d|_|jt kr\| jdO _| j
sp| jdO _|j s|d |_ | j o|p|jd t k}| j
r| j
j| j | j
j |_| j| d
| _| j
j|j| d
| _t| ||S ) NzHforce_zip64 is True, but allowZip64 was False when opening the ZIP file.zzCan't write to the ZIP file while there is another write handle open on it. Close the first handle before opening another.rm r r r i r g?Ti )rG rx r r= rm rl rk rf rb r r ri r r9 rA r> rW rj _writecheckrH r r r2 )r| r r r r r r r s4
zZipFile._open_to_writec C s* |dkrt j }n
t j|}| j|||S )a# Extract a member from the archive to the current working directory,
using its full name. Its file information is extracted as accurately
as possible. `member' may be a filename or a ZipInfo object. You can
specify a different directory using `path'.
N)ru getcwdr _extract_member)r| memberr r r r r extract6 s
zZipFile.extractc C sL |dkr| j }|dkr"tj }n
tj|}x|D ]}| j||| q2W dS )zExtract all members from the archive to the current working
directory. `path' specifies a different directory to extract to.
`members' is optional and must be a subset of the list returned
by namelist().
N)rt ru r r r )r| r membersr r r r r
extractallC s
zZipFile.extractallc C s^ | j }|s(d}tj|dt| }|| _ |j|}dd |j|D }|jdd |D }|S )z;Replace bad characters and remove trailing dots from parts.z:<>|"?*_c s s | ]}|j d V qdS ).N)rstrip)rr rD r r r <genexpr>^ s z1ZipFile._sanitize_windows_name.<locals>.<genexpr>c s s | ]}|r|V qd S )Nr )rr rD r r r r ` s )!_windows_illegal_name_trans_tablerK maketransr. translatesplitr0 )r r pathsepr Zillegalr r r _sanitize_windows_nameT s
z ZipFile._sanitize_windows_namec sN t |ts| j|}|jjdtjj}tjjrB|jtjjtjj}tjj |d }dtjj
tjj f tjjj fdd|j
tjjD }tjjdkr| j|tjj}tjj ||}tjj|}tjj|}|rtjj| rtj| |j rtjj|s
tj| |S | j||d(}t|d}tj|| W d Q R X W d Q R X |S )
zbExtract the ZipInfo object 'member' to a physical
file on the path targetpath.
rr r r c 3 s | ]}| kr|V qd S )Nr )rr rD )invalid_path_partsr r r t s z*ZipFile._extract_member.<locals>.<genexpr>\)r rF N)r r
r~ r? rw ru r rv r r curdirpardirr0 r r r dirnameexistsmakedirsr r mkdirr> shutil copyfileobj)r| r Z
targetpathr r Z upperdirssourcetargetr )r r r c s.
&
zZipFile._extract_memberc C s |j | jkr(ddl}|jd|j dd | jdkr:td | jsHtd
t|j | j sd}t
| j t krpd }n |j
tkrd }n|jtkrd
}|rt|d dS )z6Check for errors before writing a file to the archive.r NzDuplicate name: %rr )r rC rD rE z&write() requires mode 'w', 'x', or 'a'z4Attempt to write ZIP archive that was already closedz Files countZFilesizez Zipfile sizez would require ZIP64 extensions)rC rD rE )r? r@ r r r
rx r9 r rb rG r. r? ZIP_FILECOUNT_LIMITrm r rj r
)r| r r requires_zip64r r r r s(
zZipFile._writecheckc C s0 | j std| jr tdtj||}|j r>d|_d|_n|dk rN||_n| j |_|j r| j
| j rz| j j | j
| j j |_|jtkr| jdO _| j| d| _| jj| || j|j< | j j|jd | j j | _
W dQ R X n<t|d,}| j|d }tj ||d W dQ R X W dQ R X dS )
zLPut the bytes from filename into the archive under the name
arcname.z7Attempt to write to ZIP archive that was already closedz>Can't write to ZIP archive while an open writing handle existsr Nr TFr< rC i r i ) r9 rx r r
r r rl rk rb r r r rA r> rW rj r rf r rH r? r/ r@ r? r r r> r r )r| r? r rb r srcdestr r r r s8
z
ZipFile.writec C s t |tr|jd}t |tspt|tjtj dd d}| j|_|jd dkrhd|_ | j dO _ qtd|_ n|}| j
st d
| j rt d |dk r||_t
||_| j* | j|d d
}|j| W dQ R X W dQ R X dS )a Write a file into the archive. The contents is 'data', which
may be either a 'str' or a 'bytes' instance; if it is a 'str',
it is encoded as UTF-8 first.
'zinfo_or_arcname' is either a ZipInfo instance or
the name of the file in the archive.zutf-8Nr" )r? ra r rr iA r i z7Attempt to write to ZIP archive that was already closedz?Can't write to ZIP archive while an open writing handle exists.rC )r
r i Ai )r rK r r
r r r rb r? ri r9 rx r r. rm r r> r )r| Zzinfo_or_arcnamerS rb r r r r r writestr s.
zZipFile.writestrc C s | j dS )z2Call the "close()" method in case the user forgot.N)r )r| r r r __del__ s zZipFile.__del__c C s| | j dkrdS | jr tdzB| jdkr\| jr\| j" | jrJ| j j| j | j W dQ R X W d| j }d| _ | j
| X dS )zOClose the file, and for mode 'w', 'x' and 'a' write the ending
records.NzvCan't close the ZIP file while there is an open writing handle on it. Close the writing handle before closing the zip.rC rD rE )rC rD rE ) r9 r rx r
rH r r rA r> _write_end_recordrR )r| r9 r r r r s
z
ZipFile.closec C sr xP| j D ]D}|j}|d d d> |d d> B |d B }|d d> |d d> B |d d B }g }|jtksv|jtkr|j|j |j|j d
}d
}n |j}|j}|jtkr|j|j d
}n|j}|j} d}
|rt| d} t j
d d t | dd
t | f| | } t }
|j
tkr(tt|
}
n|j
tkr>tt|
}
t|
|j} t|
|j} yZ|j \}
}t j
tt| |j| |j||j
|||j||t |
t | t |jd|j |j |}W nl t k
r t tt| |j| |j|j |j
|||j||t |j!t | t |jd|j |j |ft"j#d Y nX | j$j%| | j$j%|
| j$j%| | j$j%|j q
W | j$j& }t | j }|| j' }| j'}d }|t(krd}n |tkrd}n|tkrd}|r*| j)st*|d t j
t+t,ddddd|||| }| j$j%| t j
t-t.d|d}| j$j%| t/|d}t/|d
}t/|d
}t j
t0t1dd||||t | j2 }| j$j%| | j$j%| j2 | j$j3 d S )Nr i r$ r r! r r r& r l z<HHQr )r z Files countzCentral directory offsetzCentral directory sizez would require ZIP64 extensions, r i )r )4r? ra rm r rl r/ rj r1 r6 rC r r. r rb r r\ r r r rd rT r rc re rc re rk r_ rh ri DeprecationWarningr^ rf r? ry stderrr9 r rW r> r rG r
rG rH rD rE r) rZ rY rJ r )r| r r r r r1 rm rl rj Z
extra_datar rd rT r? rf rn Zpos2Z centDirCountZ centDirSizeZ
centDirOffsetr Z zip64endrecZ zip64locrecrR r r r r s $$
zZipFile._write_end_recordc C s* | j d8 _ | j r&| j r&|j d S )Nr )rN rL r )r| r9 r r r rR r s zZipFile._fpclose)N)N)rB N)F)NN)NNN)NN)N)$r r r r r9 r r r} rT rX r rQ rt ru ry r{ r~ r rA r_ setterr; r> r r r r r r r r r r r r rR r r r r r { s< WT
u
+
*
*
$fc @ s4 e Zd ZdZdedd
fddZdd d
Zd d ZdS )r zDClass to create ZIP archives with Python library files and packages.rB Tr c C s t j| ||||d || _d S )N)r
r rS )r r} _optimize)r| r r
r rS optimizer r r r} | s zPyZipFile.__init__r Nc C s t j|}|rF|| rF| jrBt jj|r.dnd}td||f dS t jj|\}}t jj|rnt jj|d}t jj|r|rd||f }n|}| jrtd|d| | j |d d |\}} | jrtd | | j
|| t j |}
|
j d x|
D ]} t jj|| } t jj
| \}
}t jj| rRt jjt jj| dr| j| ||d q|d
kr|r|| r| jrtd| q| j | d d |\}} | jrtd | | j
|| qW n| jrtd| xt j |D ]} t jj|| } t jj
| \}
}|d
kr|r0|| r0| jrtd| q| j | d d |\}} | jrZtd | | j
|| qW nP|dd d
krtd| j |d d |\}} | jrtd| | j
|| dS )a Add all files from "pathname" to the ZIP archive.
If pathname is a package directory, search the directory and
all package subdirectories recursively for all *.py and enter
the modules into the archive. If pathname is a plain
directory, listdir *.py and enter all modules. Else, pathname
must be a Python *.py file and the module will be put into the
archive. Added modules are always module.pyc.
This method will compile the module.py into module.pyc if
necessary.
If filterfunc(pathname) is given, it is called with every argument.
When it is False, the file or directory is skipped.
r r z%s %r skipped by filterfuncNz __init__.pyz%s/%szAdding package inasr r ZAdding)
filterfuncz.pyz file %r skipped by filterfunczAdding files from directoryz.Files added with writepy() must end with ".py"z Adding filer r r r )ru r rI r r r^ r r0 isfile
_get_codenamer listdirremovesplitextwritepyr )r| pathnamebasenamer Zlabeldirr Zinitnamer r Zdirlistr? r rootZextr r r r sx
zPyZipFile.writepyc
s d fdd }|d }|d }t jj|dd}t jj|dd}t jj|dd} jdkr\tjj|rtj|jtj|jkr| } }
ntjj|rtj|jtj|jkr|}
|} ntjj|rtj|jtj|jkr|}
|} nvtjj|rtj|jtj|jkr|}
|} nD||rRt j
j d kr4|}
nt j
j dkrH|}
n|}
|} n| }
} n jd krr|}
|} n<|} jdkr|}
n& jdkr|}
nd
j j} t
| tjj|
otj|
jtj|jks|| jd s| }
} tjj| d } |rd || f } |
| fS )a Return (filename, archivename) for the path.
Given a module name path, return the correct file path and
archive name, compiling if necessary. For example, given
/python/lib/string, return (/python/lib/string.pyc, string).
r c s^ dd l } jrtd| y|j| d|d W n, |jk
rX } zt|j dS d }~X nX dS )Nr Z CompilingT)doraiser F)
py_compilerI r^ compilePyCompileErrormsg)r r r err)r| r r _compile s
z)PyZipFile._get_codename.<locals>._compilez.pyz.pycr ) optimizationr r z"invalid value for 'optimize': {!r})r z%s/%sr )r r ) importlibutilcache_from_sourcer ru r r r r ry ro r formatrx r )
r| r r r Zfile_pyZfile_pycZ pycache_opt0Z pycache_opt1Z pycache_opt2r r r Z archivenamer )r| r r s`
zPyZipFile._get_codenamer )r N)r r r r r r} r r r r r r r y s
Rc s dd l }|jd}| d kr(tjdd } | s:| d dkrLt| tjd | d dkrt| dkrvt| tjd t| d d }|j W d Q R X n|| d dkr
t| dkrt| tjd t| d d }|j }W d Q R X | rtd
j
| td n| d dkrbt| d kr8t| tjd t| d d }|j | d W d Q R X n| d dkrt| d k rt| tjd fd
d t| d dh}x`| dd D ]P}t j
j|}|st j
jt j
j|}|dt jt jfkrd} ||| qW W d Q R X d S )Nr aE Usage:
zipfile.py -l zipfile.zip # Show listing of a zipfile
zipfile.py -t zipfile.zip # Test if a zipfile is valid
zipfile.py -e zipfile.zip target # Extract zipfile into target dir
zipfile.py -c zipfile.zip src ... # Create zipfile from sources
r -l-c-e-tr rB z.The following enclosed file is corrupted: {!r}z Done testingr c sp t jj|r | j||t nPt jj|rl|r8| j|| x2t j|D ]$} | t jj||t jj|| qDW d S )N)ru r r r r r r r0 )r: r zippathZnm)addToZipr r r U s zmain.<locals>.addToZiprC r )r r r r )textwrapdedentry argvr^ exitr. r ry r{ r r ru r r r r r )argsr ZUSAGEr: Zbadfiler r r )r r main' sT
r __main__l ii i i i )N)r r ru reimportlib.utilr ry r r r rC ZbinasciirO ImportErrorZdummy_threadingr r
r r __all__ Exceptionr r
r r r r r r r r r r{ r r r rk rZ rY r rX rI rJ rK rL rM rN rO r^ r_ r` rc re rb rd Z_CD_CREATE_VERSIONZ_CD_CREATE_SYSTEMZ_CD_EXTRACT_VERSIONZ_CD_EXTRACT_SYSTEMZ
_CD_FLAG_BITSZ_CD_COMPRESS_TYPEZ_CD_TIMEZ_CD_DATEZ_CD_CRCZ_CD_COMPRESSED_SIZEZ_CD_UNCOMPRESSED_SIZErf rh ri Z_CD_DISK_NUMBER_STARTZ _CD_INTERNAL_FILE_ATTRIBUTESZ _CD_EXTERNAL_FILE_ATTRIBUTESrj r r r r Z_FH_EXTRACT_VERSIONZ_FH_EXTRACT_SYSTEMZ _FH_GENERAL_PURPOSE_FLAG_BITSZ_FH_COMPRESSION_METHODZ_FH_LAST_MOD_TIMEZ_FH_LAST_MOD_DATEZ_FH_CRCZ_FH_COMPRESSED_SIZEZ_FH_UNCOMPRESSED_SIZEr r rD rE rB rG rH rF Z_CD64_SIGNATUREZ_CD64_DIRECTORY_RECSIZEZ_CD64_CREATE_VERSIONZ_CD64_EXTRACT_VERSIONZ_CD64_DISK_NUMBERZ_CD64_DISK_NUMBER_STARTZ _CD64_NUMBER_ENTRIES_THIS_DISKZ_CD64_NUMBER_ENTRIES_TOTALZ_CD64_DIRECTORY_SIZEZ_CD64_OFFSET_START_CENTDIRr= ZStructr, r6 r: r rU r7 objectr
r r r r r r r r r r r r2 r r r r r r r r <module> sD
+= [= & -J /
B
All system for education purposes only. For more tools: Telegram @jackleet