U pdV@sdZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZzddlZejZWnek rdZejZYnXz ddlZWnek rdZYnXz ddlZWnek rdZYnXddddddd d d d d ddg ZGdddeZGdddeZeZZdZdZdZdZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'dZ(e )e'Z*dZ+dZ,dZ-dZ.d Z/d!Z0d"Z1d#Z2dZ3d$Z4d%Z5d&Z6e )e5Z7dZ8dZ9dZ:dZ;d Zd#Z?dZ@d$ZAd'ZBd(ZCdZDd)ZEdZFd*ZGd+ZHd,ZId-ZJd.ZKd/ZLe )eKZMdZNdZOdZPdZQd ZRd!ZSd"ZTd#ZUdZVd$ZWd'ZXd(ZYd0ZZd1Z[e )eZZ\d2Z]d3Z^e )e]Z_dZ`dZadZbdZcd Zdd!Zed"Zfd#ZgdZhd$Zid4Zje kd5Zld6d7Zmd8d9Znd:d Zod;d<Zpd=d>ZqGd?d d erZsdatd@dAZudBdCZvGdDdEdEZwGdFdGdGZxdHdIdJdJdJdJdKdLdMdNdKdOdPdQdRdSdTdUZydVdWZzdtdXdYZ{dZd[Z|Gd\d]d]Z}Gd^d_d_Z~Gd`dadaejZGdbdcdcejZGddd d ZGded d eZdfdgZdhdiZejZdjdkZGdldmdmeZGdndodoeZGdpddZdudqdrZedskr edS)vzP 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 LargeZipFilePathc@s eZdZdS)rN)__name__ __module__ __qualname__rr/usr/lib/python3.8/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 -.?s<4s4H2LHsPK z<4s4B4HL2L5H2LsPK z <4s2B4HL2L2HsPKz<4sLQLsPKz <4sQ2H2L4QsPKiPKr0structr/structEndArchive64LocatorstringEndArchive64LocatorrsizeEndCentDir64structEndArchive64stringEndArchive64_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 diroffsetrrr _EndRecData64s@      rXc Csh|dd|}z|t dWntk r<YdSX|}t|tkr|ddtkr|dddkrtt |}t |}| d| |tt |t |St |dtd}||d|}|t}|dkrd|||t}t|tkrdSt tt |}|t}||t|t|}| || ||t |||||SdS) 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.rrNr sr-i)rDtellsizeEndCentDirr:r>r0stringEndArchiverFr/structEndArchivelistr1rXmaxrfind_ECD_COMMENT_SIZE) rSZfilesizerVrUZmaxCommentStartr5ZrecDataZ commentSizecommentrrrr9sD        r9c@s\eZdZdZdZdddZddZdd d Zd d ZddZ e dddddZ ddZ d S)r z>Class with attributes describing each file in the ZIP archive.) orig_filenamerB date_time compress_type_compresslevelrbr3 create_systemrWextract_versionreserved flag_bitsvolume internal_attr external_attr header_offsetCRC compress_size file_size _raw_timeNoNamerrrrrcCs||_|td}|dkr(|d|}tjdkrJtj|krJ|tjd}||_||_|ddkrjtdt |_ d|_ d|_ d|_ tjdkrd|_nd|_t|_t|_d|_d|_d|_d|_d|_dS)Nr/ruz+ZIP does not support timestamps before 1980r-Zwin32r)rcfindchrossepreplacerBrd ValueErrorrrerfrbr3sysplatformrgDEFAULT_VERSIONrWrhrirjrkrlrm)selfrBrdZ null_byterrr__init__Xs0   zZipInfo.__init__cCsd|jj|jfg}|jtkr8|dt|j|j|jd?}|jd@}|rd|dt ||rv|d|| }|r|j r|d|j |r|j r|jtks|j |j kr|d|j |d d |S) Nz<%s filename=%r compress_type=%sr)rz filemode=%rz external_attr=%#xz file_size=%rz compress_size=%r>) __class__rrBrerr1compressor_namesgetrmstatfilemodeis_dirrqrpr2)rrChiloisdirrrr__repr__s0       zZipInfo.__repr__NcCs||j}|ddd>|dd>B|dB}|dd>|d d>B|ddB}|jd @rfd}}}n|j}|j}|j}|j}d} |d kr|tkp|tk}|rd } |t| dt | d ||}|tks|tkr|st d d}d}t } |j t krtt| } n|j tkrtt| } t| |j|_t| |j|_|\} } ttt|j|j| |j |||||t| t| } | | |S)z-Return the per-file header as a bytes object.rrur$rr!rrr&r rNz|jd|jdBfYSXdS)Nasciiutf-8)rBencoderjUnicodeEncodeErrorrrrrrszZipInfo._encodeFilenameFlagscCs|j}tj}t|dkr|d|dd\}}|dt|krPtd||f|dkrp|dkrv|d|dd}nV|dkr|d |dd }n:|d kr|d |dd }n|dkrd}ntd||fd}|jdkrt||krtd|||_|d7}|jdkr6t||kr$td|||_|d7}|jdkrpt||krXtd|j}|||_|d7}||dd}q dS)Nr r,z"Corrupt extra field %04x (size=%d)rz|_|rd|_|jd O_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"rurt;)rr;rNrrvrr)) isinstanceryPathLikefspathrS_ISDIRst_modetime localtimest_mtimepathnormpath splitdriverzaltseprmrqst_size) clsrBarcnamerstrmtimerdzinforrr from_files0       zZipInfo.from_filecCs|jddkS)z2Return True if this archive member is a directory.rvrBrrrrr%szZipInfo.is_dir)rsrt)N)N) rrrr __slots__rrrrr classmethodrrrrrrr >s + .2%cCs0tdD]"}|d@r"|d?dA}q|dL}q|S)Nrrl q[)range)crcr7rrr_gen_crc/s   rcsldddtdkr&ttttdatfddfdd|D] }|qLfd d }|S) NixV4igE#ixV4cs|d?||Ad@AS)z(Compute the CRC32 primitive on one byte.rr)Zchr)crctablerrcrc32Isz_ZipDecrypter..crc32cs<|d@d@ddd@d?dS)Nrrirrr)c)rkey0key1key2rr update_keysMs z"_ZipDecrypter..update_keyscsNt}|j}|D]4}dB}|||dAd?d@N}|||qt|S)zDecrypt a bytes object.rrrr) bytearrayr1bytes)rVrCr1rk)rrrr decrypterWs z _ZipDecrypter..decrypter) _crctabler^maprr)pwdprr)rrrrrrr _ZipDecrypter?s   rc@s,eZdZddZddZddZddZd S) LZMACompressorcCs d|_dSN)_comprrrrrgszLZMACompressor.__init__cCsFtdtji}tjtjttj|gd|_tdddt ||S)Nidfiltersz|rtd|j|||j|_|jW5QRSQRXdS)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.)rrr|rrDrZr)rrTwhencerrrrDs  z_SharedFile.seekrc CsX|jH|rtd|j|j|j|}|j|_|W5QRSQRXdS)NyCan't read from the ZIP file while there is an open writing handle on it. Close the writing handle before trying to read.)rrr|rrDrr>rZrnrVrrrr>s  z_SharedFile.readcCs$|jdk r |j}d|_||dSr)rr)rfileobjrrrr s z_SharedFile.closeN)r)r)rrrrrDr>r rrrrrs rc@s4eZdZddZddZddZddZd d Zd S) _TellablecCs||_d|_dSNr)r<rTrr<rrrrsz_Tellable.__init__cCs|j|}|j|7_|Sr)r<writerT)rrVrrrrr s z_Tellable.writecCs|jSr)rTrrrrrZsz_Tellable.tellcCs|jdSr)r<rrrrrrsz_Tellable.flushcCs|jdSr)r<r rrrrr sz_Tellable.closeN)rrrrrrZrr rrrrrs rcseZdZdZdZdZdZd(ddZd d Zd d Z d)ddZ d*ddZ ddZ d+ddZ ddZddZddZddZfdd Zd!d"Zd,d$d%Zd&d'ZZS)- ZipExtFilezZFile-like object for reading an archive member. Is returned by ZipFile.open(). i@iiNFcCs(||_||_||_|j|_|j|_|j|_t |j|_ d|_ d|_ d|_ d|_||_|j|_t|drz|j|_td|_nd|_d|_z4|r||_|j|_|j|_|j|_d|_Wntk rYnXd|_|r$|j d@r|j!d?d@}n|jd?d@}|"}||kr$t#d |j$dS) NFr-rroTrrrzBad password for file %r)%_fileobj_pwd_close_fileobjre_compress_typerp_compress_leftrq_leftr _decompressor_eof _readbuffer_offsetnewlinesmoderBnamer@ro _expected_crcr _running_crc _seekablerrZ_orig_compress_start_orig_compress_size_orig_file_size_orig_start_crcAttributeError _decrypterrjrr_init_decrypterrrc)rrr zipinforZ close_fileobjZ check_bytehrrrr)sF       zZipExtFile.__init__cCs4t|j|_|jd}|jd8_||dS)Nrr&)rrr*rr>r)rrrrrr+\s  zZipExtFile._init_decryptercCsvd|jj|jjfg}|jsX|d|j|jf|jtkrb|dt |j|jn |d|dd |S)N<%s.%sz name=%r mode=%rr [closed]rr) rrrclosedr1r!r rrrrr2rrCrrrrgs   zZipExtFile.__repr__rcCsL|dkr>|jd|jd}|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)rrwrioBufferedIOBasereadline)rlimitr6linerrrr5uszZipExtFile.readlinercCsr|t|j|jkr\||}t||jkrJ||j|jd|_d|_n|jt|8_|j|j|jdS)z6Returns buffered bytes without advancing the position.Nri)r0rrr>)rrchunkrrrpeeks zZipExtFile.peekcCsdSNTrrrrrreadableszZipExtFile.readablecCs|dks|dkrH|j|jd}d|_d|_|jsD|||j7}q,|S||j}|t|jkrz|j|j|}||_|S|t|j}|j|jd}d|_d|_|dkr|js||}|t|kr||_||_||d|7}q||7}|t|8}q|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. Nrr-rrr_read1MAX_Nr0)rrbufendrVrrrr>s4   zZipExtFile.readcCs@|jdkrdSt||j|_|jr<|j|jkr     zZipExtFile.read1cCs"|js|dkrdS|jtkrH|jj}|t|krR|||t|7}n ||}|jtkrj|jdk|_nx|jtkrt ||j }|j ||}|jj p|jdko|jj |_|jr||j 7}n |j |}|jj p|jdk|_|d|j}|jt|8_|jdkrd|_|||S)Nrr-T)rrrrZunconsumed_tailr0_read2rrr_ MIN_READ_SIZErrrrrBr rrrr=s4          zZipExtFile._read1cCsd|jdkrdSt||j}t||j}|j|}|jt|8_|sLt|jdk r`||}|S)Nrr-) rr_rEminrr>r0EOFErrorr*r rrrrD s      zZipExtFile._read2cs&z|jr|jW5tXdSr)superr rrrrrrr szZipExtFile.closecCs|jSr)r$rrrrr szZipExtFile.seekablercCs>|jstd|}|dkr&|}n.|dkr8||}n|dkrL|j|}ntd||jkrd|j}|dkrpd}||}||j}|dkr|t|jkr||_d}nf|dkr |j |j |j |_ |j|_|j|_d|_d|_t|j|_d|_|}|jdk r ||dkr6t|j|}||||8}q |S)N!underlying stream is not seekablerrrzCwhence must be os.SEEK_SET (0), os.SEEK_CUR (1), or os.SEEK_END (2)r-F)r$r3UnsupportedOperationrZr'r|rr0rrrDr%r(r#r&rrrrrrr*r+rF MAX_SEEK_READr>)rrTr Zcurr_posZnew_posZ read_offsetZ buff_offsetZread_lenrrrrD#sH            zZipExtFile.seekcCs0|jstd|j|jt|j|j}|S)NrJ)r$r3rKr'rr0rr)rZfileposrrrrZSs zZipExtFile.tell)NF)r)r)r)r)rrrrr>rErLrr+rr5r9r;r>rBrCr=rDr rrDrZ __classcell__rrrIrrs* 3    ! %$  0rcs@eZdZddZeddZddZddZfd d ZZ S) _ZipWriteFilecCs8||_||_||_t|j|j|_d|_d|_d|_ dSr) _zinfo_zip64_zipfilerrerf _compressor _file_size_compress_size_crc)rzfrrrrrr[sz_ZipWriteFile.__init__cCs|jjSr)rQr<rrrrresz_ZipWriteFile._fileobjcCsdSr:rrrrrwritableisz_ZipWriteFile.writablecCsf|jrtdt|}|j|7_t||j|_|jrV|j|}|jt|7_|j ||S)NzI/O operation on closed file.) r0r|r0rSrrUrRrrTrr)rrVnbytesrrrrls  z_ZipWriteFile.writec sb|jr dSzFt|jrR|j}|jt|7_|j ||j|j _ n |j |j _ |j|j _|j |j _|j jd@r|jrdnd}|j t|t|j j|j j |j j|j |j_nn|js|j tkrtd|jtkrtd|j |j_|j |j j|j |j |j|j |jj|jj|j |j |jj|j j <W5d|j_XdS)NFrzdd Zd?d!d"d#d$Zd@d%d&ZdAd'd(ZdBd)d*Zed+d,Zd-d.Zd/d0ZdCd1d2ZdDd3d4Zd5d6Z d7d8Z!d9d:Z"d;d<Z#dS)Er ai 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. NrTrc CsP|dkrtdt|||_d|_d|_i|_g|_||_||_||_ d|_ d|_ ||_ t |tjrpt|}t |trd|_||_ddd d dd d d }||}zt|||_Wqtk r||kr||}YqYqXqqnd|_||_t|dd|_d|_t|_d|_d|_z|dkr4|n|dkrd|_z|j |_!Wn2t"tfk rt#|j|_d|_!d|_Yn6Xz|j$|j!Wn t"tfk rd|_YnXnf|dkrz||j$|j!Wn6t%k r|j$ddd|_|j |_!YnXntdWn$|j} d|_|&| YnXdS)z]Open the ZIP file with mode read 'r', write 'w', exclusive create 'x', or append 'a'.)r^wxaz+ZipFile requires mode 'r', 'w', 'x', or 'a'FrNr-r?w+bx+br+bwbZxb)r^r_r`rardrbrcrr!Tr^)r_r`rarz"Mode must be 'r', 'w', 'x', or 'a')'r|r _allowZip64 _didModifydebugr\r[rrr r_comment_strict_timestampsrryrrstr _filePassedrBr3rAr<r:getattr _fileRefCnt threadingRLockrr$r_RealGetContentsrZrZr)rrDr_fpclose) rrr r allowZip64rrZmodeDictrr<rrrrs            zZipFile.__init__cCs|Srrrrrr __enter__szZipFile.__enter__cCs |dSrr )rtypevalue tracebackrrr__exit__szZipFile.__exit__cCsd|jj|jjfg}|jdk rd|jr8|d|jn|jdk rR|d|j|d|jn |d|dd|S)Nr.z file=%rz filename=%rz mode=%rr/rr) rrrr<rlr1rBr r2r1rrrr"s    zZipFile.__repr__c Cs|j}z t|}Wntk r.tdYnX|sr3BytesIOsizeCentralDirr0rFr/structCentralDir _CD_SIGNATUREstringCentralDir_CD_FILENAME_LENGTHdecoder _CD_EXTRA_FIELD_LENGTHr3_CD_COMMENT_LENGTHrb_CD_LOCAL_HEADER_OFFSETrnrWrgrhrirjrerorprqMAX_EXTRACT_VERSIONrrkrlrmrrrdrr[r1r\rB)rr<rUZsize_cdZ offset_cdconcatZinferredrVr{centdirrBflagsr`tdrrrrq0s                        zZipFile._RealGetContentscCsdd|jDS)z+Return a list of file names in the archive.cSsg|] }|jqSrr).0rVrrr sz$ZipFile.namelist..r[rrrrnamelist~szZipFile.namelistcCs|jS)zJReturn a list of class ZipInfo instances for files in the archive.rrrrrinfolistszZipFile.infolistcCsLtdd|d|jD]0}d|jdd}td|j||jf|dqdS)z+Print a table of contents for the zip file.z%-46s %19s %12s)z File Namez Modified ZSizerz%d-%02d-%02d %02d:%02d:%02dNr"z %-46s %s %12d)r|r[rdrBrq)rrrdaterrrprintdirs zZipFile.printdirc Cs^d}|jD]N}z*||jd}||r,q W5QRXWq tk rV|jYSXq dS)z%Read all the files and check the CRC.ir^N)r[rArBr>r)rZ chunk_sizerfrrrtestzips  zZipFile.testzipcCs$|j|}|dkr td||S)z,Return the instance of ZipInfo given 'name'.Nz(There is no item named %r in the archive)r\rKeyError)rr!inforrrgetinfos  zZipFile.getinfocCs6|r t|ts tdt|j|r,||_nd|_dS)z)Set default password for encrypted files.pwd: expected bytes, got %sN)rr TypeErrorrvrr)rrrrr setpasswords zZipFile.setpasswordcCs|jS)z.The comment text associated with the ZIP file.)rirrrrrbszZipFile.commentcCs^t|tstdt|jt|tkrNddl}|jdtdd|dt}||_ d|_ dS)Nzcomment: expected bytes, got %srz3Archive comment is too long; truncating to %d bytesr stacklevelT) rrrrvrr0ZIP_MAX_COMMENTwarningswarnrirg)rrbrrrrrbs   c Cs.||d|}|W5QRSQRXdS)zReturn file bytes for name.r^NrAr>)rr!rr<rrrr>sz ZipFile.readF force_zip64c s|dkrtd|r0t|ts0tdt|j|rD|dkrDtdjsRtdt|trb|}n,|dkrt|}j|_ j |_ n |}|dkrj ||dSjrtdjd 7_tj|jjjfd d }z|t}t|tkrtd tt|}|ttkr&td ||t}|trL||t|jd@r`t d|jd@rtt d|t!d@r|"d} n |"d} | |j#krtd|j#|f|jd @} | r|s҈j$}|st%d|nd}t&||||dWS|'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"rr_z'pwd is only supported for reading filesz2Attempt to use ZIP archive that was already closedrr rcsjSr)rrrrrr-zZipFile.open..zTruncated file headerz Bad magic number for file header z$compressed patched data (flag bit 5)@zstrong encryption (flag bit 6)rrrzz/File name in directory %r and header %r differ.z6File %r is encrypted, password required for extractionNT)(r|rrrrvrr<r rrerrfr_open_to_writerrnrrnrrrr>sizeFileHeaderr0rrFr/r _FH_SIGNATUREr_FH_FILENAME_LENGTH_FH_EXTRA_FIELD_LENGTHrjr_FH_GENERAL_PURPOSE_FLAG_BITSrrcrrrr ) rr!r rrrZzef_fileZfheaderfnameZ fname_strZ is_encryptedrrrrAs~               z ZipFile.opencCs|r|jstd|jr tdt|ds0d|_d|_d|_d|_|jt krZ|jdO_|j sn|jdO_|j szd|_ |jo|p|jdt k}|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.rqrrrg?T)rfr|rr@rqrprorjrerr$rmrr<rDrZrZrn _writecheckrgrrrN)rrrrrrrr(s8     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)rygetcwdr_extract_member)rmemberrrrrrextractSs  zZipFile.extractcCsH|dkr|}|dkr"t}n t|}|D]}||||q0dS)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)rryrrr)rrmembersrr,rrr extractall`s  zZipFile.extractallcCs^|j}|s(d}t|dt|}||_||}dd||D}|dd|D}|S)z;Replace bad characters and remove trailing dots from parts.z:<>|"?*_css|]}|dVqdS).N)rstriprr`rrr {sz1ZipFile._sanitize_windows_name..css|]}|r|VqdSrrrrrrr}s)!_windows_illegal_name_trans_tablerk maketransr0 translatesplitr2)rrpathseptableZillegalrrr_sanitize_windows_nameqs zZipFile._sanitize_windows_namec sLt|ts||}|jdtjj}tjjrB|tjjtjj}tj |d}dtjj tjj ftjj fdd| tjjD}tjjdkr||tjj}tj ||}tj|}tj|}|rtj|st||r tj|st||S|j||d(}t|d}t||W5QRXW5QRX|S) zbExtract the ZipInfo object 'member' to a physical file on the path targetpath. rvrrc3s|]}|kr|VqdSrrrZinvalid_path_partsrrrsz*ZipFile._extract_member..\)rre)rr rrBr{ryrrzrrcurdirpardirr2rrrdirnameexistsmakedirsrrmkdirrAshutil copyfileobj)rrZ targetpathrrZ upperdirssourcetargetrrrrs2  &       zZipFile._extract_membercCs|j|jkr(ddl}|jd|jdd|jdkr:td|jsHtdt|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.rNzDuplicate name: %rrrr_r`raz&write() requires mode 'w', 'x', or 'a'z4Attempt to write ZIP archive that was already closed Files countZFilesizez Zipfile size would require ZIP64 extensions)rBr\rrr r|r<rrerfr0r[ZIP_FILECOUNT_LIMITrqrrnr )rrrrequires_zip64rrrrs,     zZipFile._writecheckc CsP|jstd|jrtdtj|||jd}|rDd|_d|_n0|dk rT||_ n|j |_ |dk rl||_ n|j |_ |r|j |jr|j|j|j|_|j tkr|jdO_||d|_|j|||j|j<|j|d|j|_W5QRXnCan't write to ZIP archive while an open writing handle existsrrNrTFr?r_i ) r<r|rr rrjrrprorerrfrrr$rDrZrZrnrrjrrgr[r1r\rBrrrArr)rrBrrerrsrcdestrrrrsF      z ZipFile.writec Cst|tr|d}t|tsxt|ttddd}|j|_|j|_ |j ddkrpd|_ |j dO_ q|d |_ n|}|j st d |jrt d |dk r||_|dk r||_ t||_|j*|j|d d }||W5QRXW5QRXdS)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.rNr")rBrdrrviAr)rrz?Can't write to ZIP archive while an open writing handle exists.r_)r )rrkrr rrrrerrfrBrmr<r|rr0rqrrAr)rZzinfo_or_arcnamerVrerrrrrrwritestrs:    zZipFile.writestrcCs |dS)z2Call the "close()" method in case the user forgot.Nrurrrr__del__szZipFile.__del__c Cs||jdkrdS|jrtdzB|jdkr\|jr\|j"|jrJ|j|j | W5QRXW5|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<rr|rrr rgrr$rDrZ_write_end_recordrrrrr s z ZipFile.closecCs|jD]D}|j}|ddd>|dd>B|dB}|dd>|d d>B|ddB}g}|jtksr|jtkr||j||jd }d }n |j}|j}|jtkr||jd }n|j}|j} d} |r t| d } t j d d t |ddt |f|| } t } |j tkr$tt| } n|j tkr:tt| } t| |j} t| |j} zZ|\} }t tt| |j| |j||j |||j||t | t | t |jd|j|j|}Wnltk rttt| |j| |j|j |j |||j||t |j!t | t |jd|j|j|ft"j#dYnX|j$%||j$%| |j$%| |j$%|jq|j$&}t |j}||j'}|j'}d}|t(krd}n|tkrd}n|tkrd}|r$|j)st*|dt t+t,ddddd|||| }|j$%|t t-t.d|d}|j$%|t/|d}t/|d }t/|d }t t0t1dd||||t |j2 }|j$%||j$%|j2|j3dkrt|j$4|j$5dS)Nrrur$rr!rrr&r r)rr,QrrrzCentral directory offsetzCentral directory sizer,rrra)6r[rdrqrrpr1rnr3r8rFrr0rrerr_rrrrhrWrrrrgrirorbrlrmDeprecationWarningr|rjrBr}stderrr<rrZrZrrfr rJrKrGrHrFr]r\rir truncater)rrrrrr3rqrprnZ extra_datarrhrWrBrjrZpos2Z centDirCountZ centDirSizeZ centDirOffsetrZ zip64endrecZ zip64locrecrUrrrr5s $$                               zZipFile._write_end_recordcCs4|jdkst|jd8_|js0|js0|dS)Nrr)rnAssertionErrorrlr rrrrrrs zZipFile._fpclose)N)N)r^N)F)NN)NNN)NNN)NN)$rrrrr<rrrrtryrrqrrrrrrr]rbsetterr>rArrrrrrrrrrr rrrrrrrr sT ZN      b +  * 2 )hc@s4eZdZdZdeddfddZd d d Zd d ZdS)r zDClass to create ZIP archives with Python library files and packages.r^TrcCstj|||||d||_dS)N)r rrs)r r _optimize)rrr rrsoptimizerrrrs zPyZipFile.__init__rNcCst|}|rD||sD|jr@tj|r,dnd}td||fdStj|\}}tj|rhtj|d}tj|r|rd||f}n|}|jrtd|d|| |d d |\}} |jrtd | | || t t |} | d| D]} tj|| } tj| \} }tj| rRtjtj| dr|j| ||d q|d kr|r~|| s~|jrtd| q| | d d |\}} |jrtd | | || qn|jrtd|t t |D]} tj|| } tj| \} }|d kr|r,|| s,|jrtd| q| | d d |\}} |jrVtd | | || qnP|d dd krtd| |d d |\}} |jrtd| | || dS)aAdd 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. rrz%s %r skipped by filterfuncNz __init__.py%s/%szAdding package inasrZAdding) filterfunc.pyzfile %r skipped by filterfunczAdding files from directoryz.Files added with writepy() must end with ".py"z Adding file)ryrrhrrr|rr2isfile _get_codenamersortedlistdirremovesplitextwritepyr)rpathnamebasenamerZlabeldirr!ZinitnamerrZdirlistrBrrootZextrrrrs            zPyZipFile.writepyc sdfdd }|d}|d}tjj|dd}tjj|dd}tjj|d d}jdkr\tj|rt|jt|jkr|} } ntj|rt|jt|jkr|} |} ntj|rt|jt|jkr|} |} nvtj|rt|jt|jkr|} |} nD||rRt j j d kr4|} nt j j dkrH|} n|} |} n|} } njd krr|} |} n<|} jdkr|} n&jd kr|} nd j} t | tj| rt| jt|jks||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). rc sfddl}jrtd|z|j|d|dWn4|jk r`}zt|jWYdSd}~XYnXdS)NrZ CompilingT)doraiserF) py_compilerhr|compilePyCompileErrormsg)rrrerrrrr_compiles  z)PyZipFile._get_codename.._compilerz.pycr) optimizationrrrz"invalid value for 'optimize': {!r})rr)r) importlibutilcache_from_sourcerryrrrrr}rrformatr|r) rrrrZfile_pyZfile_pycZ pycache_opt0Z pycache_opt1Z pycache_opt2rrrZ archivenamerrrrsj             zPyZipFile._get_codename)rN)rrrrrrrrrrrrr s  RcCstt|ddS)a2 Given a path with elements separated by posixpath.sep, generate all parents of that path. >>> list(_parents('b/d')) ['b'] >>> list(_parents('/b/d/')) ['/b'] >>> list(_parents('b/d/f/')) ['b/d', 'b'] >>> list(_parents('b')) [] >>> list(_parents('')) [] rN) itertoolsislice _ancestry)rrrr_parentsRsrccs4|tj}|r0|tjkr0|Vt|\}}q dS)aR Given a path with elements separated by posixpath.sep, generate all elements of that path >>> list(_ancestry('b/d')) ['b/d', 'b'] >>> list(_ancestry('/b/d/')) ['/b/d', '/b'] >>> list(_ancestry('b/d/f/')) ['b/d/f', 'b/d', 'b'] >>> list(_ancestry('b')) ['b'] >>> list(_ancestry('')) [] N)r posixpathrzr)rtailrrrres rcCstt|j|S)zZ Return items in minuend not in subtrahend, retaining order with O(1) lookup. )r filterfalseset __contains__)ZminuendZ subtrahendrrr _differencesr csHeZdZdZeddZfddZddZdd Ze d d Z Z S) CompleteDirszk A ZipFile subclass that ensures that implied directories are always included in the namelist. cCs.tjtt|}dd|D}tt||S)Ncss|]}|tjVqdSr)rrz)rrrrrrsz-CompleteDirs._implied_dirs..)rchain from_iterablerr_deduper )namesparentsZas_dirsrrr _implied_dirsszCompleteDirs._implied_dirscs tt|}|t||Sr)rHr rr^r)rrrIrrrszCompleteDirs.namelistcCs t|Sr)rrrrrr _name_setszCompleteDirs._name_setcCs,|}|d}||ko||k}|r(|S|S)zx If the name represents a directory, return that name as a directory (with the trailing slash). rv)r)rr!rrZ dir_matchrrr resolve_dirszCompleteDirs.resolve_dircCsNt|tr|St|ts ||Sd|jkr.t}||}t|t||S)zl Given a source (filename or zipfile), return an appropriate CompleteDirs subclass. r^)rr r r __new__varsupdate)rrresrrrmakes    zCompleteDirs.make) rrrr staticmethodrrrrrrrMrrrIrr s   r cs,eZdZdZfddZfddZZS) FastLookupzV ZipFile subclass to ensure implicit dirs exist and are resolved rapidly. c s:tt|jW5QRSQRXtt||_|jSr) contextlibsuppressr)Z_FastLookup__namesrHrrrrIrrrs zFastLookup.namelistc s:tt|jW5QRSQRXtt||_|jSr)rrr)Z_FastLookup__lookuprHrrrrIrrrs zFastLookup._name_set)rrrrrrrMrrrIrrs rc@seZdZdZdZd#ddZeddZedd Zd d Z d d Z ddZ ddZ ddZ ddZddZddZddZddZddZeZed d!Zd"S)$ru A pathlib-compatible interface for zip files. Consider a zip file with this structure:: . ├── a.txt └── b ├── c.txt └── d └── e.txt >>> data = io.BytesIO() >>> zf = ZipFile(data, 'w') >>> zf.writestr('a.txt', 'content of a') >>> zf.writestr('b/c.txt', 'content of c') >>> zf.writestr('b/d/e.txt', 'content of e') >>> zf.filename = 'abcde.zip' Path accepts the zipfile object itself or a filename >>> root = Path(zf) From there, several path operations are available. Directory iteration (including the zip file itself): >>> a, b = root.iterdir() >>> a Path('abcde.zip', 'a.txt') >>> b Path('abcde.zip', 'b/') name property: >>> b.name 'b' join with divide operator: >>> c = b / 'c.txt' >>> c Path('abcde.zip', 'b/c.txt') >>> c.name 'c.txt' Read text: >>> c.read_text() 'content of c' existence: >>> c.exists() True >>> (b / 'missing.txt').exists() False Coercion to string: >>> str(c) 'abcde.zip/b/c.txt' z>{self.__class__.__name__}({self.root.filename!r}, {self.at!r})rcCst||_||_dSr)rrrat)rrrrrrr s z Path.__init__cCst|jj|jSr) functoolspartialrrArrrrrrA sz Path.opencCst|jdSNrv)rrrrrrrrr! sz Path.namec Os6|$}tj|f||W5QRSQRXdSr)rAr3 TextIOWrapperr>)rargskwargsstrmrrr read_text s zPath.read_textc Cs(|}|W5QRSQRXdSrr)rr$rrr read_bytes s zPath.read_bytescCst|jd|jdkSr )rrrr)rrrrr _is_child" szPath._is_childcCs t|j|Sr)rr)rrrrr_next% sz Path._nextcCs|j p|jdSr )rendswithrrrrr( sz Path.is_dircCs | Sr)rrrrris_file+ sz Path.is_filecCs|j|jkSr)rrrrrrrr. sz Path.existscCs.|stdt|j|j}t|j|S)NzCan't listdir a file)rr|rr(rrfilterr')rZsubsrrriterdir1 sz Path.iterdircCst|jj|jSr)rr2rrBrrrrr__str__7 sz Path.__str__cCs|jj|dS)Nr) _Path__reprrrrrrr: sz Path.__repr__cCs t|j|}||j|Sr)rr2rr(rr)raddnextrrrjoinpath= sz Path.joinpathcCs(t|jd}|r|d7}||Sr )rrrrr()rZ parent_atrrrparentC sz Path.parentN)r)rrrrr.rr]rAr!r%r&r'r(rr*rr,r-rr1 __truediv__r2rrrrrs*@   c sddl}d}|j|d}|jdd}|jdddd d |jd d d ddd|jdddddd|jddddd ||}|jdk r|j}t|d}|}W5QRX|rtd |tdn|j dk r|j }t|d}| W5QRXn|j dk r,|j \}}t|d}| |W5QRXn|jdk r|jd} |j} fddt| d\}| D]P} tj| } | stjtj| } | dtjtjfkrd} || | qfW5QRXdS) Nrz3A simple command-line interface for zipfile module.) descriptionT)Zrequiredz-lz--list zShow listing of a zipfile)metavarhelpz-ez --extractr)r5z zExtract zipfile into target dir)nargsr6r7z-cz--create+)zzzCreate zipfile from sourcesz-tz--testzTest if a zipfile is validr^z.The following enclosed file is corrupted: {!r}z Done testingcsptj|r|||tnPtj|rl|r8|||tt|D]$}|tj||tj||qFdSr) ryrrrrrrrr2)rVrzippathZnmaddToZiprrr<s s    zmain..addToZipr_r)argparseArgumentParserZadd_mutually_exclusive_group add_argument parse_argsZtestr rr|rr^rrrZcreatepopryrrrrr) r"r=r4parsergrouprrVZbadfilerZzip_namefilesrr:rr;rmainK s\                   rE__main__)N)N)rZbinasciirimportlib.utilrr3rryrrrrFr}rorrrr ImportErrorrr__all__ Exceptionrr rrrrrrrrrrrrrrr]r\rr[rLrMrNrOrPrQrRrar}r~rrrrZ_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_SIZErrrZ_CD_DISK_NUMBER_STARTZ_CD_INTERNAL_FILE_ATTRIBUTESZ_CD_EXTERNAL_FILE_ATTRIBUTESrrrrrZ_FH_EXTRACT_VERSIONZ_FH_EXTRACT_SYSTEMrZ_FH_COMPRESSION_METHODZ_FH_LAST_MOD_TIMEZ_FH_LAST_MOD_DATEZ_FH_CRCZ_FH_COMPRESSED_SIZEZ_FH_UNCOMPRESSED_SIZErrrGrHrErJrKrIZ_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_CENTDIRrYZStructr.r8r=r rXr9objectr rrrrrrrrrrrr4rrNr r rrdictfromkeysrr r rrrErrrrrsl             +=q& &AN/2 =