o w[eZ]@sDdZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZz ddlZejZWneyQdZejZYnwzddlZWn eycdZYnwzddlZWn eyudZYnwgdZGdddeZGdddeZeZZdZd Zd ZdZd Zd Z d Z!d Z"dZ#dZ$dZ%dZ&dZ'dZ(e )e'Z*dZ+dZ,dZ-dZ.dZ/dZ0dZ1dZ2d Z3dZ4dZ5dZ6e )e5Z7dZ8dZ9dZ:dZ;dZdZ?d Z@dZAdZBdZCd ZDdZEd ZFd ZGd!ZHd"ZId#ZJd$ZKd%ZLe )eKZMdZNdZOdZPdZQdZRdZSdZTdZUd ZVdZWdZXdZYd&ZZd'Z[e )eZZ\d(Z]d)Z^e )e]Z_dZ`dZadZbdZcdZddZedZfdZgd ZhdZid*Zje kd+Zld,d-Zmd.d/Znd0d1Zod2d3Zpd4d5ZqGd6d7d7erZsdatd8d9Zud:d;ZvGdd?d?Zxidd@ddAddBddBddBddBddCddDd dEddFddCd dGd dHd#dIdJdKdLdMdNdOZydPdQZzdsdRdSZ{dTdUZ|GdVdWdWZ}GdXdYdYZ~GdZd[d[ejZGd\d]d]ejZGd^d_d_ZGd`dadaeZdbdcZdddeZejZ dfdgZGdhdidieZGdjdkdkeZdsdldmZGdndodoZdsdpdqZedrkr edSdS)tzP 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.10/zipfile.pyr+src@seZdZdZdS)r zu Raised when writing a zipfile, the zipfile requires ZIP64 extensions and those extensions are disabled. N)rrr__doc__rrrrr /sr i -.?s<4s4H2LHsPK z<4s4B4HL2L5H2LsPK z <4s2B4HL2L2HsPKz<4sLQLsPKz <4sQ2H2L4QsPKiPK)filenameresultr@rrrr s     r c Csz ||tdWn ty|YSw|t}t|tkr#|Stt|\}}}}|tkr3|S|dks;|dkr?t d||tt d|t }t|t krV|Stt |\ }}} } } } } }}}|t krl|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)seeksizeEndCentDir64Locatorr>rBr0structr/structEndArchive64LocatorstringEndArchive64LocatorrsizeEndCentDir64structEndArchive64stringEndArchive64_ECD_SIGNATURE_ECD_DISK_NUMBER_ECD_DISK_START_ECD_ENTRIES_THIS_DISK_ECD_ENTRIES_TOTAL _ECD_SIZE _ECD_OFFSET)fpinoffsetendrecdatasigdisknoreloffdisksszcreate_version read_versiondisk_numdisk_dirdircount dircount2dirsize diroffsetrrr _EndRecData64s<      rhc Csb|dd|}z |t dWn tyYdSw|}t|tkrU|ddtkrU|dddkrUtt |}t |}| d| |tt |t |St |dtd}||d|}|t}|dkr|||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)rHtellsizeEndCentDirr>rBr0stringEndArchiverJr/structEndArchivelistr1rhmaxrfind_ECD_COMMENT_SIZE) rWfilesizerZrYmaxCommentStartr7recData commentSizecommentrrrr=sB         r=c@s\eZdZdZdZdddZddZdd d Zd d ZddZ e dddddZ ddZ d S)r z>Class with attributes describing each file in the ZIP archive.) orig_filenamerF date_time compress_type_compresslevelrvr3 create_systemr`extract_versionreserved flag_bitsvolume internal_attr external_attr header_offsetCRC compress_size file_size _raw_timeNoNamerrrrrcCs||_|td}|dkr|d|}tjdkr%tj|vr%|tjd}||_||_|ddkr5tdt |_ d|_ d|_ d|_ tjdkrJd|_nd|_t|_t|_d|_d|_d|_d|_d|_d|_d|_dS)Nr/rz+ZIP does not support timestamps before 1980r-win32r)rwfindchrossepreplacerFrx ValueErrorrryrzrvr3sysplatformr{DEFAULT_VERSIONr`r|r}r~rrrrr)selfrFrx null_byterrr__init__Zs4    zZipInfo.__init__cCsd|jj|jfg}|jtkr|dt|j|j|jd?}|jd@}|r2|dt ||r;|d|| }|rD|j rL|d|j |rQ|j rd|jtks\|j |j krd|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__rrFryrr1compressor_namesgetrstatfilemodeis_dirrrr2)rrGhiloisdirrrr__repr__s,         zZipInfo.__repr__NcCsx|j}|ddd>|dd>B|dB}|dd>|d d>B|ddB}|jd @r3d}}}n |j}|j}|j}|j}d} |d urM|tkpL|tk}|rad } |t| dt | d ||}|tksi|tkru|sot 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.rrr$rr!rrr&r rNzz |jd|jfWSty|jd|jdBfYSw)Nasciiutf-8)rFencoder~UnicodeEncodeErrorrrrrrs  zZipInfo._encodeFilenameFlagscCs6|j}tj}t|dkr|d|dd\}}|dt|kr'td||f|dkr|d|d}zD|jdvrLd}|d|dd\|_|dd}|jd krdd }|d|dd\|_|dd}|jd krvd }|d|dd\|_Wntjytd |d dw||dd}t|dks dSdS)Nr r,z"Corrupt extra field %04x (size=%d)r)lrz File sizez|_|rd|_|jd O_|S|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"rr;)rr;rNrrrr)) isinstancerPathLikefspathrS_ISDIRst_modetime localtimest_mtimepathnormpath splitdriveraltseprrst_size) clsrFarcnamerstrmtimerxzinforrr from_files4       zZipInfo.from_filecCs|jddkS)z2Return True if this archive member is a directory.rrFrrrrrszZipInfo.is_dir)rrN) rrrr __slots__rrrrr classmethodrrrrrrr @s + . %r cCs0tdD]}|d@r|d?dA}q|dL}q|S)Nrrl q[)range)crcr;rrr_gen_crcs  rcsldddtdurttttdatfddfdd|D]}|q&fd d }|S) NixV4igE#ixV4cs|d?||Ad@AS)z(Compute the CRC32 primitive on one byte.rr)chr)crctablerrcrc326sz_ZipDecrypter..crc32cs<|d@d@ddd@d?dS)Nrrirr)c)rkey0key1key2rr update_keys:s z"_ZipDecrypter..update_keyscsNt}|j}|D]}dB}|||dAd?d@N}|||qt|S)zDecrypt a bytes object.rrrr) bytearrayr1bytes)rZrGr1rk)rrrr decrypterDs z _ZipDecrypter..decrypter) _crctablernmaprr)pwdprr)rrrrrrr _ZipDecrypter,s   rc@s,eZdZddZddZddZddZd S) LZMACompressorcCs d|_dSr)_comprrrrrTs zLZMACompressor.__init__cCsFtdtji}tjtjttj|gd|_tdddt ||S)Nidfiltersz     zZipExtFile.readcCsD|jdurdSt||j|_|jr|j|jkr td|jdSdS)NzBad CRC-32 for file %r)rSrrTrMrrR)rnewdatarrr _update_crcs zZipExtFile._update_crccCs|dus|dkr-|j|jd}d|_d|_|js+||j}|r(||7} |S|jr|S||j}|t|jkrF|j|j|}||_|S|t|j}|j|jd}d|_d|_|dkr|js||}|t|kr~||_||_||d|7} |S|r||7} |S|jrb|S)z7Read up to n bytes with at most one read() system call.Nrr-)rNrOrMrorpr0)rr;rqrZrrrrrread1sH     zZipExtFile.read1cCs |js|dkr dS|jtkr$|jj}|t|kr#|||t|7}n||}|jtkr5|jdk|_n<|jtkrat ||j }|j ||}|jj pT|jdkoT|jj |_|jr`||j 7}n|j |}|jj po|jdk|_|d|j}|jt|8_|jdkrd|_|||S)Nrr-T)rMrIrrLunconsumed_tailr0_read2rrJro MIN_READ_SIZErrrrKrtr:rrrros8          zZipExtFile._read1cCsd|jdkrdSt||j}t||j}|j|}|jt|8_|s&t|jdur0||}|S)Nrr-) rJrorxminrFrBr0EOFErrorr[r:rrrrws      zZipExtFile._read2cs>z|jr|jWtdSWtdStwr)rHrFr3superrrrrr3 s  zZipExtFile.closecCs|jrtd|jSNrm)rcrrUrrrrr0szZipExtFile.seekablercCsJ|jrtd|jstd|}|dkr|}n|dkr#||}n|dkr-|j|}ntd||jkr9|j}|dkr?d}||}||j}|dkrY|t|j krY||_d}n1|dkr|j |j |j |_|j|_|j|_d|_ d|_t|j|_d|_|}|jdur||dkrt|j|}||||8}|dks|S) Nzseek on closed file.!underlying stream is not seekablerrrzCwhence must be os.SEEK_SET (0), os.SEEK_CUR (1), or os.SEEK_END (2)r-F)rcrrUrfUnsupportedOperationrjrXrOr0rNrFrHrVrYrTrWrJrKr)rIrLrMr[r\ry MAX_SEEK_READrB)rrXr8curr_posnew_pos read_offset buff_offsetread_lenrrrrHsN         zZipExtFile.seekcCs>|jrtd|jstd|j|jt|j|j }|S)Nztell on closed file.r~) rcrrUrfrrXrKr0rNrO)rfileposrrrrjJs  zZipExtFile.tell)NFr>rr=)rrrrrprxrrr\rrhrlrnrBrtrurorwr3r0rHrj __classcell__rrr|rrE s, 3   # %$  2rEcs@eZdZddZeddZddZddZfd d ZZ S) _ZipWriteFilecCs8||_||_||_t|j|j|_d|_d|_d|_ dSr@) _zinfo_zip64_zipfiler%ryrz _compressor _file_size_compress_size_crc)rzfrrrrrrTs z_ZipWriteFile.__init__cCs|jjSr)rr@rrrrrF^sz_ZipWriteFile._fileobjcCsdS)NTrrrrrwritablebz_ZipWriteFile.writablecCs|jrtdt|ttfrt|}nt|}|j}|j|7_t ||j |_ |j r:|j |}|j t|7_ |j||Sr})rcrrrrr0 memoryviewnbytesrrrrrrrFrB)rrZrrrrrBes   z_ZipWriteFile.writec sf|jrdSzt|jr(|j}|jt|7_|j||j|j _ n|j |j _ |j |j _ |j |j _|j jd@r_|jrBdnd}|jt|t|j j |j j |j j|j|j_n7|jst|j tkrktd|jtkrttd|j|j_|j|j j|j|j |j|j|jj|jj|j |j |jj|j j<Wd|j_ dSd|j_ w)Nrzd!d"d#d$Zd?d%d&Zd@d'd(ZdAd)d*Zed+d,Zd-d.Zd/d0Z  dAd1d2Z d@d3d4Zd5d6Z d7d8Z!d9d:Z"d;d<Z#dS)Br 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 Cs@|dvrtdt|||_d|_d|_i|_g|_||_||_||_ d|_ d|_ ||_ t |tjr8t|}t |trpd|_||_ddd d dd d d }||} z t|||_Wntym||vrl||}YqQw n d|_||_t|dd|_d|_t|_d|_d|_z|dkr|WdS|dvrd|_z|j |_!Wnt"tfyt#|j|_d|_!d|_YWdSwz |j$|j!WWdSt"tfyd|_YWdSw|dkr z||j$|j!WWdSt%y |j$ddd|_|j |_!YWdSwtd|j} d|_|&| )z]Open the ZIP file with mode read 'r', write 'w', exclusive create 'x', or append 'a'.)rwxaz+ZipFile requires mode 'r', 'w', 'x', or 'a'FrNr-rCw+bx+br+bwbxb)rrrrrrrTrrRr)rrrrz"Mode must be 'r', 'w', 'x', or 'a')'rr _allowZip64 _didModifydebugrrrr$rQr_comment_strict_timestampsrrrrstr _filePassedrFrfrEr@r>getattr _fileRefCnt threadingRLockr.rUr/_RealGetContentsrjrrZr?rHr_fpclose) rr1rQr allowZip64r$rmodeDictrr@rrrrs         zZipFile.__init__cCs|Srrrrrr __enter__rzZipFile.__enter__cCs |dSrr3)rtypevalue tracebackrrr__exit__ zZipFile.__exit__cCsd|jj|jjfg}|jdur2|jr|d|jn |jdur)|d|j|d|jn|d|dd|S)Nraz file=%rz filename=%rz mode=%rrbrr) rrrr@rr1rFrQr2rdrrrr"s     zZipFile.__repr__c Cs|j}zt|}Wn tytdw|std|jdkr$t||t}|t}|t|_ |t ||}|t t krE|t t8}|jdkrU||}td||||||_|jdkrctd||jd||}t|}d}||kra|t} t| tkrtdtt| } | ttkrtd|jdkrt| || t} | d } | d @r| d } n| d } t| } || t| _|| t | _!| t"| _#| dd \ | _$| _%| _&| _'| _(| _)} }| _*| _+| _,| j&t-krt.d| j&d| dd\| _/| _0| _1| | _2|d?d|d ?d@|d@| d?| d ?d@| d@df| _3| 4| j#|| _#|j56| | |j7| j8<|t| t| t| t }|jdkr[td|||ks{dSdS)z/Read in the table of contents for the ZIP file.zFile is not a zip filerrzgiven, inferred, offsetrz Bad offset for central directoryzTruncated central directoryz&Bad magic number for central directoryr!rrcp437rzzip file version %.1fr%r(r+r$rrr&rtotalN)9r@r=r>rrprintrUrV _ECD_COMMENTr _ECD_LOCATIONrPrOrMrIrrHrBrfBytesIOsizeCentralDirr0rJr/structCentralDir _CD_SIGNATUREstringCentralDir_CD_FILENAME_LENGTHdecoder _CD_EXTRA_FIELD_LENGTHr3_CD_COMMENT_LENGTHrv_CD_LOCAL_HEADER_OFFSETrr`r{r|r}r~ryrrrMAX_EXTRACT_VERSIONrrrrrrxrrr1rrF)rr@rYsize_cd offset_cdconcatinferredrZrcentdirrFflagsrtdrrrr0s                             zZipFile._RealGetContentscCsdd|jDS)z+Return a list of file names in the archive.cSsg|]}|jqSrr).0rZrrr sz$ZipFile.namelist..rrrrrnamelistszZipFile.namelistcCr6)zJReturn a list of class ZipInfo instances for files in the archive.rrrrrinfolistzZipFile.infolistcCsLtdd|d|jD]}d|jdd}td|j||jf|dq dS)z+Print a table of contents for the zip file.z%-46s %19s %12s)z File Namez Modified Size)r1z%d-%02d-%02d %02d:%02d:%02dNr"z %-46s %s %12d)rrrxrFr)rr1rdaterrrprintdirs zZipFile.printdirc Csxd}|jD]4}z$||jd}||r ||sWdn1s%wYWqty9|jYSwdS)z%Read all the files and check the CRC.irN)rrErFrBr)r chunk_sizerfrrrtestzips    zZipFile.testzipcCs$|j|}|durtd||S)z,Return the instance of ZipInfo given 'name'.Nz(There is no item named %r in the archive)rrKeyError)rrRinforrrgetinfos zZipFile.getinfocCs8|rt|tstdt|j|r||_dSd|_dS)z)Set default password for encrypted files.pwd: expected bytes, got %sN)rr TypeErrorrrr)rrrrr setpasswords   zZipFile.setpasswordcCr6)z.The comment text associated with the ZIP file.)rrrrrrvrzZipFile.commentcCs^t|tstdt|jt|tkr'ddl}|jdtdd|dt}||_ d|_ dS)Nzcomment: expected bytes, got %srz3Archive comment is too long; truncating to %d bytesr stacklevelT) rrrrrr0ZIP_MAX_COMMENTwarningswarnrr)rrvrrrrrvs    cCs:||d| }|WdS1swYdS)zReturn file bytes for name.rNrErB)rrRrr@rrrrBs$z ZipFile.readF force_zip64c s|dvrtd|rt|tstdt|j|r"|dkr"tdjs)tdt|tr1|}n|dkrBt|}j|_ j |_ n |}|dkrRj ||dSjrYtdjd 7_tj|jjjfd d }z|t}t|tkrtd tt|}|ttkrtd ||t}|tr||t|jd@rt d|jd@rt d|t!d@r|"d} n|"d} | |j#krtd|j#|f|jd @} | r|s߈j$}|st%d|nd}t&||||dWS|')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. >rrzopen() requires mode "r" or "w"rrz'pwd is only supported for reading filesz2Attempt to use ZIP archive that was already closedrr9rcsjSr)r/rrrrszZipFile.open..zTruncated file headerz Bad magic number for file header z$compressed patched data (flag bit 5)@zstrong encryption (flag bit 6)rrrz/File name in directory %r and header %r differ.z6File %r is encrypted, password required for extractionNT)(rrrrrrr@r rryr$rzr_open_to_writer/rr*rrr.rBsizeFileHeaderr0rrJr/r _FH_SIGNATUREr_FH_FILENAME_LENGTH_FH_EXTRA_FIELD_LENGTHr~r_FH_GENERAL_PURPOSE_FLAG_BITSrrwrrrEr3) rrRrQrrrzef_filefheaderfname fname_str is_encryptedrrrrEs|                z ZipFile.opencCs|r |js td|jrtdd|_d|_d|_|jtkr%|jdO_|js/|jdO_|j s5d|_ |jo@|p@|j dt k}|jrK|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.rrrg?T)rrr/rrr~ryrrUrrrr@rHrrjr _writecheckrrBrr)rrrrrrrr*s4     zZipFile._open_to_writecCs*|dur t}nt|}||||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)rgetcwdr_extract_member)rmemberrrrrrextractSs  zZipFile.extractcCsH|dur|}|durt}nt|}|D] }||||qdS)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)rrrrr)rrmembersrr]rrr extractall`s  zZipFile.extractallcCs^|j}|sd}t|dt|}||_||}dd||D}|dd|D}|S)z;Replace bad characters and remove trailing dots from parts.z:<>|"?*_css|]}|dVqdS).N)rstriprrrrr {z1ZipFile._sanitize_windows_name..css|]}|r|VqdSrrrrrrr}s)!_windows_illegal_name_trans_tabler maketransr0 translatesplitr2)rrpathseptableillegalrrr_sanitize_windows_nameqs zZipFile._sanitize_windows_namec st|ts ||}|jdtjj}tjjr!|tjjtjj}tj |d}dtjj tjj ftjj fdd| tjjD}tjjdkrS||tjj}tj ||}tj|}tj|}|rstj|sst||rtj|st||S|j||d,}t|d}t||Wd n1swYWd |SWd |S1swY|S) zbExtract the ZipInfo object 'member' to a physical file on the path targetpath. rrrc3s|] }|vr|VqdSrrrinvalid_path_partsrrrs z*ZipFile._extract_member..\rrN)rr rrFrrrrrrcurdirpardirr2r rrdirnameexistsmakedirsrrmkdirrEshutil copyfileobj)rr targetpathrr upperdirssourcetargetrrrrs<  &      ( zZipFile._extract_membercCs|j|jvrddl}|jd|jdd|jdvrtd|js$tdt|j|j sOd}t |j t kr8d }n|j tkr@d }n|jtkrGd }|rQt|d dSdS) z6Check for errors before writing a file to the archive.rNzDuplicate name: %rrrrrrz&write() requires mode 'w', 'x', or 'a'z4Attempt to write ZIP archive that was already closed Files countFilesizez Zipfile size would require ZIP64 extensions)rFrrrrQrr@rryrr0rZIP_FILECOUNT_LIMITrrrr )rrrrequires_zip64rrrrs0     zZipFile._writecheckc Cs|jstd|jrtdtj|||jd}|r"d|_d|_n|dur*||_ n|j |_ |dur6||_ n|j |_ |r|j H|jrL|j|j|j|_|j tkr^|jdO_||d|_|j|||j|j<|j|d|j|_WddS1swYdSt|d .}||d }t||d Wdn1swYWddSWddS1swYdS) zLPut the bytes from filename into the archive under the name arcname.7Attempt to write to ZIP archive that was already closedz>Can't write to ZIP archive while an open writing handle existsrrNrTFrCri ) r@rr/r rrrrrryrrzr$r.rUrHrrjrrr~rrrr1rrFrBrrErr)rrFrryr$rsrcdestrrrrBsJ     "Pz ZipFile.writec Cs2t|tr |d}t|ts|dd>B|dB}|dd>|d d>B|ddB}g}|jtks8|jtkrI||j||jd }d }n|j}|j}|jtkr]||jd }n|j}|j} d} |rt| d } t j d d t |ddt |g|R| } t } |j tkrtt| } n |j tkrtt| } t| |j} t| |j} |\} }t tt| |j| |j||j |||j||t | t | t |jd|j|j|}|j||j| |j| |j|jq|j }t |j}||j!}|j!}d}|t"krd}n|tkrd}n|tkrd}|rV|j#s#t$|dt t%t&ddddd|||| }|j|t t't(d|d}|j|t)|d}t)|d }t)|d }t t*t+dd||||t |j, }|j||j|j,|j-dkr~|j.|j/dS)Nrrr$rr!rrr&r rrr,Qrr!zCentral directory offsetzCentral directory sizer#,rrr)0rrxrrrr1rr3r<rJrr0rryrrorrrr|r`rrrr{r}rrvrrr@rBrjrr$rr rNrOrKrLryrmrlrrQtruncater)rrrrrr3rrr extra_datarr|r`rFr~rpos2 centDirCount centDirSize centDirOffsetr% zip64endrec zip64locrecrYrrrr,5s $$                                    zZipFile._write_end_recordcCs<|jdksJ|jd8_|js|js|dSdSdS)Nrr)rrr3rArrrrs   zZipFile._fpcloser)rN)F)NN)NNN)$rrrrr@rrrrrrrrrrrrrrrvsetterrBrErrrrrrrrBr*r+r3r,rrrrrr sR ZP        b )  * 2 ) ^r c@s4eZdZdZdeddfddZd d d Zd d ZdS)r zDClass to create ZIP archives with Python library files and packages.rTrcCstj|||||d||_dS)N)rQrr)r r _optimize)rr1rQrroptimizerrrrs  zPyZipFile.__init__rNcCst|}|r"||s"|jr tj|rdnd}td||fdStj|\}}tj|r-tj|d}tj|r|rGd||f}n|}|jrStd|d|| |d d |\}} |jrgtd | | || t t |} | d| D]Y} tj|| } tj| \} }tj| rtjtj| dr|j| ||d q{|d kr|r|| s|jrtd| q{| | d d |\}} |jrtd | | || q{dS|jrtd|t t |D]D} tj|| } tj| \} }|d kr*|r|| s|jrtd| q| | d d |\}} |jr$td | | || qdS|d dd kr:td| |d d |\}} |jrOtd| | || 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. rr1z%s %r skipped by filterfuncNz __init__.py%s/%szAdding package inasrAdding) filterfunc.pyzfile %r skipped by filterfunczAdding files from directoryz.Files added with writepy() must end with ".py"z Adding file)rrrrrrr r2isfile _get_codenamerBsortedlistdirremovesplitextwritepyr)rpathnamebasenamer>labeldirrRinitnamerrdirlistrFrrootextrrrrFs                 zPyZipFile.writepyc sdfdd }|d}|d}tjj|dd}tjj|dd}tjj|d d}jdkrtj|rCt|jt|jkrC|} } ntj|rZt|jt|jkrZ|} |} ntj|rqt|jt|jkrq|} |} nztj|rt|jt|jkr|} |} nc||rt j j d kr|} n t j j dkr|} n|} |} nH|} } nCjd kr|} |} n|} jdkr|} njd kr|} n d 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 sbddl}jr td|z |j|d|dWdS|jy0}z t|jWYd}~dSd}~ww)Nr CompilingT)doraiser9F) py_compilerrcompilePyCompileErrormsg)r1r9rQerrrrr_compiles  z)PyZipFile._get_codename.._compiler?z.pycr) optimizationrrrz"invalid value for 'optimize': {!r})r9r:Nr>) importlibutilcache_from_sourcer8rrr@rrrrr9formatrr ) rrGrHrVfile_pyfile_pyc pycache_opt0 pycache_opt1 pycache_opt2rrrT archivenamerrrrAs`                zPyZipFile._get_codename)rN)rrrrrrrFrArrrrr s  Rr cCstt|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_parentsHsreccsN|tj}|r!|tjkr%|Vt|\}}|r#|tjksdSdSdSdS)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 posixpathrr )rtailrrrrd[s  rdcCstt|j|S)zZ Return items in minuend not in subtrahend, retaining order with O(1) lookup. )rb filterfalseset __contains__)minuend subtrahendrrr _differenceusrmcsTeZdZdZeddZfddZddZdd Zfd d Z e 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)rfr)rrrrrrrz-CompleteDirs._implied_dirs..)rbchain from_iterablerre_deduperm)namesparentsas_dirsrrr _implied_dirsszCompleteDirs._implied_dirscs tt|}|t||Sr)r{rnrrnru)rrrr|rrrszCompleteDirs.namelistcCs t|Sr)rirrrrr _name_setrzCompleteDirs._name_setcCs,|}|d}||vo||v}|r|S|S)zx If the name represents a directory, return that name as a directory (with the trailing slash). r)rv)rrRrrr dir_matchrrr resolve_dirs zCompleteDirs.resolve_dircsDzt|WSty!|dr||vrt|dYSw)z6 Supplement getinfo for implied dirs. rr)r{rrendswithrvr )rrRr|rrrs zCompleteDirs.getinfocCs8t|tr|St|ts||Sd|jvrt}||_|S)zl Given a source (filename or zipfile), return an appropriate CompleteDirs subclass. r)rrnr rQr)rrrrrmakes   zCompleteDirs.make) rrrr staticmethodrurrvrxrrrzrrrr|rrn}s    rncs,eZdZdZfddZfddZZS) FastLookupzV ZipFile subclass to ensure implicit dirs exist and are resolved rapidly. cFtt |jWdS1swYtt||_|jSr) contextlibsuppressrZ_FastLookup__namesr{r|rrr|rrr  zFastLookup.namelistcr}r)r~rrZ_FastLookup__lookupr{r|rvrr|rrrvrzFastLookup._name_set)rrrrrrvrrrr|rr|s r|cOst|d||fS)Nr)rf text_encoding)encodingargskwargsrrr_extract_text_encodingsrc@seZdZdZdZd'ddZd(ddd d Zed d Zed dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zd#d$ZeZed%d&ZdS))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 = 'mem/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('mem/abcde.zip', 'a.txt') >>> b Path('mem/abcde.zip', 'b/') name property: >>> b.name 'b' join with divide operator: >>> c = b / 'c.txt' >>> c Path('mem/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: >>> import os >>> str(c).replace(os.sep, posixpath.sep) 'mem/abcde.zip/b/c.txt' At the root, ``name``, ``filename``, and ``parent`` resolve to the zipfile. Note these attributes are not valid and will raise a ``ValueError`` if the zipfile has no filename. >>> root.name 'abcde.zip' >>> str(root.filename).replace(os.sep, posixpath.sep) 'mem/abcde.zip' >>> str(root.parent) 'mem' z>{self.__class__.__name__}({self.root.filename!r}, {self.at!r})rcCst||_||_dS)aX Construct a Path from a ZipFile or filename. Note: When the source is an existing ZipFile object, its type (__class__) will be mutated to a specialized type. If the caller wishes to retain the original type, the caller should either create a separate ZipFile object or pass a filename. N)r|rzrMat)rrMrrrrr! s z Path.__init__rNrcOs|rt||d}|s|dkrt||jj|j||d}d|vr0|s*|r.td|St|i|\}}}t j ||g|Ri|S)z Open this entry as text or binary following the semantics of ``pathlib.Path.open()`` by passing arguments through to io.TextIOWrapper(). rrrbz*encoding args invalid for binary operation) rIsADirectoryErrorrFileNotFoundErrorrMrErrrrf TextIOWrapper)rrQrrrzip_modestreamrrrrrE. sz Path.opencCst|jjp |jjSr)pathlibrrrRrFrrrrrRB z Path.namecCst|jj|jSr)rrrMrFjoinpathrrrrrrFF rz Path.filenamecOsZt|i|\}}}|jd|g|Ri| }|WdS1s&wYdS)Nr)rrErB)rrrrstrmrrr read_textJ s$zPath.read_textcCs6|d }|WdS1swYdS)NrCr)rrrrr read_bytesO s $zPath.read_bytescCst|jd|jdkSNr)rfrrr)rrrrr _is_childS szPath._is_childcCs||j|Sr)rrM)rrrrr_nextV rDz Path._nextcCs|j p |jdSr)rryrrrrrY sz Path.is_dircCs|o| Sr)rrrrrris_file\ z Path.is_filecCs|j|jvSr)rrMrvrrrrr_ sz Path.existscCs.|stdt|j|j}t|j|S)NzCan't listdir a file)rrrrrMrfilterr)rsubsrrriterdirb s z Path.iterdircCst|jj|jSr)rfr2rMrFrrrrr__str__h rz Path.__str__cCs|jj|dS)Nr) _Path__reprr[rrrrrk rDz Path.__repr__cGs&tj|jg|R}||j|Sr)rfr2rrrMrx)rothernextrrrrn sz Path.joinpathcCs6|js|jjSt|jd}|r|d7}||Sr)rrFparentrfrrr)r parent_atrrrrt s  z Path.parent)r)r)rrrrrrrErrRrFrrrrrrrrrrr __truediv__rrrrrrs.M    rc s ddl}d}|j|d}|jdd}|jdddd d |jd d d ddd|jdddddd|jddddd ||}|jdurm|j}t|d }|}Wdn1sYwY|rgtd |tddS|j dur|j }t|d }| WddS1swYdS|j dur|j \}}t|d}| |WddS1swYdS|jdur|jd} |j} fddt| d1}| D]%} tj| } | stjtj| } | dtjtjfvrd} || | qWddS1swYdSdS) Nrz3A simple command-line interface for zipfile module.) descriptionT)requiredz-lz--list zShow listing of a zipfile)metavarhelpz-ez --extractr)rz zExtract zipfile into target dir)nargsrrz-cz--create+)zzzCreate zipfile from sourcesz-tz--testzTest if a zipfile is validrz.The following enclosed file is corrupted: {!r}z Done testingcsvtj|r|||tdStj|r7|r|||tt|D]}|tj||tj||q$dSdSr) rrr@rBrrrBrCr2)rrzippathnmaddToZiprrr s   zmain..addToZiprr)argparseArgumentParseradd_mutually_exclusive_group add_argument parse_argstestr rrr[rnrrrcreatepoprrrHrrr) rrrparsergroupr'rbadfilerzip_namefilesrrrrrmain~ sh              "    "     $r__main__r)rbinasciiimportlib.utilrXrfrbrrfrrrJrrrr~rrr ImportErrorrr__all__ Exceptionrr rrrr$rrrrrrrrrrrmrlrrkrPrQrRrSrTrUrVrqrrrrrr_CD_CREATE_VERSION_CD_CREATE_SYSTEM_CD_EXTRACT_VERSION_CD_EXTRACT_SYSTEM _CD_FLAG_BITS_CD_COMPRESS_TYPE_CD_TIME_CD_DATE_CD_CRC_CD_COMPRESSED_SIZE_CD_UNCOMPRESSED_SIZErrr_CD_DISK_NUMBER_START_CD_INTERNAL_FILE_ATTRIBUTES_CD_EXTERNAL_FILE_ATTRIBUTESrrrrr_FH_EXTRACT_VERSION_FH_EXTRACT_SYSTEMr_FH_COMPRESSION_METHOD_FH_LAST_MOD_TIME_FH_LAST_MOD_DATE_FH_CRC_FH_COMPRESSED_SIZE_FH_UNCOMPRESSED_SIZErrrKrLrIrNrOrM_CD64_SIGNATURE_CD64_DIRECTORY_RECSIZE_CD64_CREATE_VERSION_CD64_EXTRACT_VERSION_CD64_DISK_NUMBER_CD64_DISK_NUMBER_START_CD64_NUMBER_ENTRIES_THIS_DISK_CD64_NUMBER_ENTRIES_TOTAL_CD64_DIRECTORY_SIZE_CD64_OFFSET_START_CENTDIRrStructr.r<rAr rhr=objectr rrrrrrrr%r)r*r?rgrErr r rerddictfromkeysrqrmrnr|rrrrrrrrs              +=\&      (KUw/<  . =