HEX

File: //usr/lib/python3.7/__pycache__/zipfile.cpython-37.pyc
B

��{a�9�@s�dZddlZddlZddlZddlZddlZddlZddlZddl	Z	ddl
Z
ddlZyddlZej
Z
Wnek
r�dZe
j
Z
YnXyddlZWnek
r�dZYnXyddlZWnek
r�dZYnXddddddd	d
ddd
dgZGdd�de�ZGdd�de�ZeZZdZdZdZdZdZdZdZdZdZdZ dZ!dZ"dZ#dZ$e	�%e#�Z&dZ'dZ(dZ)dZ*dZ+d Z,d!Z-d"Z.dZ/d#Z0d$Z1d%Z2e	�%e1�Z3dZ4dZ5dZ6dZ7dZ8d Z9d!Z:d"Z;dZ<d#Z=d&Z>d'Z?dZ@d(ZAdZBd)ZCd*ZDd+ZEd,ZFd-ZGd.ZHe	�%eG�ZIdZJdZKdZLdZMdZNd ZOd!ZPd"ZQdZRd#ZSd&ZTd'ZUd/ZVd0ZWe	�%eV�ZXd1ZYd2ZZe	�%eY�Z[dZ\dZ]dZ^dZ_dZ`d Zad!Zbd"ZcdZdd#Zed3Zfe	�gd4�Zhd5d6�Zid7d8�Zjd9d
�Zkd:d;�Zld<d=�ZmGd>d�den�Zodapd?d@�ZqdAdB�ZrGdCdD�dD�ZsGdEdF�dF�ZtdGdHdIdIdIdIdJdKdLdMdJdNdOdPdQdRdSdT�ZudUdV�ZvdhdWdX�ZwdYdZ�ZxGd[d\�d\�ZyGd]d^�d^�ZzGd_d`�d`ej{�Z|Gdadb�dbej{�Z}Gdcd�d�Z~Gddd
�d
e~�Zdidedf�Z�e�dgk�r�e��dS)jzP
Read and write ZIP files.

XXX references to utf-8 need further investigation.
�N�
BadZipFile�
BadZipfile�error�
ZIP_STORED�ZIP_DEFLATED�	ZIP_BZIP2�ZIP_LZMA�
is_zipfile�ZipInfo�ZipFile�	PyZipFile�LargeZipFilec@seZdZdS)rN)�__name__�
__module__�__qualname__�rr�/usr/lib/python3.7/zipfile.pyr&sc@seZdZdZdS)r
zu
    Raised when writing a zipfile, the zipfile requires ZIP64 extensions
    and those extensions are disabled.
    N)rrr�__doc__rrrrr
*si���i�������-�.�?s<4s4H2LHsPK��������	z<4s4B4HL2L5H2LsPK�
��
����z<4s2B4HL2L2HsPKz<4sLQLsPKz
<4sQ2H2L4QsPKiPKz<HHc
Cs�tj}d}g}d}}xf|dt|�kr|||||d��\}}|d|}	||krv||krn|�|||��|	}d}|	}qW|s�|Sd�|�S)NFrrT�)�_EXTRA_FIELD_STRUCT�unpack�len�append�join)
�extraZxidsr,Zmodified�buffer�start�iZxidZxlen�jrrr�_strip_extra�s r5cCs*yt|�rdSWntk
r$YnXdS)NTF)�_EndRecData�OSError)�fprrr�_check_zipfile�sr9c	CsVd}y8t|d�rt|d�}nt|d��}t|�}WdQRXWntk
rPYnX|S)z�Quickly 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.
    F�read)r8�rbN)�hasattrr9�openr7)�filename�resultr8rrrr	�s
c
Csy|�|td�Wntk
r(|SX|�t�}t|�tkrD|St�t|�\}}}}|tkrd|S|dkst|dkr|t	d��|�|tt
d�|�t
�}t|�t
kr�|St�t|�\
}}}	}
}}}
}}}|tkr�|S||t
<||t<||t<|
|t<||t<||t<||t<|S)zM
    Read the ZIP64 end-of-archive records and use that to update endrec
    rrrz3zipfiles that span multiple disks are not supported)�seek�sizeEndCentDir64Locatorr7r:r-�structr,�structEndArchive64Locator�stringEndArchive64Locatorr�sizeEndCentDir64�structEndArchive64�stringEndArchive64�_ECD_SIGNATURE�_ECD_DISK_NUMBER�_ECD_DISK_START�_ECD_ENTRIES_THIS_DISK�_ECD_ENTRIES_TOTAL�	_ECD_SIZE�_ECD_OFFSET)�fpin�offset�endrec�dataZsigZdisknoZreloffZdisksZsz�create_versionZread_versionZdisk_numZdisk_dirZdircountZ	dircount2ZdirsizeZ	diroffsetrrr�
_EndRecData64�s6

 rTc	Csf|�dd�|��}y|�td�Wntk
r:dSX|��}t|�tkr�|dd�tkr�|dd�dkr�t�t	|�}t
|�}|�d�|�|t�t|t|�St
|dtd�}|�|d�|��}|�t�}|dk�rb|||t�}t|�tk�rdSt
t�t	|��}|t}||t|t|�}|�|�|�||�t|||||�SdS)	z�Return 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.rrNr�����sr*i)r@�tell�sizeEndCentDirr7r:r-�stringEndArchiverBr,�structEndArchive�listr.rT�max�rfind�_ECD_COMMENT_SIZE)	rOZfilesizerRrQZmaxCommentStartr2ZrecDataZcommentSize�commentrrrr6�s>



r6c@sVeZdZdZdZddd�Zdd�Zdd
d�Zdd
�Zdd�Z	e
ddd��Zdd�Zd	S)r
z>Class with attributes describing each file in the ZIP archive.)�
orig_filenamer>�	date_time�
compress_type�_compresslevelr^r0�
create_systemrS�extract_version�reserved�	flag_bits�volume�
internal_attr�
external_attr�
header_offset�CRC�
compress_size�	file_size�	_raw_time�NoName�i�rrrrrcCs�||_|�td��}|dkr(|d|�}tjdkrJtj|krJ|�tjd�}||_||_|ddkrjtd��t	|_
d|_d|_d|_
tjdkr�d|_nd|_t|_t|_d|_d|_d|_d|_d|_dS)Nr�/i�z+ZIP does not support timestamps before 1980r*Zwin32r)r_�find�chr�os�sep�replacer>r`�
ValueErrorrrarbr^r0�sys�platformrc�DEFAULT_VERSIONrSrdrerfrgrhri)�selfr>r`Z	null_byterrr�__init__Ss0
zZipInfo.__init__cCs�d|jj|jfg}|jtkr8|�dt�|j|j��|jd?}|jd@}|rd|�dt	�
|��|rv|�d|�|��}|r�|jr�|�d|j�|r�|j
r�|jtks�|j|j
kr�|�d|j
�|�d	�d
�|�S)Nz<%s filename=%rz compress_type=%sr'i��z filemode=%rz external_attr=%#xz
 file_size=%rz compress_size=%r�>�)�	__class__rr>rarr.�compressor_names�getri�stat�filemode�is_dirrmrlr/)r{r?�hi�lo�isdirrrr�__repr__~s(






zZipInfo.__repr__NcCs||j}|ddd>|dd>B|dB}|dd>|d	d>B|ddB}|jd
@rfd}}}n|j}|j}|j}|j}d}	|dkr�|tkp�|tk}|r�d}
|t�|
dt�	|
�d	||�}|tks�|tkr�|s�t
d
��d}d}t}	|jt
k�rtt|	�}	n|jtk�rtt|	�}	t|	|j�|_t|	|j�|_|��\}}t�tt|j|j||j|||||t|�t|��
}
|
||S)z-Return the per-file header as a bytes object.ri�r"rrrrr$rrNz<HHQQz'Filesize would require ZIP64 extensionsl��)r`rfrkrlrmr0�ZIP64_LIMITrB�pack�calcsizer
�
ZIP64_VERSIONrarr[�
BZIP2_VERSIONr�LZMA_VERSIONrdrS�_encodeFilenameFlags�structFileHeader�stringFileHeaderrer-)r{�zip64�dt�dosdate�dostimerkrlrmr0�min_version�fmtr>rf�headerrrr�
FileHeader�sF$$



zZipInfo.FileHeadercCs>y|j�d�|jfStk
r8|j�d�|jdBfSXdS)N�asciizutf-8i)r>�encoderf�UnicodeEncodeError)r{rrrr��szZipInfo._encodeFilenameFlagscCsJ|j}tj}�x6t|�dk�rD|d|dd��\}}|dt|�krTtd||f��|dk�r2|dkrz|d|dd��}nV|dkr�|d	|dd
��}n:|dkr�|d|dd
��}n|dkr�d}ntd||f��d}|jdkr�|||_|d7}|jdk�r|||_|d7}|jdk�r2|j}|||_|d7}||dd�}qWdS)Nrz<HHz"Corrupt extra field %04x (size=%d)r�z<QQQ�r'z<QQrrz<Qrrr)l����l��l��)r0rBr,r-rrmrlrj)r{r0r,�tpZlnZcounts�idx�oldrrr�_decodeExtra�s8




zZipInfo._decodeExtracCs�t|tj�rt�|�}t�|�}t�|j�}t�|j	�}|dd�}|dkrP|}tj
�tj
�|�d�}x$|dtj
tjfkr�|dd�}qjW|r�|d7}|||�}|jd@d>|_|r�d|_|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).
        rr Nrrqi��r')�
isinstancert�PathLike�fspathr��S_ISDIR�st_mode�time�	localtime�st_mtime�path�normpath�
splitdriveru�altseprirm�st_size)�clsr>�arcname�str��mtimer`�zinforrr�	from_file�s(



zZipInfo.from_filecCs|jddkS)z2Return True if this archive member is a directory.�����rq)r>)r{rrrr�szZipInfo.is_dir)rorp)N)N)
rrrr�	__slots__r|r�r�r�r��classmethodr�r�rrrrr
9s
+
.&!cCs4x.td�D]"}|d@r$|d?dA}q
|dL}q
W|S)Nrrl q[)�range)�crcr4rrr�_gen_crcs
r�cspd�d�d�tdkr&ttttd���at��fdd������fdd��x|D]}�|�qNW��fd	d
�}|S)NixV4i�gE#i�xV4�cs|d?�||Ad@AS)z(Compute the CRC32 primitive on one byte.r��r)Zchr�)�crctablerr�crc324sz_ZipDecrypter.<locals>.crc32cs<�|�����d@d@��ddd@���d?���dS)Nr�l��i�rr�r)�c)r��key0�key1�key2rr�update_keys8s
z"_ZipDecrypter.<locals>.update_keyscsRt�}|j}x<|D]4}�dB}|||dAd?d@N}�|�||�qWt|�S)zDecrypt a bytes object.rrrr�)�	bytearrayr.�bytes)rRr?r.r��k)r�r�rr�	decrypterBs
z _ZipDecrypter.<locals>.decrypter)�	_crctablerZ�mapr�r�)�pwd�pr�r)r�r�r�r�r�r�r�
_ZipDecrypter*s
r�c@s,eZdZdd�Zdd�Zdd�Zdd�Zd	S)
�LZMACompressorcCs
d|_dS)N)�_comp)r{rrrr|RszLZMACompressor.__init__cCsFt�dtji�}tjtjt�tj|�gd�|_t�dddt	|��|S)N�id)�filtersz<BBHr"r)
�lzmaZ_encode_filter_properties�FILTER_LZMA1r��
FORMAT_RAW�_decode_filter_propertiesr�rBr�r-)r{Zpropsrrr�_initUszLZMACompressor._initcCs*|jdkr|��|j�|�S|j�|�S)N)r�r��compress)r{rRrrrr�\s
zLZMACompressor.compresscCs&|jdkr|��|j��S|j��S)N)r�r��flush)r{rrrr�as
zLZMACompressor.flushN)rrrr|r�r�r�rrrrr�Psr�c@seZdZdd�Zdd�ZdS)�LZMADecompressorcCsd|_d|_d|_dS)Nr*F)�_decomp�_unconsumed�eof)r{rrrr|iszLZMADecompressor.__init__c	Cs�|jdkr�|j|7_t|j�dkr*dSt�d|jdd��\}t|j�d|krXdStjtjt�tj	|jdd|��gd�|_|jd|d�}|`|j�
|�}|jj|_|S)Nrr*z<Hr)r�)r�r�r-rBr,r�r�r�r�r��
decompressr�)r{rRZpsizer?rrrr�ns

zLZMADecompressor.decompressN)rrrr|r�rrrrr�gsr�ZstoreZshrink�reduceZimplode�tokenizeZdeflateZ	deflate64Zbzip2r�ZterseZlz77ZwavpackZppmd)rrrrrrr r!rr"r#rrr)��a�bcCsX|tkr
nJ|tkr tsTtd��n4|tkr6tsTtd��n|tkrLtsTtd��ntd��dS)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)	rr�zlib�RuntimeErrorr�bz2rr��NotImplementedError)�compressionrrr�_check_compression�sr�cCsj|tkr2|dk	r t�|tjd�St�tjtjd�S|tkrT|dk	rLt�|�St��S|tkrbt	�SdSdS)Ni����)
rr�ZcompressobjZDEFLATEDZZ_DEFAULT_COMPRESSIONrr�Z
BZ2Compressorrr�)ra�
compresslevelrrr�_get_compressor�s
r�cCsn|tkrdS|tkrt�d�S|tkr.t��S|tkr<t�St	�
|�}|r\td||f��ntd|f��dS)Ni����zcompression type %d (%s)zcompression type %d)rrr�Z
decompressobjrr�ZBZ2Decompressorrr�r�r�r�)raZdescrrrr�_get_decompressor�s

r�c@s0eZdZdd�Zddd�Zd
dd�Zd	d
�ZdS)�_SharedFilecCs2||_||_||_||_||_|j|_|j|_dS)N)�_file�_pos�_close�_lock�_writing�seekablerV)r{�file�pos�close�lockZwritingrrrr|�sz_SharedFile.__init__rc	CsB|j�2|��rtd��|j�||�|j��|_|jSQRXdS)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�rwr�r@rVr�)r{rP�whencerrrr@�sz_SharedFile.seekr�c	CsL|j�<|��rtd��|j�|j�|j�|�}|j��|_|SQRXdS)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�rwr�r@r�r:rV)r{�nrRrrrr:�sz_SharedFile.readcCs$|jdk	r |j}d|_|�|�dS)N)r�r�)r{�fileobjrrrr��s
z_SharedFile.closeN)r)r�)rrrr|r@r:r�rrrrr��s	


r�c@s4eZdZdd�Zdd�Zdd�Zdd�Zd	d
�ZdS)�	_TellablecCs||_d|_dS)Nr)r8rP)r{r8rrrr|�sz_Tellable.__init__cCs|j�|�}|j|7_|S)N)r8�writerP)r{rRr�rrrr��sz_Tellable.writecCs|jS)N)rP)r{rrrrV�sz_Tellable.tellcCs|j��dS)N)r8r�)r{rrrr��sz_Tellable.flushcCs|j��dS)N)r8r�)r{rrrr�sz_Tellable.closeN)rrrr|r�rVr�r�rrrrr��s
r�cs�eZdZdZdZdZdZd&dd�Zd	d
�Zd'dd
�Z	d(dd�Z
dd�Zd)dd�Zdd�Z
dd�Zdd�Zdd�Z�fdd�Zdd �Zd*d"d#�Zd$d%�Z�ZS)+�
ZipExtFilezZFile-like object for reading an archive member.
       Is returned by ZipFile.open().
    i@iiNFcCs�||_||_||_|j|_|j|_|j|_t	|j�|_
d|_d|_d|_
d|_|jdk	rf|jd8_||_|j|_t|d�r�|j|_td�|_nd|_d|_y4|��r�|��|_|j|_|j|_|j|_d|_Wntk
r�YnXdS)NFr*rrrkT)�_fileobj�
_decrypter�_close_fileobjra�_compress_typerl�_compress_leftrm�_leftr��
_decompressor�_eof�_readbuffer�_offset�newlines�moder>�namer<rk�
_expected_crcr��_running_crc�	_seekabler�rV�_orig_compress_start�_orig_compress_size�_orig_file_size�_orig_start_crc�AttributeError)r{r�r�zipinfor�Z
close_fileobjrrrr|s:



zZipExtFile.__init__cCsvd|jj|jjfg}|jsX|�d|j|jf�|jtkrb|�dt	�
|j|j��n
|�d�|�d�d�|�S)Nz<%s.%sz name=%r mode=%rz compress_type=%sz	 [closed]r}r~)rrr�closedr.r	rrrr�r�r/)r{r?rrrr�>s


zZipExtFile.__repr__r�cCsL|dkr>|j�d|j�d}|dkr>|j|j|�}||_|Stj�||�S)zrRead and return a line from the stream.

        If limit is specified, at most limit bytes will be read.
        r�
r)rrrr�io�BufferedIOBase�readline)r{�limitr3�linerrrrLszZipExtFile.readlinercCsr|t|j�|jkr\|�|�}t|�|jkrJ||j|jd�|_d|_n|jt|�8_|j|j|jd�S)z6Returns buffered bytes without advancing the position.Nri)r-rrr:)r{r��chunkrrr�peek\s
zZipExtFile.peekcCsdS)NTr)r{rrr�readableiszZipExtFile.readablecCs|dks|dkrL|j|jd�}d|_d|_x|jsF||�|j�7}q.W|S||j}|t|j�kr~|j|j|�}||_|S|t|j�}|j|jd�}d|_d|_x^|dk�r|j�s|�|�}|t|�kr�||_||_||d|�7}P||7}|t|�8}q�W|S)z�Read and return up to n bytes.
        If the argument is omitted, None, or negative, data is read and returned until EOF is reached..
        Nrr*)rrr�_read1�MAX_Nr-)r{r��buf�endrRrrrr:ls4

zZipExtFile.readcCs@|jdkrdSt||j�|_|jr<|j|jkr<td|j��dS)NzBad CRC-32 for file %r)r
r�rrrr	)r{�newdatarrr�_update_crc�s

zZipExtFile._update_crccCs|dks|dkrV|j|jd�}d|_d|_x$|jsP|�|j�}|r.||7}Pq.W|S||j}|t|j�kr�|j|j|�}||_|S|t|j�}|j|jd�}d|_d|_|dk�rxN|j�s
|�|�}|t|�kr�||_||_||d|�7}P|r�||7}Pq�W|S)z7Read up to n bytes with at most one read() system call.Nrr*)rrrrrr-)r{r�rrRr rrr�read1�s>



zZipExtFile.read1cCs"|js|dkrdS|jtkrH|jj}|t|�krR||�|t|��7}n
|�|�}|jtkrj|jdk|_nx|jtkr�t	||j
�}|j�||�}|jjp�|jdko�|jj|_|jr�||j�
�7}n |j�|�}|jjp�|jdk|_|d|j�}|jt|�8_|jdk�rd|_|�|�|S)Nrr*T)rrrrZunconsumed_tailr-�_read2rrr[�
MIN_READ_SIZEr�r�r�rr")r{r�rRrrrr�s2





zZipExtFile._read1cCsd|jdkrdSt||j�}t||j�}|j�|�}|jt|�8_|sLt�|jdk	r`|�|�}|S)Nrr*)	rr[r%�minr�r:r-�EOFErrorr�)r{r�rRrrrr$�s


zZipExtFile._read2cs&z|jr|j��Wdt���XdS)N)r�r�r��super)r{)rrrr��szZipExtFile.closecCs|jS)N)r)r{rrrr��szZipExtFile.seekablercCs*|jst�d��|��}|dkr&|}n.|dkr8||}n|dkrL|j|}ntd��||jkrd|j}|dkrpd}||}||j}|dkr�|t|j�kr�||_d}nP|dkr�|j	�
|j�|j|_
|j|_|j|_d|_d|_t|j�|_d|_|}x,|dk�r t|j|�}|�|�||8}q�W|��S)Nz!underlying stream is not seekablerrrzCwhence must be os.SEEK_SET (0), os.SEEK_CUR (1), or os.SEEK_END (2)r*F)rr�UnsupportedOperationrVrrwrr-rr�r@r
rrrrrr�rrrr&�
MAX_SEEK_READr:)r{rPr�Zcurr_posZnew_posZread_offsetZbuff_offsetZread_lenrrrr@�sD




zZipExtFile.seekcCs0|jst�d��|j|jt|j�|j}|S)Nz!underlying stream is not seekable)rrr)rrr-rr)r{ZfileposrrrrV(s
zZipExtFile.tell)NF)r�)r)r�)r)rrrrrr%r*r|r�rrrr:r"r#rr$r�r�r@rV�
__classcell__rr)rrr�s&
*



!
%$
.r�cs@eZdZdd�Zedd��Zdd�Zdd�Z�fd	d
�Z�Z	S)�
_ZipWriteFilecCs8||_||_||_t|j|j�|_d|_d|_d|_	dS)Nr)
�_zinfo�_zip64�_zipfiler�rarb�_compressor�
_file_size�_compress_size�_crc)r{�zfr�r�rrrr|0s
z_ZipWriteFile.__init__cCs|jjS)N)r/r8)r{rrrr�:sz_ZipWriteFile._fileobjcCsdS)NTr)r{rrr�writable>sz_ZipWriteFile.writablecCsf|jrtd��t|�}|j|7_t||j�|_|jrV|j�|�}|jt|�7_|j	�
|�|S)NzI/O operation on closed file.)rrwr-r1r�r3r0r�r2r�r�)r{rR�nbytesrrrr�Asz_ZipWriteFile.writec	sX|jr
dSt���|jrN|j��}|jt|�7_|j�|�|j|j	_
n
|j|j	_
|j|j	_
|j|j	_|j	jd@r�|jr�dnd}|j�t�|t|j	j
|j	j
|j	j��|j��|j_nn|js�|jtkr�td��|jtkr�td��|j��|j_|j�|j	j�|j�|j	�|j��|j�|jj�d|j_|jj�|j	�|j	|jj|j	j <dS)Nrz<LLQQz<LLLLz+File size unexpectedly exceeded ZIP64 limitz1Compressed size unexpectedly exceeded ZIP64 limitF)!rr(r�r0r�r2r-r�r�r-rlr1r3rkrmrfr.rBr��
_DD_SIGNATURErVr/�	start_dirr�r�r@rjr�r��filelistr.�
NameToInfor>)r{rr�)rrrr�Ms8






z_ZipWriteFile.close)
rrrr|�propertyr�r5r�r�r+rr)rrr,/s

r,c@seZdZdZdZdZdeddfdd�Zdd�Zd	d
�Z	dd�Z
d
d�Zdd�Zdd�Z
d<dd�Zdd�Zdd�Zdd�Zedd��Zejdd��Zd=dd�Zd>d d!�d"d#�Zd?d$d%�Zd@d&d'�ZdAd(d)�Zed*d+��Zd,d-�Zd.d/�ZdBd0d1�ZdCd2d3�Zd4d5�Z d6d7�Z!d8d9�Z"d:d;�Z#dS)Drai Class with methods to open, read, write, close, list zip files.

    z = ZipFile(file, mode="r", compression=ZIP_STORED, allowZip64=True,
                compresslevel=None)

    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.
    compresslevel: None (default for the given compression type) or an integer
                   specifying the level to pass to the compressor.
                   When using ZIP_STORED or ZIP_LZMA this keyword has no effect.
                   When using ZIP_DEFLATED integers 0 through 9 are accepted.
                   When using ZIP_BZIP2 integers 1 through 9 are accepted.

    N�rTc		CsL|dkrtd��t|�||_d|_d|_i|_g|_||_||_||_	d|_
d|_t|t
j�rjt
�|�}t|t�r�d|_||_ddd	d
dddd
�}||}x^yt�||�|_Wn(tk
r�||kr�||}w��YnXPq�Wnd|_||_t|dd�|_d|_t��|_d|_d|_�y|dk�r0|��n�|dk�r�d|_y|j��|_ Wn2t!tfk
�r�t"|j�|_d|_ d|_Yn6Xy|j�#|j �Wn t!tfk
�r�d|_YnXnf|dk�ry|��|j�#|j �Wn6t$k
�r|j�#dd�d|_|j��|_ YnXntd��Wn$|j}d|_|�%|��YnXdS)z]Open the ZIP file with mode read 'r', write 'w', exclusive create 'x',
        or append 'a'.)r<�w�x�az+ZipFile requires mode 'r', 'w', 'x', or 'a'FrNr*r;zw+bzx+bzr+b�wbZxb)r<r=r>r?zr+bzw+bzx+brr	Tr<)r=r>r?rz"Mode must be 'r', 'w', 'x', or 'a')&rwr��_allowZip64�
_didModify�debugr:r9r�r�rr��_commentr�rtr�r��str�_filePassedr>rr=r8r7�getattr�_fileRefCnt�	threading�RLockr�rr��_RealGetContentsrVr8rr�r@r�_fpclose)	r{r�rr��
allowZip64r�ZmodeDictr�r8rrrr|�s�







zZipFile.__init__cCs|S)Nr)r{rrr�	__enter__�szZipFile.__enter__cCs|��dS)N)r�)r{�type�value�	tracebackrrr�__exit__�szZipFile.__exit__cCs�d|jj|jjfg}|jdk	rd|jr8|�d|j�n|jdk	rR|�d|j�|�d|j�n
|�d�|�d�d�|�S)Nz<%s.%sz file=%rz filename=%rz mode=%rz	 [closed]r}r~)	rrrr8rFr.r>rr/)r{r?rrrr��s



zZipFile.__repr__cCs�|j}yt|�}Wntk
r.td��YnX|s<td��|jdkrNt|�|t}|t}|t|_	|t
||}|ttkr�|t
t8}|jdkr�||}td|||�|||_|�|jd�|�|�}t�|�}d}�x�||k�r�|�t�}	t|	�tk�rtd��t�t|	�}	|	ttk�r0td��|jdk�rDt|	�|�|	t�}
|	d}|d	@�rp|
�d
�}
n
|
�d�}
t|
�}|�|	t�|_|�|	t �|_!|	t"|_#|	dd�\|_$|_%|_&|_'|_(|_)}
}|_*|_+|_,|j&t-k�r�t.d
|j&d��|	dd�\|_/|_0|_1|
|_2|d?d|d?d@|d@|
d?|
d?d@|
d@df|_3|�4�|j#||_#|j5�6|�||j7|j8<|t|	t|	t|	t }|jdkr�td|�q�WdS)z/Read in the table of contents for the ZIP file.zFile is not a zip filerrzgiven, inferred, offsetrzTruncated central directoryz&Bad magic number for central directoryrizutf-8�cp437rzzip file version %.1fr#r&r)r"i��r$r�totalN)9r8r6r7rrC�printrMrN�_ECD_COMMENTrD�
_ECD_LOCATIONrHrGrErAr8r@r:r�BytesIO�sizeCentralDirr-rBr,�structCentralDir�
_CD_SIGNATURE�stringCentralDir�_CD_FILENAME_LENGTH�decoder
�_CD_EXTRA_FIELD_LENGTHr0�_CD_COMMENT_LENGTHr^�_CD_LOCAL_HEADER_OFFSETrjrSrcrdrerfrarkrlrm�MAX_EXTRACT_VERSIONr�rgrhrirnr`r�r9r.r:r>)r{r8rQZsize_cdZ	offset_cd�concatZinferredrRrU�centdirr>�flagsr>�t�drrrrKsp









4  
zZipFile._RealGetContentscCsdd�|jD�S)z+Return a list of file names in the archive.cSsg|]
}|j�qSr)r>)�.0rRrrr�
<listcomp>Qsz$ZipFile.namelist.<locals>.<listcomp>)r9)r{rrr�namelistOszZipFile.namelistcCs|jS)zJReturn a list of class ZipInfo instances for files in the
        archive.)r9)r{rrr�infolistSszZipFile.infolistcCsPtdd|d�x:|jD]0}d|jdd�}td|j||jf|d�qWdS)z+Print a table of contents for the zip file.z%-46s %19s %12s)z	File NamezModified    ZSize)r�z%d-%02d-%02d %02d:%02d:%02dNr z
%-46s %s %12d)rVr9r`r>rm)r{r�r�Zdaterrr�printdirXszZipFile.printdirc
Cs^d}xT|jD]J}y.|�|jd��}x|�|�r0q$WWdQRXWqtk
rT|jSXqWdS)z%Read all the files and check the CRC.ir<N)r9r=r>r:r)r{Z
chunk_sizer��frrr�testzipaszZipFile.testzipcCs$|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	�inforrr�getinfons

zZipFile.getinfocCs6|r t|t�s tdt|�j��|r,||_nd|_dS)z)Set default password for encrypted files.zpwd: expected bytes, got %sN)r�r��	TypeErrorrOrr�)r{r�rrr�setpasswordws
zZipFile.setpasswordcCs|jS)z.The comment text associated with the ZIP file.)rD)r{rrrr^�szZipFile.commentcCs^t|t�stdt|�j��t|�tkrNddl}|jdtdd�|dt�}||_	d|_
dS)Nzcomment: expected bytes, got %srz3Archive comment is too long; truncating to %d bytesr)�
stacklevelT)r�r�rsrOrr-�ZIP_MAX_COMMENT�warnings�warnrDrB)r{r^rwrrrr^�s
c	Cs"|�|d|��
}|��SQRXdS)zReturn file bytes for name.r<N)r=r:)r{r	r�r8rrrr:�szZipFile.readF)�force_zip64csv|dkrtd��|r0t|t�s0tdt|�j��|rD|dkrDtd���jsRtd��t|t�rb|}n,|dkr�t|�}�j|_	�j
|_n
��|�}|dkr��j
||d�S�jr�td���jd	7_t�j|j�j�j�fd
d��}�yx|�t�}t|�tk�rtd��t�t|�}|ttk�r&td
��|�|t�}|t�rL|�|t�|jd@�r`t d��|jd@�rtt d��|jd@�r�|�!d�}	n
|�!d�}	|	|j"k�r�td|j"|f��|jd	@}
d}|
�rL|�sԈj#}|�s�t$d|��t%|�}|�d�}||dd��}
|jd@�r$|j&d?d@}n|j'd?d@}|
d|k�rLt$d|��t(||||d�S|�)��YnXdS) auReturn 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.
        >r=r<zopen() requires mode "r" or "w"zpwd: expected bytes, got %sr=z'pwd is only supported for reading filesz2Attempt to use ZIP archive that was already closed)ryzyCan't read from the ZIP file while there is an open writing handle on it. Close the writing handle before trying to read.rcs�jS)N)r�r)r{rr�<lambda>�r*zZipFile.open.<locals>.<lambda>zTruncated file headerz Bad magic number for file header� z$compressed patched data (flag bit 5)�@zstrong encryption (flag bit 6)izutf-8rSz/File name in directory %r and header %r differ.Nz6File %r is encrypted, password required for extractionrrrr�r�r$zBad password for file %rT)*rwr�r�rsrOrr8r
r�rar�rbrr�_open_to_writer�rHr�rjrLr�r:�sizeFileHeaderr-rrBr,r��
_FH_SIGNATUREr��_FH_FILENAME_LENGTH�_FH_EXTRA_FIELD_LENGTHrfr�r_r_r�r�r�rnrkr�r�)r{r	rr�ryr�Zzef_fileZfheader�fnameZ	fname_strZis_encryptedZzdr��hZ
check_byter)r{rr=�s�








zZipFile.opencCs�|r|jstd��|jr td��t|d�s0d|_d|_d|_d|_|jt	krZ|jdO_|j
sn|jdO_|jszd|_|jo�|p�|jdtk}|j
r�|j
�|j�|j
��|_|�|�d	|_|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.rmrrri�g�������?T)rArwr�r<rmrlrkrfrarrrir�r8r@r8rVrj�_writecheckrBr�r�r,)r{r�ryr�rrrr}	s4



zZipFile._open_to_writecCs*|dkrt��}n
t�|�}|�|||�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)rt�getcwdr��_extract_member)r{�memberr�r�rrr�extract4s

zZipFile.extractcCsL|dkr|��}|dkr"t��}n
t�|�}x|D]}|�|||�q2WdS)z�Extract 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)rkrtr�r�r�)r{r��membersr�rrrr�
extractallAs


zZipFile.extractallcCs^|j}|s(d}t�|dt|��}||_|�|�}dd�|�|�D�}|�dd�|D��}|S)z;Replace bad characters and remove trailing dots from parts.z:<>|"?*�_css|]}|�d�VqdS)�.N)�rstrip)rir>rrr�	<genexpr>\sz1ZipFile._sanitize_windows_name.<locals>.<genexpr>css|]}|r|VqdS)Nr)rir>rrrr�^s)�!_windows_illegal_name_trans_tablerE�	maketransr-�	translate�splitr/)r�r��pathsep�tableZillegalrrr�_sanitize_windows_nameRs
zZipFile._sanitize_windows_namec
sLt|t�s|�|�}|j�dtjj�}tjjrB|�tjjtjj�}tj�	|�d}dtjj
tjjf�tjj��fdd�|�
tjj�D��}tjjdkr�|�|tjj�}tj�||�}tj�|�}tj�|�}|r�tj�|�s�t�|�|���rtj�|��st�|�|S|j||d��(}t|d��}t�||�Wd	QRXWd	QRX|S)
zbExtract the ZipInfo object 'member' to a physical
           file on the path targetpath.
        rqrr~c3s|]}|�kr|VqdS)Nr)rir>)�invalid_path_partsrrr�rsz*ZipFile._extract_member.<locals>.<genexpr>�\)r�r@N)r�r
rrr>rvrtr�rur�r��curdir�pardirr/r�r�r��dirname�exists�makedirsr�r��mkdirr=�shutil�copyfileobj)r{r�Z
targetpathr�r�Z	upperdirs�source�targetr)r�rr�as.

&


 zZipFile._extract_membercCs�|j|jkr(ddl}|jd|jdd�|jdkr:td��|jsHtd��t|j�|j	s�d}t
|j�tkrpd	}n|j
tkr�d
}n|jtkr�d}|r�t|d��dS)
z6Check for errors before writing a file to the archive.rNzDuplicate name: %rr)ru)r=r>r?z&write() requires mode 'w', 'x', or 'a'z4Attempt to write ZIP archive that was already closedzFiles countZFilesizezZipfile sizez would require ZIP64 extensions)r>r:rwrxrrwr8r�rarAr-r9�ZIP_FILECOUNT_LIMITrmr�rjr
)r{r�rw�requires_zip64rrrr��s(



zZipFile._writecheckc
CsJ|jstd��|jrtd��t�||�}|��r>d|_d|_n0|dk	rN||_n|j	|_|dk	rf||_
n|j|_
|���r
|j��|j
r�|j�|j�|j��|_|jtkr�|jdO_|�|�d|_|j�|�||j|j<|j�|�d��|j��|_WdQRXn<t|d��,}|�|d	��}t�||d
�WdQRXWdQRXdS)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 existsrNrTFr;r=i )r8rwr�r
r�r�rlrkrar�rbr�r�rr@r8rVrjrrfr�rBr9r.r:r>r�r�r=r�r�)r{r>r�rar�r��src�destrrrr��s>


z
ZipFile.writec
Cs�t|t�r|�d�}t|t�sxt|t�t���dd�d�}|j|_|j|_	|j
ddkrpd|_|jdO_q|d	|_n|}|js�t
d
��|jr�t
d��|dk	r�||_|dk	r�||_	t|�|_|j�*|j|dd
��}|�|�WdQRXWdQRXdS)aWrite 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>r`r�rqi�Ar'i�z7Attempt to write to ZIP archive that was already closedz?Can't write to ZIP archive while an open writing handle exists.r=)r)r�rEr�r
r�r�r�rar�rbr>rir8rwr�r-rmr�r=r�)r{Zzinfo_or_arcnamerRrar�r�r�rrr�writestr�s4



zZipFile.writestrcCs|��dS)z2Call the "close()" method in case the user forgot.N)r�)r{rrr�__del__�szZipFile.__del__c	Cs||jdkrdS|jrtd��zB|jdkr\|jr\|j�"|jrJ|j�|j�|�	�WdQRXWd|j}d|_|�
|�XdS)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.)r=r>r?)r8r�rwrrBr�rr@r8�_write_end_recordrL)r{r8rrrr��s
z
ZipFile.closecCsr�xP|jD�]D}|j}|ddd>|dd>B|dB}|dd>|d	d>B|ddB}g}|jtksv|jtkr�|�|j�|�|j�d
}d
}n|j}|j}|jtkr�|�|j�d
}n|j}|j}	d}
|�rt|	d�}	t	j
dd
t|�ddt|�f|��|	}	t}
|j
tk�r(tt|
�}
n|j
tk�r>tt|
�}
t|
|j�}t|
|j�}yZ|��\}
}t	�
tt||j||j||j
|||j||t|
�t|	�t|j�d|j|j|�}Wnltk
�rttt||j||j|j |j
|||j||t|j!�t|	�t|j�d|j|j|ft"j#d��YnX|j$�%|�|j$�%|
�|j$�%|	�|j$�%|j�q
W|j$�&�}t|j�}||j'}|j'}d}|t(k�r�d}n|tk�r�d}n|tk�r�d}|�r*|j)�s�t*|d��t	�
t+t,ddddd||||�}|j$�%|�t	�
t-t.d|d�}|j$�%|�t/|d�}t/|d
�}t/|d
�}t	�
t0t1dd||||t|j2��	}|j$�%|�|j$�%|j2�|j$�3�dS)Nri�r"rrrrr$rl��)rz<HH�Qr)r�zFiles countzCentral directory offsetzCentral directory sizez would require ZIP64 extensions�,ri��)4r9r`rmr�rlr.rjr0r5rBr�r-r�rarr[r�rr�rdrSr�r[r]rcrerkr^rhri�DeprecationWarningrVrfr>rx�stderrr8r�rVr8r�rAr
rFrGrCrDr&rYrXrDr�)r{r�r�r�r�r0rmrlrjZ
extra_datar�rdrSr>rfreZpos2ZcentDirCountZcentDirSizeZ
centDirOffsetr�Zzip64endrecZzip64locrecrQrrrr�s�$$



















zZipFile._write_end_recordcCs4|jdkst�|jd8_|js0|js0|��dS)Nrr)rH�AssertionErrorrFr�)r{r8rrrrL{szZipFile._fpclose)N)N)r<N)F)NN)NNN)NNN)NN)$rrrrr8r�rr|rNrRr�rKrkrlrmrorrrtr;r^�setterr:r=r}r�r�r�r�r�r�r�r�r�r�r�rLrrrrrzs@XN
	
		
r
+

*
0
(fc@s4eZdZdZdeddfdd�Zd
d	d
�Zdd�ZdS)rzDClass to create ZIP archives with Python library files and packages.r<Tr�cCstj|||||d�||_dS)N)rr�rM)rr|�	_optimize)r{r�rr�rM�optimizerrrr|�szPyZipFile.__init__r~NcCs�t�|�}|rD||�sD|jr@tj�|�r,dnd}td||f�dStj�|�\}}tj�|��rptj�|d�}tj�|��r�|r�d||f}n|}|jr�td|d|�|�	|d	d
�|�\}}	|jr�td|	�|�
||	�tt�|��}
|
�
d�x�|
D]�}tj�||�}tj�|�\}
}tj�|��rTtj�tj�|d���r�|j|||d�q�|d
kr�|�r�||��s�|jr�td|�q�|�	|d	d
�|�\}}	|j�r�td|	�|�
||	�q�Wn�|j�r�td|�x�tt�|��D]�}tj�||�}tj�|�\}
}|d
k�r�|�r2||��s2|j�r�td|��q�|�	|d	d
�|�\}}	|j�r\td|	�|�
||	��q�WnP|d
d�d
k�r�td��|�	|d	d
�|�\}}	|j�r�td|	�|�
||	�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 in�asr�����ZAdding)�
filterfuncz.pyzfile %r skipped by filterfunczAdding files from directoryz.Files added with writepy() must end with ".py"zAdding file)rtr�rCr�r�rVr�r/�isfile�
_get_codenamer��sorted�listdir�remove�splitext�writepyr�)r{�pathname�basenamer�Zlabel�dirr	Zinitnamer�r�Zdirlistr>r��rootZextrrrr��sx











zPyZipFile.writepyc
sd�fdd�	}|d}|d}tjj|dd�}tjj|dd�}tjj|d	d�}�jdk�r\tj�|�r�t�|�jt�|�jkr�|}	}
n�tj�|�r�t�|�jt�|�jkr�|}
|}	n�tj�|�r�t�|�jt�|�jkr�|}
|}	nvtj�|��rt�|�jt�|�jk�r|}
|}	nD||��rRt	j
jd
k�r4|}
nt	j
jdk�rH|}
n|}
|}	n|}
}	n��jd
k�rr|}
|}	n<|}	�jdk�r�|}
n&�jd	k�r�|}
nd��j�}t
|��tj�|
��r�t�|
�jt�|�jk�s�||�jd��s�|}
}	tj�|	�d}|�rd
||f}|
|fS)aReturn (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`ddl}�jrtd|�y|j|d|d�Wn.|jk
rZ}zt|j�dSd}~XYnXdS)NrZ	CompilingT)�doraiser�F)�
py_compilerCrV�compile�PyCompileError�msg)r�r�r��err)r{rr�_compile�s

z)PyZipFile._get_codename.<locals>._compilez.pyz.pycr~)�optimizationrrrz"invalid value for 'optimize': {!r})r�z%s/%s)r�)�	importlib�util�cache_from_sourcer�rtr�r�r�r�rxrfr��formatrwr�)
r{r�r�r�Zfile_pyZfile_pycZpycache_opt0Zpycache_opt1Zpycache_opt2r�r�r�Zarchivenamer)r{rr��s`


zPyZipFile._get_codename)r~N)rrrrrr|r�r�rrrrr�s

Rc
	s�ddl}d}|j|d�}|jdd�}|jdddd	d
�|jddd
ddd�|jdddddd�|jddddd
�|�|�}|jdk	r�|j}t|d��}|��}WdQRX|r�td�	|��td��n|j
dk	r�|j
}t|d��}|��WdQRXn�|jdk	�r.|j\}}t|d��}|�
|�WdQRXn�|jdk	�r�|j�d�}	|j}
�fdd��t|	d��`}xX|
D]P}tj�|�}|�s�tj�tj�|��}|dtjtjfk�r�d}�|||��qjWWdQRXdS) Nrz3A simple command-line interface for zipfile module.)�descriptionT)Zrequiredz-lz--listz	<zipfile>zShow listing of a zipfile)�metavar�helpz-ez	--extractr)z	<zipfile>z<output_dir>zExtract zipfile into target dir)�nargsr�r�z-cz--create�+)z<name>z<file>zCreate zipfile from sourcesz-tz--testzTest if a zipfile is validr<z.The following enclosed file is corrupted: {!r}zDone testingcsttj�|�r|�||t�nTtj�|�rp|r8|�||�x6tt�|��D]$}�|tj�||�tj�||��qHWdS)N)	rtr�r�r�rr�r�r�r/)r4r��zippathZnm)�addToZiprrr�Xszmain.<locals>.addToZipr=r~)�argparse�ArgumentParserZadd_mutually_exclusive_group�add_argument�
parse_argsZtestrrorVr�rZrmr�r�Zcreate�poprtr�r�r�r�r�)
�argsr�r��parser�groupr�r4Zbadfiler�Zzip_nameZfilesr�r�r)r�r�main0sT








r��__main__)N)N)�rrrt�importlib.utilr�rxr�r�r�rBZbinasciirIr�r��ImportErrorr�r��__all__�	Exceptionrr
rrr�r�rvrrrrrzr�r�r�rcrYrXr�rWrHrIrJrKrLrMrNr]rWrXr[r]rZr\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_SIZEr^r`raZ_CD_DISK_NUMBER_STARTZ_CD_INTERNAL_FILE_ATTRIBUTESZ_CD_EXTERNAL_FILE_ATTRIBUTESrbr�r�r~rZ_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�rCrDrArFrGrEZ_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_CENTDIRr7ZStructr+r5r9r	rTr6�objectr
r�r�r�r�r�r�r�r�r�r�r�rr�r,rrr�rrrrr�<module>s@









+=a&
&,K/
<