o
0ib @ sL d Z ddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
Z
ddlZddlZddlZz ddlZejZW n eyU dZejZY nw zddlZW n eyg dZY nw zddlZW n eyy dZY nw g dZG dd deZG dd deZe ZZdZ d Z d Z dZ d
Z d Z!d Z"d
Z#dZ$dZ%dZ&dZ'dZ(dZ)e *e(Z+dZ,dZ-dZ.dZ/dZ0dZ1dZ2dZ3d
Z4dZ5dZ6d Z7e *e6Z8dZ9dZ:dZ;dZdZ?dZ@d
ZAdZBd ZCd ZDd ZEd ZFd ZGd ZHd!ZId"ZJd#ZKd$ZLd%ZMe *eLZNdZOdZPdZQdZRdZSdZTdZUdZVd
ZWdZXd ZYd ZZd&Z[d'Z\e *e[Z]d(Z^d)Z_e *e^Z`dZadZbdZcdZddZedZfdZgdZhd
ZidZjd*Zke ld+Zmd,d- Znd.d/ Zod0d1 Zpd2d3 Zqd4d5 ZrG d6d7 d7esZtdaud8d9 Zvd:d; ZwG dd? d?Zyi dd@ddAddBddBddBddBddCddDd
dEddFd dCd dGd dHd#dIdJdKdLdMdNdOZzdPdQ Z{dsdRdSZ|dTdU Z}G dVdW dWZ~G dXdY dYZG dZd[ d[ejZG d\d] d]ejZG d^d_ d_ZG d`da daeZdbdc Zddde ZejZ dfdg ZG dhdi dieZG djdk dkeZdsdldmZG dndo doZdsdpdqZedrkr$e dS dS )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 e Zd ZdS )r N)__name__
__module__ __qualname__ r r ./opt/alt/python310/lib64/python3.10/zipfile.pyr , s r 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
0 s r
i - . ? s <4s4H2LHs PK z<4s4B4HL2L5H2Ls PK
z <4s2B4HL2L2Hs PKz<4sLQLs PKz
<4sQ2H2L4Qs PKiPK r0 r: structr/ structEndArchive64LocatorstringEndArchive64Locatorr sizeEndCentDir64
startswithstringEndArchive64structEndArchive64_ECD_SIGNATURE_ECD_DISK_NUMBER_ECD_DISK_START_ECD_ENTRIES_THIS_DISK_ECD_ENTRIES_TOTAL _ECD_SIZE _ECD_OFFSET
_ECD_LOCATION)fpinoffsetendrecdataZsigZdisknoZreloffZdisksZextraszZszcreate_versionZ read_versionZdisk_numZdisk_dirZdircountZ dircount2ZdirsizeZ diroffsetr r r
_EndRecData64 sV
rZ c C s` | dd | }z | t d W n
ty Y dS w | }t|tkrV|dd tkrV|dd dkrVtt |}t
|}| d | |t t | |t |S t
|d t d}| |d | }|t}|dkr|||t }t|tkrdS t
tt |}|t }||t |t | }| | | || 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- i )rE tellsizeEndCentDirr: r> r0 stringEndArchiverF r/ structEndArchivelistr1 rZ maxrfind_ECD_COMMENT_SIZE) rU ZfilesizerX rW ZmaxCommentStartr5 ZrecDataZ commentSizecommentr r r r9 s>
r9 c @ s\ e Zd ZdZdZdddZdd Zdd
d Zd d
Zdd Z e
dddddZ dd Z d S )r
z>Class with attributes describing each file in the ZIP archive.)
orig_filenamerB date_time
compress_type_compresslevelrd r3
create_systemrY extract_versionreserved flag_bitsvolume
internal_attr
external_attr
header_offsetCRC
compress_size file_size _raw_time _end_offsetNoName r r r r r c C s || _ |td}|dkr|d| }tjdkr%tj|v r%|tjd}|| _|| _|d dk r5tdt | _
d | _ d| _ d| _
tjdkrJd| _nd| _t| _t| _d| _d| _d| _d| _d| _d| _d| _d | _d S )Nr /rx z+ZIP does not support timestamps before 1980r- Zwin32r ) re findchrossepreplacerB rf
ValueErrorr rg rh rd r3 sysplatformri DEFAULT_VERSIONrY rj rk rl rm rn ro rr rs ru )selfrB rf Z null_byter r r __init__l s6
zZipInfo.__init__c C s d| j j| jf g}| 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) r z filemode=%rz external_attr=%#xz
file_size=%rz compress_size=%r> ) __class__r rB rg r r1 compressor_namesgetro statfilemodeis_dirrs rr r2 )r rC hiloisdirr r r __repr__ s,
zZipInfo.__repr__Nc C sx | j }|d d d> |d d> B |d B }|d d> |d d> B |d d B }| jd
@ r3d } }}n | j}| j}| j}| j}d} |d u rM|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.r rx r$ r r! r r r& r r Nz z
| j d| jfW S ty | j d| jdB f Y S w )Nasciiutf-8 )rB encoderl UnicodeEncodeErrorr r r r r s
z ZipInfo._encodeFilenameFlagsc C s6 | j }tj}t|dkr|d|d d \}}|d t|kr'td||f |dkr|d|d }zD| jdv rLd}|d|d d \| _|dd }| jd krdd
}|d|d d \| _|dd }| jd krvd }|d|d d \| _W n tjy td | d
d w ||d d }t|dks d S d S )Nr r, z"Corrupt extra field %04x (size=%d)r )l r z 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).
r r" rx rw ; )r r ; r Nr ry r r) )
isinstancer| PathLikefspathr S_ISDIRst_modetime localtimest_mtimepathnormpath
splitdriver} altsepro rs st_size) clsrB arcnamer str mtimerf zinfor r r from_file s4
zZipInfo.from_filec C s | j d dkS )z2Return True if this archive member is a directory.ry rB r r r r r % s zZipInfo.is_dir)rv rw N)
r r r r __slots__r r r r r classmethodr r r r r r r
Q s
,
. %r
c C s0 t dD ]}| d@ r| d? dA } q| dL } q| S )Nr r l q[ )range)crcr7 r r r _gen_crc/ s
r c sl dddt d u rttttda t fdd fdd| D ]}| q&fd d
}|S ) NixV4igE#ixV4 c s |d? || A d@ A S )z(Compute the CRC32 primitive on one byte.r r )Zchr )crctabler r crc32I s z _ZipDecrypter..crc32c s< | d@ d@ d d d@ d? d S )Nr r ir r )c)r key0key1key2r r update_keysM s
z"_ZipDecrypter..update_keysc sN t }|j}| D ]} dB }|||dA d? d@ N }| || qt|S )zDecrypt a bytes object.r r r r ) bytearrayr1 bytes)rX rC r1 r k)r r r r decrypterW s
z _ZipDecrypter..decrypter) _crctabler` mapr r )pwdpr r )r r r r r r r
_ZipDecrypter? s
r c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
LZMACompressorc C s
d | _ d S r )_compr r r r r g s
zLZMACompressor.__init__c C sF t dt ji}t jt jt t j|gd| _tdddt || S )Nidfiltersz r\ r nrX r r r r> s $z_SharedFile.readc C s( | j d ur| j }d | _ | | d S d S r )r r )r fileobjr r r r s
z_SharedFile.closeNr r )r r r r r\ rE 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 )r< rV r r< r r r r
s
z_Tellable.__init__c C s | j |}| j|7 _|S r )r< writerV )r rX r r r r r s z_Tellable.writec C r r )rV r r r r r\ r z_Tellable.tellc C | j d S r )r< r r r r r r z_Tellable.flushc C r r )r< r r r r r r r z_Tellable.closeN)r r r r r r\ r r r r r r r s r c s e Zd ZdZdZdZdZ d(ddZd d
Zd d Z d)ddZ
d*ddZ dd Z d)ddZ
dd Zdd Zdd Zd d 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().
i @i i NFc C s& || _ || _|| _|j| _|j| _|j| _t | j| _
d| _ d| _ d| _
d | _|| _|j| _t|dr=|j| _td| _nd | _d| _z| r\| | _|j| _|j| _ | j| _ d| _W n t yf Y nw d | _ |r|j d@ ry|j!d? d@ }n|jd? d@ }| " }||krt#d |j$ d S d S )
NFr- r rq Tr r r zBad password for file %r)%_fileobj_pwd_close_fileobjrg _compress_typerr _compress_leftrs _leftr
_decompressor_eof _readbuffer_offsetnewlinesmoderB namer@ rq
_expected_crcr _running_crc _seekabler r\ _orig_compress_start_orig_compress_size_orig_file_size_orig_start_crcAttributeError
_decrypterrl rt _init_decrypterr re )r r r& zipinfor Z
close_fileobjZ
check_bytehr r r r + sN
zZipExtFile.__init__c C s4 t | j| _| jd}| jd8 _| |d S )Nr r& )r r r0 r r> r )r r r r r r1 ^ s zZipExtFile._init_decrypterc C sv d| j j| j jf g}| js,|d| j| jf | jtkr+|dt
| j| j n|d |d d |S )N<%s.%sz name=%r mode=%rr [closed]r r ) r r r closedr1 r' r& r r r r r2 r rC r r r r i s
zZipExtFile.__repr__r c C sL |dk r | j d| jd }|dkr | j | j| }|| _|S tj| |S )zrRead and return a line from the stream.
If limit is specified, at most limit bytes will be read.
r
r )r# rz r$ ioBufferedIOBasereadline)r limitr6 liner r r r; w s zZipExtFile.readliner c C sr |t | j| j kr.| |}t || jkr%|| j| jd | _d| _n | jt |8 _| j| j| jd S )z6Returns buffered bytes without advancing the position.Nr i )r0 r# r$ r> )r r chunkr r r peek s
zZipExtFile.peekc C s | j rtddS )N I/O operation on closed file.T)r6 r r r r r readable s zZipExtFile.readablec C s" | j rtd|du s|dk r-| j| jd }d| _d| _| js+|| | j7 }| jr |S || j }|t| jk rF| j| j| }|| _|S |t| j }| j| jd }d| _d| _|dkr| js| |}|t|k r~|| _|| _||d| 7 } |S ||7 }|t|8 }|dkr| jrb|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.
zread from closed file.Nr r- )r6 r r# r$ r" _read1MAX_Nr0 )r r bufendrX r r r r> s>
zZipExtFile.readc C sD | j d u rd S t|| j| _| jr | j| j kr td| j d S 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 |du s|dk r-| j | jd }d| _ d| _| js+| | j}|r(||7 } |S | jr|S || j }|t| j k rF| j | j| }|| _|S |t| j }| j | jd }d| _ d| _|dkr| js| |}|t|k r~|| _ || _||d| 7 } |S |r||7 } |S | jrb|S )z7Read up to n bytes with at most one read() system call.Nr r- )r# r$ r" rB rC r0 )r r rD rX rE r r r read1 sH
zZipExtFile.read1c C s | j s|dkr dS | jtkr$| jj}|t|kr#|| |t| 7 }n| |}| jtkr5| jdk| _ n<| jtkrat || j
}| j ||}| jj pT| jdkoT| jj | _ | j r`|| j
7 }n| j |}| jj po| jdk| _ |d | j }| jt|8 _| jdkrd| _ | | |S )Nr r- T)r" r r r! Zunconsumed_tailr0 _read2r r ra
MIN_READ_SIZEr r r r rG r r r r rB s8
zZipExtFile._read1c C sd | j dkrdS t|| j}t|| j }| j|}| j t|8 _ |s&t| jd ur0| |}|S )Nr r- ) r ra rJ minr r> r0 EOFErrorr0 r r r r rI s
zZipExtFile._read2c s> z| j r| j W t d S W t d S t w r )r r r superr r r r r s zZipExtFile.closec C s | j rtd| jS Nr@ )r6 r r* r r r r r & s zZipExtFile.seekabler c C sJ | j rtd| jstd| }|dkr|}n|dkr#|| }n|dkr-| j| }ntd|| jkr9| j}|dk r?d}|| }|| j }|dkrY|t| j k rY|| _d}n1|dk r| j
| j | j
| _| j| _| j| _d| _ d| _t| j| _d| _|}| jd ur| |dkrt| j|}| | ||8 }|dks| S ) Nzseek on closed file.!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)r6 r r* r9 UnsupportedOperationr\ r- r$ r0 r# r rE r+ r. r) r, r r r r r! r" r0 r1 rK
MAX_SEEK_READr> )r rV r
Zcurr_posZnew_posZ read_offsetZ buff_offsetZread_lenr r r rE + sN
zZipExtFile.seekc C s> | j rtd| jstd| j| j t| j | j }|S )Nztell on closed file.rP )
r6 r r* r9 rQ r- r r0 r# r$ )r Zfileposr r r r\ ] s
zZipExtFile.tell)NFr r r )r r r r rC rJ rR r r1 r r; r? rA r> rG rH rB rI r r rE r\
__classcell__r r rN r r s,
3
#
%$
2r c s@ e Zd Zdd Zedd Zdd Zdd Z fd d
Z Z S )
_ZipWriteFilec C s8 || _ || _|| _t|j|j| _d| _d| _d| _ d S r )
_zinfo_zip64_zipfiler rg rh _compressor
_file_size_compress_size_crc)r zfr r r r r r g s
z_ZipWriteFile.__init__c C s | j jS r )rX r< r r r r r q s z_ZipWriteFile._fileobjc C s dS )NTr r r r r writableu z_ZipWriteFile.writablec C s | j rtdt|ttfrt|}nt|}|j}| j|7 _t || j
| _
| j r:| j |}| j
t|7 _
| j| |S rO )r6 r r r r r0
memoryviewnbytesrZ r r\ rY r r[ r r )r rX ra r r r r x s
z_ZipWriteFile.writec sf | j rd S zt | 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 < W d| j_ d S d| j_ w )Nr zd!d"d#d$Zd?d%d&Zd@d'd(ZdAd)d*Zed+d, Zd-d. Z d/d0 Z dAd1d2Z d@d3d4Z d5d6 Z d7d8 Z!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.
NrTr c
C s@ |dvrt dt| || _d| _d| _i | _g | _|| _|| _|| _ d| _
d| _ || _ t
|tjr8t|}t
|trpd| _|| _ddd d
dd d d
}|| } z t||| _W n tym ||v rl|| }Y qQ w n
d| _|| _t|dd| _d| _t | _ d| _ d| _ z|dkr| W dS |dv rd| _z| j | _!W n t"tfy t#| j| _d| _!d| _ Y W dS w z | j$| j! W W dS t"tfy d| _ Y W dS w |dkr
z| | j$| j! W W dS t%y | j$dd d| _| j | _!Y W dS w t d | j} d| _| &| )z]Open the ZIP file with mode read 'r', write 'w', exclusive create 'x',
or append 'a'.)rg wxaz+ZipFile requires mode 'r', 'w', 'x', or 'a'Fr Nr- r? w+bx+br+bwbZxb)rg rh ri rj rm rk rl Tr r' rg )rh ri rj r z"Mode must be 'r', 'w', 'x', or 'a')'r r _allowZip64
_didModifydebugre rd r r r& r _comment_strict_timestampsr r| r r str _filePassedrB r9 rA r< r: getattr _fileRefCnt threadingRLockr r* r _RealGetContentsr\ rc r/ r rE r _fpclose)
r r r& r
allowZip64r r ZmodeDictr r< r r r r s
zZipFile.__init__c C s | S r r r r r r __enter__/ r_ zZipFile.__enter__c C s | d S r r )r typevalue tracebackr r r __exit__2 zZipFile.__exit__c C s d| j j| j jf g}| jd ur2| jr |d| j n
| jd ur)|d| j |d| j n|d |d d|S )Nr4 z file=%rz filename=%rz mode=%rr5 r r ) r r r r< ru r1 rB r& r2 r7 r r r r 5 s
zZipFile.__repr__c C s | j }zt|}W n ty tdw |std| jdkr$t| |t }|t }|t | _ |t
| | }| jdkrI|| }td||| || | _ | j dk rWtd| | j d |
|}t|}d}||k rS|
t} t| tkr~tdtt| } | t tkrtd| jdkrt| |
| t }
| d } | d
@ r|
d }
n|
d }
t|
} |
| t | _|
| t | _ | t | _ | dd
\ | _ | _!| _"| _#| _$| _%}
}| _&| _'| _(| j"t)krt*d| j"d | dd \| _+| _,| _-|
| _.|d? d |d ? d@ |d@ |
d? |
d ? d@ |
d@ d f| _/| 0 | j | | _ | j12| | | j3| j4< |t | t | t | t }| jdkrOtd| ||k so| j }t5| j1dd ddD ] }||_6|j }q`d S ) z/Read in the table of contents for the ZIP file.zFile is not a zip filer r zgiven, inferred, offsetr z Bad offset for central directoryzTruncated central directoryz&Bad magic number for central directoryr! r r cp437r zzip file version %.1fr% r( r+ r$ rx r r& r totalc S r r )rp )r r r r z*ZipFile._RealGetContents..T)keyreverseN)7r< r9 r: r rq printrR rS _ECD_COMMENTrr rT rc rE r> r9 BytesIOsizeCentralDirr0 rF r/ structCentralDir
_CD_SIGNATUREstringCentralDir_CD_FILENAME_LENGTHdecoder
_CD_EXTRA_FIELD_LENGTHr3 _CD_COMMENT_LENGTHrd _CD_LOCAL_HEADER_OFFSETrp rY ri rj rk rl rg rq rr rs MAX_EXTRACT_VERSIONr rm rn ro rt rf r rd r1 re rB sortedru )r r< rW Zsize_cdZ offset_cdconcatZinferredrX r centdirrB flagsri td
end_offsetr r r r rz C s
/
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 ).0rX r r r
s z$ZipFile.namelist..rd r r r r namelist s zZipFile.namelistc C r )zJReturn a list of class ZipInfo instances for files in the
archive.r r r r r infolist zZipFile.infolistc C sL t dd |d | jD ]}d|jdd }t d|j||jf |d q dS )z+Print a table of contents for the zip file.z%-46s %19s %12s)z File Namez Modified ZSize)r z%d-%02d-%02d %02d:%02d:%02dNr" z
%-46s %s %12d)r rd rf rB rs )r r r dater r r printdir s
zZipFile.printdirc C sx d}| j D ]4}z$| |jd}||r ||sW d n1 s%w Y W q ty9 |j Y S w dS )z%Read all the files and check the CRC.i rg N)rd rA rB r> r )r Z
chunk_sizer fr r r testzip s
zZipFile.testzipc C s$ | j |}|du rtd| |S )z,Return the instance of ZipInfo given 'name'.Nz(There is no item named %r in the archive)re r KeyError)r r' infor r r getinfo s zZipFile.getinfoc C s8 |rt |tstdt|j |r|| _dS d| _dS )z)Set default password for encrypted files.pwd: expected bytes, got %sN)r r TypeErrorr r r )r r r r r setpassword s
zZipFile.setpasswordc C r )z.The comment text associated with the ZIP file.)rr r r r r rd r zZipFile.commentc C s^ t |tstdt|j t|tkr'dd 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 r r r0 ZIP_MAX_COMMENTwarningswarnrr rp )r rd r r r r rd s
c C s: | |d| }| W d S 1 sw Y dS )zReturn file bytes for name.rg NrA r> )r r' r r< r r r r> s $z ZipFile.readF force_zip64c s* |dvrt d|rt|tstdt|j |r"|dkr"t d js)t dt|tr1|}n|dkrBt|} j|_ j
|_ n |}|dkrR j
||dS jrYt d jd 7 _t j|j j j fd
d }z|t}t|tkrtd tt|}|t t krtd
||t }|t r||t |j d@ rt d|j d@ rt d|t! d@ r|"d} n|"d} | |j#krtd|j#|f |j$dur|% |j& |j$krtd|j#d|j d @ }
|
r|s j'}|st(d| nd}t)||||dW S |* )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.
> rh rg z open() requires mode "r" or "w"r rh z'pwd is only supported for reading filesz2Attempt to use ZIP archive that was already closedr r r c s j S r )r r r r r r r z ZipFile.open..zTruncated file headerz Bad magic number for file header z$compressed patched data (flag bit 5)@ z strong encryption (flag bit 6)r r r z/File name in directory %r and header %r differ.NzOverlapped entries: z (possible zip bomb)z6File %r is encrypted, password required for extractionT)+r r r r r r r< r
r rg r rh r _open_to_writer rw r rp r{ r r> sizeFileHeaderr0 r rF r/ r
_FH_SIGNATUREr _FH_FILENAME_LENGTH_FH_EXTRA_FIELD_LENGTHrl r _FH_GENERAL_PURPOSE_FLAG_BITSr re ru r\ rr r r r r ) r r' r& r r r Zzef_fileZfheaderfnameZ fname_strZ is_encryptedr r r rA s
z ZipFile.openc C s |r | j s td| jrtdd|_d|_d|_|jtkr%| jdO _| js/| jdO _|j s5d|_ | j o@|p@|j
d t 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.r r r g?T)ro r r rr rq rl rg r r* ro rs r r< rE rc r\ rp _writecheckrp r r rU )r r r r r r r r D s4
zZipFile._open_to_writec C s* |du r 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)r| getcwdr _extract_member)r memberr r r r r extractm s
zZipFile.extractc C sH |du r| }|du rt }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)r r| r r r )r r membersr r2 r r r
extractallz s
zZipFile.extractallc C s^ | j }|sd}t|dt| }|| _ ||}dd ||D }|dd |D }|S )z;Replace bad characters and remove trailing dots from parts.z:<>|"?*_c s s | ]}| d V qdS ).N)rstripr ri r r r z1ZipFile._sanitize_windows_name..c s s | ]}|r|V qd S r r r r r r r s )!_windows_illegal_name_trans_tablert maketransr0 translatesplitr2 )r r pathseptableZillegalr r r _sanitize_windows_name s
z ZipFile._sanitize_windows_namec s t |ts
| |}|jdtjj}tjjr!|tjjtjj}tj |d }dtjj
tjj f tjj fdd|
tjjD }tjjdkrS| |tjj}tj ||}tj|}tj|}|rstj|sst| | rtj|st| |S | j||d,}t|d}t|| W d n1 sw Y W d |S W d |S 1 sw Y |S )
zbExtract the ZipInfo object 'member' to a physical
file on the path targetpath.
ry r r c 3 s | ] }| vr|V qd S r r r Zinvalid_path_partsr r r s z*ZipFile._extract_member..\r rn N)r r
r rB r~ r| r r} r r curdirpardirr2 r r r dirnameexistsmakedirsr r mkdirrA shutil copyfileobj)r r Z
targetpathr r Z upperdirssourcetargetr r r r s<
&
( zZipFile._extract_memberc C s |j | jv rddl}|jd|j dd | jdvr td| js$tdt|j | j sOd}t
| j t kr8d }n|j
tkr@d
}n|jtkrGd }|rQt|d dS dS )
z6Check for errors before writing a file to the archive.r NzDuplicate name: %rr r rh ri rj z&write() requires mode 'w', 'x', or 'a'z4Attempt to write ZIP archive that was already closed Files countZFilesizez Zipfile size would require ZIP64 extensions)rB re r r r& r r< r rg ro r0 rd ZIP_FILECOUNT_LIMITrs r rp r
)r r r requires_zip64r r r r s0
zZipFile._writecheckc C s | j std| 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 | _W d dS 1 sw Y dS t |d .}| |d
}t ||d W d n1 sw Y W d dS W d dS 1 sw Y dS ) 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 existsr r Nr TFr? rh i ) r< r r r
r rs r rr rq rg r rh r r r* rE rc r\ rp r rl r rp rd r1 re rB r r rA r r )r rB r rg r r srcdestr r r r sJ
"Pz
ZipFile.writec C s2 t |tr
|d}t |ts |d d> B |d B }|d d> |d d> B |d d B }g }|jtks8|jtkrI||j ||j d
}d
}n|j}|j}|jtkr]||j d
}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 |j q| j }t | j }|| j! }| j!}d }|t"krd}n|tkrd}n|tkrd}|rX| j#s#t$|d t
t%t&t'd dddd|||| }| 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 0 d S )Nr rx r$ r r! r r r& r r rS r, Qr r zCentral directory offsetzCentral directory sizer r r r rj )1rd rf rs r rr r1 rp r3 r8 rF r r0 r rg r ra r r r rj rY r r r ri rk rq rd rn ro r< r r\ rc r ro r
rL rK rI rG rH rK r_ r^ rr r& truncater )r r r r r r3 rs rr rp Z
extra_datar rj rY rB rl r Zpos2Z centDirCountZ centDirSizeZ
centDirOffsetr Z zip64endrecZ zip64locrecrW r r r r O s
$$
zZipFile._write_end_recordc C s< | j dksJ | j d8 _ | j s| js | d S d S d S )Nr r )rw ru r r r r r r{ s
zZipFile._fpcloser )rg N)F)NN)NNN)$r r r r r< r r r r} r r rz r r r r r r rf rd setterr> rA r r r r r r r r r r r r r{ r r r r r sR ZS
f
)
*
2
) ^r c @ s4 e Zd ZdZdeddfddZd
d d
Zd d ZdS )r zDClass to create ZIP archives with Python library files and packages.rg Tr c C s t j| ||||d || _d S )N)r& r r| )r r _optimize)r r r& r r| optimizer r r r s
zPyZipFile.__init__r Nc C s t |}|r"||s"| jr t j|rdnd}td||f dS t j|\}}t j|r-t j|d}t j|r|rGd||f }n|}| jrStd|d| | |d d
|\}} | jrgtd | |
|| t t |}
|
d |
D ]Y} t j|| } t j| \}
}t j| rt jt j| dr| j| ||d q{|d
kr|r|| s| jrtd| q{| | d d
|\}} | jrtd | |
|| q{dS | jrtd| t t |D ]D} t j|| } t j| \}
}|d
kr*|r|| s| jrtd| q| | d d
|\}} | jr$td | |
|| qdS |d
d d
kr:td| |d d
|\}} | jrOtd| |
|| 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__.py%s/%szAdding package inasr ZAdding)
filterfunc.pyz file %r skipped by filterfunczAdding files from directoryz.Files added with writepy() must end with ".py"z Adding file)r| r rq r r r r r2 isfile
_get_codenamer r listdirremovesplitextwritepyr )r pathnamebasenamer Zlabeldirr' Zinitnamer r ZdirlistrB r rootZextr r r r s
zPyZipFile.writepyc
s d fdd }|d }|d }t jj|dd}t jj|dd}t jj|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| }
} nC jd
kr|}
|} n |} jdkr|}
n jd kr|}
n
d j} t
| tj|
rt|
jt|jks|| jd s| }
} tj| 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
sb dd l } jr td| z |j| d|d W dS |jy0 } z t|j W Y d }~dS d }~ww )Nr Z CompilingT)doraiser F)
py_compilerq r compilePyCompileErrormsg)r r r errr r r _compile s
z)PyZipFile._get_codename.._compiler z.pycr ) optimizationr r r z"invalid value for 'optimize': {!r})r r Nr ) importlibutilcache_from_sourcer r| r r r r r r r formatr r )
r r r r Zfile_pyZfile_pycZ pycache_opt0Z pycache_opt1Z pycache_opt2r r r Z archivenamer r r r s`
zPyZipFile._get_codename)r N)r r r r r r r r r r r r r s
Rr c C s t t| 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(''))
[]
r N) itertoolsislice _ancestry)r r r r _parentsb s r c c sB | tj} | tjr | V t| \} }| tjs
dS dS )a
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(''))
[]
Multiple separators are treated like a single.
>>> list(_ancestry('//b//d///f//'))
['//b//d///f', '//b//d', '//b']
N)r posixpathr} r )r tailr r r r u s r c C s t t|j| S )zZ
Return items in minuend not in subtrahend, retaining order
with O(1) lookup.
)r filterfalseset __contains__)ZminuendZ
subtrahendr r r _difference s r c sT e Zd ZdZedd Z fddZdd Zdd Z fd
d Z e
d d
Z Z S ) CompleteDirszk
A ZipFile subclass that ensures that implied directories
are always included in the namelist.
c C s. t jtt| }dd |D }tt|| S )Nc s s | ]}|t j V qd S r )r r} )r r r r r r r z-CompleteDirs._implied_dirs..)r chain
from_iterabler r _deduper )namesparentsZas_dirsr r r
_implied_dirs s zCompleteDirs._implied_dirsc s t t| }|t| | S r )rM r r r` r )r r rN r r r s zCompleteDirs.namelistc C s t | S r )r r r r r r _name_set r zCompleteDirs._name_setc C s, | }|d }||vo||v }|r|S |S )zx
If the name represents a directory, return that name
as a directory (with the trailing slash).
ry )r )r r' r r Z dir_matchr r r resolve_dir s zCompleteDirs.resolve_dirc sD zt |W S ty! |dr|| vr t|d Y S w )z6
Supplement getinfo for implied dirs.
ry r )rM r r endswithr r
)r r' rN r r r s zCompleteDirs.getinfoc C s8 t |tr|S t |ts| |S d|jvrt} | |_|S )zl
Given a source (filename or zipfile), return an
appropriate CompleteDirs subclass.
rg )r r r r& r )r r r r r make s
zCompleteDirs.make)
r r r r staticmethodr r r r r r r rT r r rN r r s
r c s, e Zd ZdZ fddZ fddZ ZS )
FastLookupzV
ZipFile subclass to ensure implicit
dirs exist and are resolved rapidly.
c F t t | jW d S 1 sw Y tt| | _| jS r )
contextlibsuppressr/ Z_FastLookup__namesrM r" r r rN r r r
zFastLookup.namelistc r# r )r$ r% r/ Z_FastLookup__lookuprM r" r r rN r r r r& zFastLookup._name_set)r r r r r r rT r r rN r r" s r" c O s t | d||fS )Nr )r9
text_encoding)encodingargskwargsr r r _extract_text_encoding s r+ c @ s e Zd ZdZdZd'ddZd(ddd d
Zed d Zed
d Z dd Z
dd Z dd Z dd Z
dd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ ZeZed%d& ZdS ))r u,
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})r c C s t || _|| _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" r r at)r r r, r r r r @ s
z
Path.__init__rg Nr c O s | rt| |d }| s|dkrt| | jj| j||d}d|v r0|s*|r.td|S t|i |\}}}t j
||g|R i |S )z
Open this entry as text or binary following the semantics
of ``pathlib.Path.open()`` by passing arguments through
to io.TextIOWrapper().
r rg r bz*encoding args invalid for binary operation) r IsADirectoryErrorr FileNotFoundErrorr rA r, r r+ r9
TextIOWrapper)r r& r r) r* Zzip_modestreamr( r r r rA M s z Path.openc C s t | jjp
| jjS r )pathlibZ
PurePosixPathr, r' rB r r r r r' a z Path.namec C s t | jj| jS r )r2 r r rB joinpathr, r r r r rB e r3 z
Path.filenamec O sZ t |i |\}}}| jd|g|R i | }| W d S 1 s&w Y d S )Nrg )r+ rA r> )r r) r* r( strmr r r read_texti s $zPath.read_textc C s6 | d }| W d S 1 sw Y d S )Nr? r )r r5 r r r
read_bytesn s $zPath.read_bytesc C s t |jd| jdkS Nry )r r r, r )r r r r r _is_childr s zPath._is_childc C s | | j|S r )r r )r r, r r r _nextu r z
Path._nextc C s | j p | j dS r8 )r, r r r r r r x s z Path.is_dirc C s | o| S r )r r r r r r is_file{ z Path.is_filec C s | j | j v S r )r, r r r r r r r ~ s z Path.existsc C s. | stdt| j| j }t| j|S )NzCan't listdir a file)r r r r: r r filterr9 )r Zsubsr r r iterdir s z Path.iterdirc C s t | jj| jS r )r r2 r rB r, r r r r __str__ r< z Path.__str__c C s | j j| dS )Nr ) _Path__reprr r r r r r r z
Path.__repr__c G s& t j| jg|R }| | j|S r )r r2 r, r: r r )r othernextr r r r4 s z
Path.joinpathc C s6 | j s| jjS t| j d}|r|d7 }| |S r8 )r, rB parentr r r r: )r Z parent_atr r r rC s
z Path.parent)r )rg )r r r r r@ r rA rf r' rB r6 r7 r9 r: r r; r r> r? r r4 __truediv__rC r r r r r s. M
r c
s dd l }d}|j|d}|jdd}|jdddd d
|jd d d
ddd |jdddddd |jddddd
|| } | jd urm| j}t|d }| }W d n1 sYw Y |rgtd | td d S | j
d ur| j
}t|d
}| W d d S 1 sw Y d S | j d ur| j \}}t|d}|
| W d d S 1 sw Y d S | jd ur| jd} | j}
fd d t| d 1}|
D ]%} tj| } | stjtj| } | d tjtjfv rd } || | qW d d S 1 sw Y d S d S ) Nr z3A simple command-line interface for zipfile module.) descriptionT)Zrequiredz-lz--list zShow listing of a zipfile)metavarhelpz-ez --extractr )rF z z Extract zipfile into target dir)nargsrG rH z-cz--create+)zzzCreate zipfile from sourcesz-tz--testzTest if a zipfile is validrg z.The following enclosed file is corrupted: {!r}z Done testingc sv t j|r| ||t d S t j|r7|r | || tt |D ]} | t j||t j|| q$d S d S r ) r| r r r r r r r r2 )r] r zippathZnmaddToZipr r rM s zmain..addToZiprh r )argparseArgumentParserZ add_mutually_exclusive_group add_argument
parse_argstestr r r r r` r r r Zcreatepopr| r r r r r )
r) rN rE parsergroupr r] Zbadfiler Zzip_namefilesr rK r rL r main sh
"
" $rW __main__r )r Zbinasciiimportlib.utilr r9 r r| r rer r rF r rx r r$ r2 r r ImportErrorr r __all__ Exceptionr r
r r r r r r r r r r r r r r r_ r^ r r] rM rN rO rP rQ rR rS rc r rT r r r r 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 r Z_CD_DISK_NUMBER_STARTZ _CD_INTERNAL_FILE_ATTRIBUTESZ _CD_EXTERNAL_FILE_ATTRIBUTESr r r r r Z_FH_EXTRACT_VERSIONZ_FH_EXTRACT_SYSTEMr Z_FH_COMPRESSION_METHODZ_FH_LAST_MOD_TIMEZ_FH_LAST_MOD_DATEZ_FH_CRCZ_FH_COMPRESSED_SIZEZ_FH_UNCOMPRESSED_SIZEr r rG rH rD rL rK rI 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_CENTDIRrb ZStructr. r8 r= r rZ r9 objectr
r r r r r r r r r r r r: r rU r r r r dictfromkeysr r r r" r+ r rW r r r r r s
<< ^&
( KU ~ /<
.
=