a
XC?h�X � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl Z d dl
Z
zd dlZejZW n ey� dZe jZY n0 z d dlZW n ey� dZY n0 z d dlZW n ey� dZY n0 g d�ZG dd� de�ZG dd� de�Ze ZZdZdZdZd Z d Z d
Z d Z d Z d
Z!dZ"dZ#dZ$dZ%dZ&e �'e%�Z(d Z)dZ*dZ+dZ,dZ-dZ.dZ/dZ0d Z1dZ2dZ3dZ4e �'e3�Z5d Z6dZ7dZ8dZ9dZ:dZ;dZdZ?d Z@d ZAd
ZBd ZCd ZDd ZEd ZFd!ZGd"ZHd#ZId$ZJe �'eI�ZKd ZLdZMdZNdZOdZPdZQdZRdZSd ZTdZUd ZVd ZWd%ZXd&ZYe �'eX�ZZd'Z[d(Z\e �'e[�Z]d Z^dZ_dZ`dZadZbdZcdZddZed ZfdZgd)Zhe �id*�Zjd+d,� Zkd-d.� Zld/d0� Zmd1d2� Znd3d4� ZoG d5d6� d6ep�Zqdard7d8� Zsd9d:� ZtG d;d<� d<�ZuG d=d>� d>�Zvd?d@dAdAdAdAdBdCdDdEdBdFdGdHdIdJdKdL�ZwdMdN� ZxdndOdP�ZydQdR� ZzG dSdT� dT�Z{G dUdV� dV�Z|G dWdX� dXej}�Z~G dYdZ� dZej}�ZG d[d\� d\�Z�G d]d^� d^e��Z�d_d`� Z�dadb� Z�e�j�Z�dcdd� Z�G dedf� dfe��Z�G dgdh� dhe��Z�G didj� dj�Z�dodkdl�Z�e�dmk�r�e�� dS )p� N)
�
BadZipFile�
BadZipfile�error�
ZIP_STORED� ZIP_DEFLATED� ZIP_BZIP2�ZIP_LZMA�
is_zipfile�ZipInfo�ZipFile� PyZipFile� LargeZipFile�Pathc @ s e Zd ZdS )r N��__name__�
__module__� __qualname__� r r �,/opt/alt/python39/lib64/python3.9/zipfile.pyr * s r c @ s e Zd ZdS )r
Nr r r r r r
. s r
i������ � � � � �- �. �? s <4s4H2LHs PK� � � � � � � � z<4s4B4HL2L5H2Ls PK�
� �
� � � � z <4s2B4HL2L2Hs PKz<4sLQLs PKz
<4sQ2H2L4Qs PKiPK� r0 �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_versionZ read_versionZdisk_numZdisk_dirZdircountZ dircount2ZdirsizeZ diroffsetr r r �
_EndRecData64� s:
�rX c C sf | � dd� | �� }z| � t d� W n ty: Y d S 0 | �� }t|�tkr�|dd� tkr�|dd � dkr�t�t |�}t
|�}|� d� |� |t � t | t |�S t
|d t d�}| � |d� | �� }|�t�}|dk�rb|||t � }t|�tk�rd S t
t�t |��}|t }||t |t | � }|� |� |� || � t | || | |�S d S )Nr r r �����s r- i )rD �tell�sizeEndCentDirr: r> r0 �stringEndArchiverF r/ �structEndArchive�listr1 rX �max�rfind�_ECD_COMMENT_SIZE) rS ZfilesizerV rU ZmaxCommentStartr5 ZrecDataZ commentSize�commentr r r r9 sD ��
�r9 c @ sX e Zd ZdZddd�Zdd� Zdd d
�Zd d � Zd
d� Ze ddd�dd��Z
dd� Z dS )r
)�
orig_filenamerB � date_time�
compress_type�_compresslevelrb r3 �
create_systemrW �extract_version�reserved� flag_bits�volume�
internal_attr�
external_attr�
header_offset�CRC�
compress_size� file_size� _raw_time� _end_offset�NoName�� r r r r r c C s� || _ |�td��}|dkr(|d|� }tjdkrJtj|v rJ|�tjd�}|| _|| _|d dk rjtd��t | _
d | _ d| _ d| _
tjdkr�d| _nd| _t| _t| _d| _d| _d| _d| _d| _d| _d| _d | _d S )Nr �/rv z+ZIP does not support timestamps before 1980r- Zwin32r ) rc �find�chr�os�sep�replacerB rd �
ValueErrorr re rf rb r3 �sys�platformrg �DEFAULT_VERSIONrW rh ri rj rk rl rm rp rq rs )�selfrB rd Z null_byter r r �__init__X s6
zZipInfo.__init__c C s� d| j j| jf g}| 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) r z filemode=%rz external_attr=%#xz
file_size=%rz compress_size=%r�>� )� __class__r rB re r r1 �compressor_names�getrm �stat�filemode�is_dirrq rp r2 )r� rC �hi�lo�isdirr r r �__repr__� s0
��
�
�
zZipInfo.__repr__Nc C s| | j }|d d d> |d d> B |d B }|d d> |d d> B |d d B }| jd
@ rfd } }}n| j}| j}| j}| j}d} |d u r�|tkp�|tk}|r�d }
|t�|
dt� |
�d ||� }|tks�|tkr�|s�t
d ��d
}d
}t } | j t
k�rtt| �} n| j tk�rtt| �} t| | j�| _t| | j�| _| �� \} } t�tt| j| j| | j |||||t| �t|��
}
|
| | S )Nr rv r$ r r! r r r& r r z |_|r�d|_| jd
O _n|j|_|S ) Nr r" rv ru �; )r� r � � �; r� r rw r r) )�
isinstancerz �PathLike�fspathr� �S_ISDIR�st_mode�time� localtime�st_mtime�path�normpath�
splitdriver{ �altseprm rq �st_size) �clsrB �arcnamer� �str� �mtimerd �zinfor r r � from_file� s0
zZipInfo.from_filec C s | j d dkS )N�����rw �rB r� r r r r� s zZipInfo.is_dir)rt ru )N)N) r r r � __slots__r� r� r� r� r� � classmethodr� r� r r r r r
= s
,
. %r
c C s0 t d�D ]"}| d@ r"| d? dA } q| dL } q| S )Nr r l q[ )�range)�crcr7 r r r �_gen_crc s
r� c sl d�d�d�t d u r&ttttd���a t ��fdd�� � ���fdd��| D ] }�|� qL��fd d
�}|S ) NixV4i�gE#i�xV4� c s |d? � || A d@ A S )Nr �� r )Zchr� )�crctabler r �crc325 s z _ZipDecrypter..crc32c s< � | �����d@ d@ ��d d d@ �� �d? ���d S )Nr� r� i�r � r )�c)r� �key0�key1�key2r r � update_keys9 s
z"_ZipDecrypter..update_keysc sN t � }|j}| D ]4}� dB }|||dA d? d@ N }�|� ||� qt|�S )Nr r r r� )� bytearrayr1 �bytes)rV rC r1 r� �k)r� r� r r � decrypterC s
z _ZipDecrypter..decrypter)� _crctabler^ �mapr� r� )�pwd�pr� 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 �N)�_compr� r r r r� S s zLZMACompressor.__init__c C sF t �dt ji�}t jt jt �t j|�gd�| _t�dddt |��| S )N�id��filtersz rZ �r� �nrV r r r r> � s z_SharedFile.readc C s$ | j d ur | j }d | _ | �|� d S r� )r� r )r� �fileobjr r r r � s
z_SharedFile.closeN)r )r� )r r r r� rZ rD 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< rT �r� r< r r r r� � s z_Tellable.__init__c C s | j �|�}| j|7 _|S r� )r< �writerT )r� rV r r r r r � s z_Tellable.writec C s | j S r� )rT r� r r r rZ � s z_Tellable.tellc C s | j �� d S r� )r< r� r� r r r r� s z_Tellable.flushc C s | j �� d S r� )r< r r� r r r r s z_Tellable.closeN)r r r r� r rZ r� r r r r r r � s
r c s� e Zd ZdZdZdZd'dd�Zdd � Zd
d � Zd(d
d�Z d)dd�Z
dd� Z d*dd�Z dd� Z
dd� Zdd� Zd d � Z� fd d �Zd d!� Zd+d#d$�Zd%d&� Z� ZS ),�
ZipExtFilei @i i NFc C s& || _ || _|| _|j| _|j| _|j| _t | j�| _
d| _ d| _ d| _
d | _|| _|j| _t|d�rz|j| _td�| _nd | _d| _z4|�� r�|�� | _|j| _|j| _ | j| _ d| _W n t y� Y n0 d | _ |�r"|j d@ r�|j!d? d@ }n|jd? d@ }| �"� }||k�r"t#d |j$ ��d S )
NFr- r ro Tr r� r� zBad password for file %r)%�_fileobj�_pwd�_close_fileobjre �_compress_typerp �_compress_leftrq �_leftr� �
_decompressor�_eof� _readbuffer�_offset�newlines�moderB �namer@ ro �
_expected_crcr� � _running_crc� _seekabler rZ �_orig_compress_start�_orig_compress_size�_orig_file_size�_orig_start_crc�AttributeError�
_decrypterrj rr �_init_decrypterr� rc )r� r
r �zipinfor� Z
close_fileobjZ
check_byte�hr r r r� sF
zZipExtFile.__init__c C s4 t | j�| _| j�d�}| jd8 _| �|�d S )Nr r& )r� r r( r r> r )r� r� r r r r) J s zZipExtFile._init_decrypterc C sv d| j j| j jf g}| jsX|�d| j| jf � | jtkrb|�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� U s �
��
zZipExtFile.__repr__r� c C sL |dk r>| j �d| j�d }|dkr>| j | j|� }|| _|S tj�| |�S )Nr �
r )r rx r �io�BufferedIOBase�readline)r� �limitr6 �liner r r r3 c s zZipExtFile.readliner c C sr |t | j�| j kr\| �|�}t |�| jkrJ|| j| jd � | _d| _n| jt |�8 _| j| j| jd � S )Nr i )r0 r r r> )r� r �chunkr r r �peeks s
zZipExtFile.peekc C s | j rtd��dS )N� I/O operation on closed file.T)r. r} r� r r r �readable� s zZipExtFile.readablec C s | j rtd��|d u s |dk rV| j| jd � }d| _d| _| jsR|| �| j�7 }q:|S || j }|t| j�k r�| j| j|� }|| _|S |t| j� }| j| jd � }d| _d| _|dk�r| j�s| �|�}|t|�k r�|| _|| _||d |� 7 }�q||7 }|t|�8 }q�|S )Nzread from closed file.r r- )r. r} r r r �_read1�MAX_Nr0 )r� r �buf�endrV r r r r> � s8
zZipExtFile.readc C s@ | j d u rd S t|| j�| _| jr<| j| j kr
zZipExtFile.read1c C s" | j s|dkrdS | jtkrH| jj}|t|�krR|| �|t|� �7 }n
| �|�}| jtkrj| jdk| _ nx| jtkr�t || j
�}| j� ||�}| jj p�| jdko�| jj | _ | j r�|| j�
� 7 }n | j� |�}| jj p�| jdk| _ |d | j� }| jt|�8 _| jdk�rd| _ | �|� |S )Nr r- T)r r r r Zunconsumed_tailr0 �_read2r r r_ �
MIN_READ_SIZEr� r� r� r r? r r r r r: � s4
�
zZipExtFile._read1c C sd | j dkrdS t|| j�}t|| j �}| j�|�}| j t|�8 _ |sLt�| jd ur`| �|�}|S r@ ) r r_ rC �minr r> r0 �EOFErrorr( r r r r rB � s
zZipExtFile._read2c s0 z | j r| j�� W t� �� n t� �� 0 d S r� )r r r �superr� �r� r r r s zZipExtFile.closec C s | j rtd��| jS �Nr8 )r. r} r" r� r r r r s zZipExtFile.seekabler c C sL | j rtd��| js t�d��| �� }|dkr4|}n.|dkrF|| }n |dkrZ| j| }ntd��|| jkrr| j}|dk r~d}|| }|| j }|dkr�|t| j �k r�|| _d}nf|dk �r| j
� | j � | j
| _| j| _| j| _d| _ d| _t| j�| _d| _|}| jd u�r| �� |dk�rDt| j|�}| �|� ||8 }�q| �� 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)r. r} r" r1 �UnsupportedOperationrZ r% r r0 r r rD r# r& r! r$ r r r� r r r r( r) rD �
MAX_SEEK_READr> )r� rT r Zcurr_posZnew_posZ read_offsetZ buff_offsetZread_lenr r r rD sL
zZipExtFile.seekc C s> | j rtd��| js t�d��| j| j t| j� | j }|S )Nztell on closed file.rI )
r. r} r" r1 rJ r% r r0 r r )r� Zfileposr r r rZ I s
zZipExtFile.tell)NF)r� )r )r� )r )r r r r; rC rK r� r) r� r3 r7 r9 r> r? rA r: rB r r rD rZ �
__classcell__r r rG 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� re rf � _compressor�
_file_size�_compress_size�_crc)r� �zfr� r� r r r r� S s �z_ZipWriteFile.__init__c C s | j jS r� )rP r< r� r r r r ] s z_ZipWriteFile._fileobjc C s dS )NTr r� r r r �writablea s z_ZipWriteFile.writablec C s� | j rtd��t|ttf�r&t|�}nt|�}|j}| j|7 _t || j
�| _
| j rt| j � |�}| j
t|�7 _
| j�|� |S rH )r. r} r� r� r� r0 �
memoryview�nbytesrR r� rT rQ r� rS r r )r� rV rX r r r r d s
z_ZipWriteFile.writec sj | j r
d S �zNt� �� | jrR| 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�| j tkr�td��| jtkr�td��| j�� | j_| j�| j j� | j�| j �| j�� | j�| jj� | jj � | j � | j | jj | j j < W d| j_ n
d| j_ 0 d S )Nr zd d!�d"d#�Zd?d$d%�Zd@d&d'�ZdAd(d)�Zed*d+� �Zd,d-� Zd.d/� Z dBd0d1�Z dCd2d3�Z d4d5� Z d6d7� Z d8d9� Z!d:d;� Z"dS )Dr N�rTr� c
C sH |dvrt d��t|� || _d| _d| _i | _g | _|| _|| _|| _ d | _
d| _ || _ t
|tj�rpt�|�}t
|t�r�d| _|| _dddd dd
d d �}|| }zt�||�| _W q� ty� ||v r�|| }Y q�� Y q�0 q�q�nd
| _|| _t|dd �| _d
| _t�� | _ d| _ d| _ �z|dk�r2| � � n�|dv �r�d| _z| j� � | _!W n0 t"tf�y� t#| j�| _d| _!d| _ Y n40 z| j�$| j!� W n t"tf�y� d| _ Y n0 nd|dk�rz| � � | j�$| j!� W n4 t%�y | j�$dd� d| _| j� � | _!Y n0 nt d��W n$ | j} d | _| �&| � � Y n0 d S )N)r^ �w�x�az+ZipFile requires mode 'r', 'w', 'x', or 'a'Fr r- r? �w+b�x+b�r+b�wbZxb)r^ r_ r` ra rd rb rc r r Tr^ )r_ r` ra r z"Mode must be 'r', 'w', 'x', or 'a')'r} r� � _allowZip64�
_didModify�debugr\ r[ r� r� r r� �_comment�_strict_timestampsr� rz r� r� �str� _filePassedrB r1 rA r< r: �getattr� _fileRefCnt� threading�RLockr r" r �_RealGetContentsrZ rZ r' r rD 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__ s zZipFile.__enter__c C s | � � d S r� �r )r� �type�value� tracebackr r r �__exit__ s zZipFile.__exit__c C s� d| j j| j jf g}| jd urd| jr8|�d| j � n| jd urR|�d| j � |�d| j � n
|�d� |�d� d�|�S )Nr, z file=%rz filename=%rz mode=%rr- r� r� ) r� r r r< rl r1 rB r r2 r/ r r r r� ! s �
zZipFile.__repr__c C s� | j }z t|�}W n ty, td��Y n0 |s:td��| jdkrLt|� |t }|t }|t | _ |t
| | }|t t kr�|t
t 8 }| jdkr�|| }td|||� || | _|�| jd� |�|�}t�|�}d}||k �r�|�t�} t| �tk�rtd��t�t| �} | t tk�r*td��| jdk�r>t| � |�| t �}
| d } | d @ �rj|
� 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@ d f| _3| �4� | j#| | _#| j5�6| � | | j7| j8< |t | t | t | t }| jdkr�td|� q�| j}t9| j5dd� dd�D ]}||_:|j#}�q�d S )NzFile is not a zip filer r zgiven, inferred, offsetr zTruncated central directoryz&Bad magic number for central directoryr! r� r� �cp437r zzip file version %.1fr% r( r+ r$ rv r� r& r �totalc S s | j S r� )rn )r� r r r �~ r- z*ZipFile._RealGetContents..T)�key�reverse);r< r9 r: r rh �printrQ rR � _ECD_COMMENTri �
_ECD_LOCATIONrL rK rI rE rZ rD r> r1 �BytesIO�sizeCentralDirr0 rF r/ �structCentralDir�
_CD_SIGNATURE�stringCentralDir�_CD_FILENAME_LENGTH�decoder
�_CD_EXTRA_FIELD_LENGTHr3 �_CD_COMMENT_LENGTHrb �_CD_LOCAL_HEADER_OFFSETrn rW rg rh ri rj re ro rp rq �MAX_EXTRACT_VERSIONr� rk rl rm rr rd r� r[ r1 r\ rB �sortedrs )r� r< rU Zsize_cdZ offset_cd�concatZinferredrV r{ �centdirrB �flagsr` �t�dZ
end_offsetr� r r r rq / s�
� �� ��
�
zZipFile._RealGetContentsc C s dd� | j D �S )Nc S s g | ]
}|j �qS r r� )�.0rV r r r �
� r- z$ZipFile.namelist..�r[ r� r r r �namelist� s zZipFile.namelistc C s | j S r� r� r� r r r �infolist� s zZipFile.infolistc C sL t dd |d� | jD ]0}d|jd d� }t d|j||jf |d� qd S )Nz%-46s %19s %12s)z File Namez Modified ZSize)r z%d-%02d-%02d %02d:%02d:%02dr" z
%-46s %s %12d)r r[ rd rB rq )r� r r� �dater r r �printdir� s �
�zZipFile.printdirc C sp d}| j D ]`}z>| �|jd�� }|�|�r,q W d � n1 s@0 Y W q
tyh |j Y S 0 q
d S )Ni r^ )r[ rA rB r> r )r� Z
chunk_sizer� �fr r r �testzip� s
$ zZipFile.testzipc C s$ | j �|�}|d u r td| ��|S )Nz(There is no item named %r in the archive)r\ r� �KeyError)r� r �infor r r �getinfo� s �zZipFile.getinfoc C s6 |r t |t�s tdt|�j ��|r,|| _nd | _d S )N�pwd: expected bytes, got %s)r� r� � TypeErrorrv r r� )r� r� r r r � setpassword� s
zZipFile.setpasswordc C s | j S r� )ri r� r r r rb � s zZipFile.commentc C s^ t |t�s tdt|�j ��t|�tkrNdd l}|jdt dd� |d t� }|| _ d| _
d S )Nz comment: expected bytes, got %sr z3Archive comment is too long; truncating to %d bytesr ��
stacklevelT) r� r� r� rv r r0 �ZIP_MAX_COMMENT�warnings�warnri rg )r� rb r� r r r rb � s
�� c C s: | � |d|��}|�� W d � S 1 s,0 Y d S �Nr^ �rA r> )r� r r� r< r r r r> � s z ZipFile.readF�� force_zip64c sJ |dvrt d��|r0t|t�s0tdt|�j ��|rD|dkrDt d��� jsRt d��t|t�rb|}n,|dkr�t|�}� j|_ � j
|_ n
� � |�}|dkr�� j
||d�S � jr�t d��� jd 7 _t� j|j� j� j� fd
d ��}�zL|�t�}t|�tk�rtd ��t�t|�}|t t k�r&td
��|�|t �}|t �rL|�|t � |j d@ �r`t d��|j d@ �rtt d��|t! d@ �r�|�"d�} n
|�"d�} | |j#k�r�td|j#|f ��|j$d u�r�|�%� |j& |j$k�r�td|j#�d���|j d @ }
|
�r|�s� j'}|�s t(d| ��nd }t)||||d�W S |�*� � Y n0 d S )N> r^ r_ z open() requires mode "r" or "w"r� r_ 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� rz z/File name in directory %r and header %r differ.zOverlapped entries: z (possible zip bomb)z6File %r is encrypted, password required for extractionT)+r} r� r� r� rv r r< r
r� re r� rf r� �_open_to_writer rn r� rn rr r r> �sizeFileHeaderr0 r rF r/ r� �
_FH_SIGNATUREr� �_FH_FILENAME_LENGTH�_FH_EXTRA_FIELD_LENGTHrj r� � _FH_GENERAL_PURPOSE_FLAG_BITSr� rc rs rZ rp r� r� r r ) r� r r r� r� r� Zzef_fileZfheader�fnameZ fname_strZ is_encryptedr r� r rA � s� �
�
�� �
�z ZipFile.openc C s� |r| j std��| jr td��d|_d|_d|_|jtkrJ| jdO _| js^| jdO _|j sjd|_ | j o�|p�|j
d t k}| jr�| 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)rf r} r rp ro rj re r r" rm rq r� r< rD rZ rZ rn � _writecheckrg r r� rM )r� r� r� r� r r r r� 1 s4
�
�
zZipFile._open_to_writec C s* |d u rt �� }n
t �|�}| �|||�S r� )rz �getcwdr� �_extract_member)r� �memberr� r� r r r �extractZ s
zZipFile.extractc C sH |d u r| � � }|d u r"t�� }n
t�|�}|D ]}| �|||� q0d S r� )r� rz r� r� r� )r� r� �membersr� r* r r r �
extractallg s
zZipFile.extractallc C s^ | j }|s(d}t�|dt|� �}|| _ |�|�}dd� |�|�D �}|�dd� |D ��}|S )Nz:<>|"?*�_c s s | ]}|� d �V qdS )�.N)�rstrip�r� r` r r r � � r- z1ZipFile._sanitize_windows_name..c s s | ]}|r|V qd S r� r r� r r r r� � r- )�!_windows_illegal_name_trans_tablerk � maketransr0 � translate�splitr2 )r� r� �pathsep�tableZillegalr r r �_sanitize_windows_namex s
z ZipFile._sanitize_windows_namec sx t |t�s| �|�}|j�dtjj�}tjjrB|�tjjtjj�}tj� |�d }dtjj
tjj f� tjj� � fdd�|�
tjj�D ��}tjjdkr�| �|tjj�}tj� ||�}tj�|�}tj�|�}|r�tj�|�s�t�|� |�� �r tj�|��st�|� |S | j||d��H}t|d�� }t�||� W d � n1 �sJ0 Y W d � n1 �sj0 Y |S ) Nrw r r� c 3 s | ]}|� vr|V qd S r� r r� �Zinvalid_path_partsr r r� � s �z*ZipFile._extract_member..�\�r� re )r� r
r� rB r| rz r� r{ r� r� �curdir�pardirr2 r� r� r� �dirname�exists�makedirsr� r� �mkdirrA �shutil� copyfileobj)r� r� Z
targetpathr� r� Z upperdirs�source�targetr r� r r� � s2
&
�LzZipFile._extract_memberc C s� |j | jv r(dd l}|jd|j dd� | jdvr:td��| jsHtd��t|j� | j s�d }t
| j �t krpd}n |j
tkr�d }n|jtkr�d
}|r�t|d ��d S ) Nr zDuplicate name: %rr r� �r_ r` ra 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 r\ r� r� r r} r< r� re rf r0 r[ �ZIP_FILECOUNT_LIMITrq r� rn r
)r� r� r� �requires_zip64r r r r� � s,
�
�zZipFile._writecheckc C s� | j std��| jr td��tj||| jd�}|�� rDd|_d|_n0|d urT||_ n| j
|_ |d url||_ n| j |_ |�� �r&| j
�� | jr�| j �| j� | j �� |_|j tkr�| jdO _| �|� d| _| j�|� || j|j< | j �|� d�� | j �� | _W d � n1 �s0 Y nht |d��L}| � |d �� }t � ||d
� W d � n1 �sd0 Y W d � n1 �s�0 Y d S ) N�7Attempt to write to ZIP archive that was already closedz>Can't write to ZIP archive while an open writing handle existsr� r r TFr? r_ i ) r< r} r r
r� rj r� rp ro re r� rf r� r r" rD rZ rZ rn r rj r� rg r[ r1 r\ rB r r� rA r� r� )r� rB r� re r� r� �src�destr r r r � sF ���
.z
ZipFile.writec C s" t |t�r|�d�}t |t�sxt|t�t�� �d d� d�}| j|_| j|_ |j
d dkrpd|_ | j dO _ q|d|_ n|}| j s�t
d ��| jr�t
d
��|d ur�||_|d ur�||_ t|�|_| j�H | j|d d ��}|�|� W d � n1 s�0 Y W d � n1 �s0 Y d S )
Nr� r" )rB rd r� rw i �Ar) r� r� z?Can't write to ZIP archive while an open writing handle exists.r_ )r )r� rk r� r
r� r� r� re r� rf rB rm r< r} r r0 rq r rA r )r� Zzinfo_or_arcnamerV re r� r� r� r r r �writestr� s:
���
zZipFile.writestrc C s | � � d S r� ru r� r r r �__del__" s zZipFile.__del__c C s� | j d u rd S | jr td��zl| jdv rp| jrp| j�, | jrJ| j �| j� | � � W d � n1 sf0 Y W | j }d | _ | �
|� n| j }d | _ | �
|� 0 d S )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 r} r rg r r" rD rZ �_write_end_recordrr r r r r r & s
( �z
ZipFile.closec C s | j D �]�}|j}|d d d> |d d> B |d B }|d d> |d d> B |d d B }g }|jtksr|jtkr�|�|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
tk�r&tt|
�}
n|j
tk�r rA r� r� r� r� r� r� r� r r� r� r r� rr r r r r r � sN �� ZT
f
)
* �
2 �
)^r c @ s0 e Zd Zdeddfdd�Zd dd �Zd
d � ZdS )
r r^ Tr� c C s t j| ||||d� || _d S )N)r r� rs )r r� � _optimize)r� r r r� rs �optimizer r r r� � s �zPyZipFile.__init__r� Nc C s� t �|�}|rD||�sD| jr@t j�|�r,dnd}td||f � d S t j�|�\}}t j�|��rht j�|d�}t j�|��r�|r�d||f }n|}| jr�td|d|� | � |dd � |�\}} | jr�td
| � | �
|| � t t � |��}
|
�
d� |
D ]�} t j�|| �} t j�| �\}
}t j�| ��rRt j�t j�| d���r�| j| ||d � q�|d kr�|�r~|| ��s~| jr�td
| � q�| � | dd � |�\}} | j�r�td
| � | �
|| � q�n�| j�r�td|� t t � |��D ]�} t j�|| �} t j�| �\}
}|d k�r�|�r,|| ��s,| j�r�td
| � �q�| � | dd � |�\}} | j�rVtd
| � | �
|| � �q�nP|d d � d k�r�td��| � |dd � |�\}} | j�r�td| � | �
|| � d S )Nr� r z%s %r skipped by filterfuncz __init__.py�%s/%szAdding package in�asr �����ZAdding)�
filterfunc�.pyz file %r skipped by filterfunczAdding files from directoryz.Files added with writepy() must end with ".py"z Adding file)rz r� rh r� r� r r� r2 �isfile�
_get_codenamer r� �listdir�remove�splitext�writepyr� )r� �pathname�basenamer� Zlabel�dirr 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�}� 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
j d
k�r4|}
nt j
j dk�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 )Nr� c
sd dd l }� jrtd| � z|j| d|d� W n2 |jy^ } zt|j� W Y d }~dS d }~0 0 dS )Nr Z CompilingT)�doraiser� F)�
py_compilerh r �compile�PyCompileError�msg)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� )r� )� importlib�util�cache_from_sourcer� rz 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� � sj �
� ��
� zPyZipFile._get_codename)r� N)r r r r r� r� r� r r r r r � s
�
Rr c C s t �t| �dd �S r� )� itertools�islice� _ancestry)r� r r r �_parentsO s r c c s2 | � tj�} | � tj�r.| V t�| �\} }q d S r� )r� � posixpathr{ r� )r� �tailr r r r b s r c C s t �t|�j| �S r� )r � filterfalse�set� __contains__)ZminuendZ
subtrahendr r r � _difference� s r c sD e Zd Zedd� �Z� fdd�Zdd� Zdd� Zed d
� �Z � Z
S ) � CompleteDirsc C s. t j�tt| ��}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 )�names�parentsZas_dirsr r r �
_implied_dirs� s zCompleteDirs._implied_dirsc s t t| ��� }|t| �|�� S r� )rF r r� r^ r )r� r rG r r r� � s zCompleteDirs.namelistc C s t | �� �S r� )r r� r� r r r � _name_set� s zCompleteDirs._name_setc C s, | � � }|d }||vo ||v }|r(|S |S �Nrw )r )r� r r r� Z dir_matchr r r � resolve_dir� s zCompleteDirs.resolve_dirc C sN t |t�r|S t |t�s | |�S d|jvr.t} | �| �}t|��t|�� |S r� )r� r r r �__new__�vars�update)r� r� �resr r r �make� s
zCompleteDirs.make) r r r � staticmethodr r� r r r� r rL r r rG r r � s
r c s( e Zd Z� fdd�Z� fdd�Z� ZS )�
FastLookupc sF t �t�� | jW d � S 1 s&0 Y tt| ��� | _| jS r� )�
contextlib�suppressr' Z_FastLookup__namesrF r r� r� rG r r r� � s $zFastLookup.namelistc sF t �t�� | jW d � S 1 s&0 Y tt| ��� | _| jS r� )r r r' Z_FastLookup__lookuprF r r r� rG r r r � s $zFastLookup._name_set)r r r r� r rL r r rG r r � s r c @ s� e Zd ZdZd#dd�Zd$dd�Zedd � �Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zd d � ZeZed d!� �Zd"S )%r z>{self.__class__.__name__}({self.root.filename!r}, {self.at!r})r� c C s t �|�| _|| _d S r� )r r r� �at)r� r� r r r r r� s z
Path.__init__r^ c O s\ |� dd �}|d }| jj| j||d�}d|v rD|s8|r@td��|S tj|g|�R i |��S )Nr� r r� �bz*encoding args invalid for binary operation)�popr� rA r r} r1 �
TextIOWrapper)r� r �args�kwargsr� Zzip_mode�streamr r r rA s z Path.openc C s t �| j�d��S r )r r� r r� r� r r r r # s z Path.namec O sD | j dg|�R i |���}|�� W d � S 1 s60 Y d S r� r� )r� r# r$ �strmr r r � read_text' s zPath.read_textc C s6 | � d��}|�� W d � S 1 s(0 Y d S )Nr? r� )r� r& r r r �
read_bytes+ s zPath.read_bytesc C s t �|j�d��| j�d�kS r )r r� r r� )r� r� r r r � _is_child/ s zPath._is_childc C s t | j|�S r� )r r� )r� r r r r �_next2 s z
Path._nextc C s | j p| j �d�S r )r �endswithr� r r r r� 5 s z Path.is_dirc C s
| � � S r� )r� r� r r r �is_file8 s 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. | � � std��t| j| j�� �}t| j|�S )NzCan't listdir a file)r� r} r� r* r� r� �filterr) )r� Zsubsr r r �iterdir> s z Path.iterdirc C s t �| jj| j�S r� )r r2 r� rB r r� r r r �__str__D s z Path.__str__c C s | j j| d�S )Nr� )� _Path__reprr r� r r r r� G s z
Path.__repr__c C s t �| j|�}| �| j�|��S r� )r r2 r r* r� r )r� �add�nextr r r �joinpathJ s z
Path.joinpathc C s( t �| j�d��}|r |d7 }| �|�S r )r r� r r� r* )r� Z parent_atr r r �parentP s z Path.parentN)r� )r^ )r r r r0 r� rA r] r r' r( r) r* r� r, r� r. r/ r� r3 � __truediv__r4 r r r r r � s& A
r c
s dd l }d}|j|d�}|jdd�}|jdddd d
� |jd d d
ddd� |jdddddd� |jddddd
� |�| �} | jd ur�| j}t|d��}|�� }W d � n1 s�0 Y |r�td� |�� td� �nB| j
d u�r"| j
}t|d��}|� � W d � n1 �s0 Y n�| j d u�rp| j \}}t|d��}|�
|� W d � n1 �sd0 Y n�| jd u�r | j�d�} | j}
� fd d �� t| d ��f}|
D ]P} tj�| �} | �s�tj�tj�| ��} | d tjtjfv �r�d } � || | � �q�W d � n1 �s0 Y d S ) Nr z3A simple command-line interface for zipfile module.)� descriptionT)Zrequiredz-lz--list� zShow listing of a zipfile)�metavar�helpz-ez --extractr )r7 z z Extract zipfile into target dir)�nargsr8 r9 z-cz--create�+)zzzCreate zipfile from sourcesz-tz--testzTest if a zipfile is validr^ z.The following enclosed file is corrupted: {!r}z Done testingc sp t j�|�r | �||t� nPt j�|�rl|r8| �||� tt �|��D ]$}� | t j�||�t j�||�� qFd S r� ) rz r� r� r r r� r� r� r2 )rU r� �zippathZnm��addToZipr r r> � s �zmain..addToZipr_ r� )�argparse�ArgumentParserZ add_mutually_exclusive_group� add_argument�
parse_args�testr r� r r r^ r� r� r� Zcreater! rz r� r� r� r� r� )
r# r? r6 �parser�groupr� rU Zbadfiler� Zzip_name�filesr� r< r r= r �mainX s\
�
�
�
�
& *
, rG �__main__)N)N)�Zbinascii�importlib.utilr� r1 r rz r r� r� rF r~ ro r� r 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[ rL rM rN rO rP rQ rR ra r� r� 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 rE rJ 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_CENTDIRrY ZStructr. r8 r= r rX r9 �objectr
r� r� r� r� r� r� r� r� r� r� r r2 r rM r r r r �dict�fromkeysr r r r r rG r r r r r � sR
+= ^& �
( KU /2
=