U
e5d�V � @ 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
d dlZzd dlZejZW n ek
r� dZe jZY nX z d dlZW n ek
r� dZY nX z d dlZW n ek
r� dZY nX dddddddd d
d d d
dg
ZG dd� de�ZG dd
� d
e�Ze ZZdZdZdZ d Z dZ dZ dZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'e
�(e&�Z)d Z*d Z+d Z,d Z-d Z.d Z/d!Z0d"Z1dZ2d#Z3d$Z4d%Z5e
�(e4�Z6d Z7d Z8d Z9d Z:d Z;d ZdZ?d#Z@d&ZAd'ZBdZCd(ZDdZEd)ZFd*ZGd+ZHd,ZId-ZJd.ZKe
�(eJ�ZLd ZMd ZNd ZOd ZPd ZQd ZRd!ZSd"ZTdZUd#ZVd&ZWd'ZXd/ZYd0ZZe
�(eY�Z[d1Z\d2Z]e
�(e\�Z^d Z_d Z`d Zad Zbd Zcd Zdd!Zed"ZfdZgd#Zhd3Zie
�jd4�Zkd5d6� Zld7d8� Zmd9d � Znd:d;� Zodd
� d
eq�Zrdasd?d@� ZtdAdB� ZuG dCdD� dD�ZvG dEdF� dF�ZwdGdHdIdIdIdIdJdKdLdMdJdNdOdPdQdRdSdT�ZxdUdV� ZydsdWdX�ZzdYdZ� Z{G d[d\� d\�Z|G d]d^� d^�Z}G d_d`� d`ej~�ZG dadb� dbej~�Z�G dcd � d �Z�G ddd � d e��Z�dedf� Z�dgdh� Z�e�j�Z�didj� Z�G dkdl� dle��Z�G dmdn� dne��Z�G dod� d�Z�dtdpdq�Z�e�drk�re�� dS )u� 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 � /usr/lib64/python3.8/zipfile.pyr + s c @ s e Zd ZdS )r
Nr r r r r r
/ s 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 sh | � dd� | �� }z| � t d� W n tk
r< Y d S X | �� }t|�tkr�|dd� tkr�|dd � dkr�t�t |�}t
|�}|� d� |� |t � t | t |�S t
|d t d�}| � |d� | �� }|�t�}|dk�rd|||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�NoName�� r r r r r c C s� || _ |�td��}|dkr(|d|� }tjdkrJtj|krJ|�tjd�}|| _|| _|d dk rjtd��t | _
d | _ d| _ d| _
tjdkr�d| _nd| _t| _t| _d| _d| _d| _d| _d| _d S )Nr �/ru 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 )�selfrB rd Z null_byter r r �__init__X s0
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 kr�|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 ru r$ r r! r r r& r r z | j �d�| jdB f Y S X d S )N�ascii�utf-8� )rB �encoderj �UnicodeEncodeError�r� r r r r� � s z ZipInfo._encodeFilenameFlagsc C s� | j }tj}t|�dk�r�|d|d d� �\}}|d t|�krPtd||f ��|dk�rp|dkrv|d|dd� �}nV|dkr�|d |dd
� �}n:|d kr�|d |dd
� �}n |dkr�d}ntd||f ��d}| jdk�rt|�|kr�td��|| | _|d7 }| jdk�r6t|�|k�r$td��|| | _|d7 }| jdk�rpt|�|k�rXtd��| j}|| | _|d7 }||d d � }q d S )Nr r, z"Corrupt extra field %04x (size=%d)r � z |_|r�d|_| jd
O _n|j|_|S ) Nr r" ru rt �; )r� r � � �; r� r rv r r) )�
isinstancery �PathLike�fspathr� �S_ISDIR�st_mode�time� localtime�st_mtime�path�normpath�
splitdriverz �altseprm rq �st_size) �clsrB �arcnamer� �str� �mtimerd �zinfor r r � from_file� s0
zZipInfo.from_filec C s | j d dkS )N�����rv �rB r� r r r r� % s zZipInfo.is_dir)rs rt )N)N) r r r � __slots__r� r� r� r� r� � classmethodr� r� r r r r r
> s
+
.2%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 kr&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 �crc32I s z _ZipDecrypter..crc32c s< � | �����d@ d@ ��d d d@ �� �d? ���d S )Nr� r� i�r r� r )�c)r� �key0�key1�key2r r � update_keysM 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 � decrypterW 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� g s zLZMACompressor.__init__c C sF t �dt ji�}t jt jt �t j|�gd�| _t�dddt |��| S )N�id��filtersz | �� rtd��| j�||� | j�� | _| jW 5 Q R � S Q R X d S )Nz}Can't reposition in the ZIP file while there is an open writing handle on it. Close the writing handle before trying to read.)r r r| r rD rZ r )r� rT �whencer r r rD � s z_SharedFile.seekr� c
C sX | j �H | �� rtd��| j�| j� | j�|�}| j�� | _|W 5 Q R � S Q R X d S )N�yCan't read from the ZIP file while there is an open writing handle on it. Close the writing handle before trying to read.)r r r| r rD r r> rZ �r� �nrV r r r r> � s z_SharedFile.readc C s$ | j d k r | j }d | _ | �|� d S r� )r r )r� �fileobjr r r r s
z_SharedFile.closeN)r )r� )r r r r� 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 k
r� Y nX 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+ \ 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� g s �
��
zZipExtFile.__repr__r� c C sL |dk r>| j �d| j�d }|dkr>| j | j|� }|| _|S tj�| |�S )Nr �
r )r rw r �io�BufferedIOBase�readline)r� �limitr6 �liner r r r5 u 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 �peek� s
zZipExtFile.peekc C s dS �NTr r� r r r �readable� s zZipExtFile.readablec C s |d ks|dk rH| j | jd � }d| _ d| _| jsD|| �| j�7 }q,|S || j }|t| j �k rz| j | j|� }|| _|S |t| j � }| j | jd � }d| _ d| _|dkr�| js�| �|�}|t|�k r�|| _ || _||d |� 7 }q�||7 }|t|�8 }q�|S �Nr r- �r r r �_read1�MAX_Nr0 )r� r �buf�endrV r r r r> � s4
zZipExtFile.readc C s@ | j d krd 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 rC r
r r r r> � s4
�
zZipExtFile._read1c C sd | j dkrdS t|| j�}t|| j �}| j�|�}| j t|�8 _ |sLt�| jd k r`| �|�}|S r< ) r r_ rF �minr r> r0 �EOFErrorr* r
r r r rE s
zZipExtFile._read2c s& z| jr| j�� W 5 t � �� X d S r� )�superr r r r� �r� r r r s zZipExtFile.closec C s | j S r� )r$ r� r r r r s zZipExtFile.seekabler c C s> | j st�d��| �� }|dkr&|}n.|dkr8|| }n |dkrL| j| }ntd��|| jkrd| j}|dk rpd}|| }|| j }|dkr�|t| j�k r�|| _d}nf|dk �r
| j �
| j � | j | _
| j| _| j| _d| _d| _t| j�| _d| _|}| jd k �r
| �� |dk�r6t| j|�}| �|� ||8 }�q
| �� S )N�!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$ r3 �UnsupportedOperationrZ r' r| r r0 r r rD r% r( r# r& r r r� r r r r* r+ rG �
MAX_SEEK_READr> )r� rT r Zcurr_posZnew_posZ read_offsetZ buff_offsetZread_lenr r r rD # sH
zZipExtFile.seekc C s0 | j st�d��| j| j t| j� | j }|S )NrK )r$ r3 rL r' r r0 r r )r� Zfileposr r r rZ S s
zZipExtFile.tell)NF)r� )r )r� )r )r r r r? rF rM r� r+ r� r5 r9 r; r> rC rD r> rE r r rD rZ �
__classcell__r r rJ r r s( �
3
!
%$
0r 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 �z_ZipWriteFile.__init__c C s | j jS r� )rR r< r� r r r r e s z_ZipWriteFile._fileobjc C s dS r: r r� r r r �writablei s z_ZipWriteFile.writablec C sf | j rtd��t|�}| j|7 _t|| j�| _| jrV| j�|�}| jt|�7 _| j �
|� |S )Nz I/O operation on closed file.) r0 r| r0 rT r� rV rS r� rU r r )r� rV �nbytesr r r r l s z_ZipWriteFile.writec sb | j r
d S �zFt� �� | 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 5 d| j_X d S )NFr 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 sP |dkrt d��t|� || _d| _d| _i | _g | _|| _|| _|| _ d | _
d| _ || _ t
|tj�rpt�|�}t
|t�r�d| _|| _dddd dd
d d �}|| }zt�||�| _W q� tk
r� ||kr�|| }Y q�� Y q�X q�q�nd
| _|| _t|dd �| _d
| _t�� | _ d| _ d| _ �z|dk�r4| � � n�|dk�r�d| _z| j� � | _!W n2 t"tfk
�r� t#| j�| _d| _!d| _ Y n6X z| j�$| j!� W n t"tfk
�r� d| _ Y nX nf|dk�r z| � � | j�$| j!� W n6 t%k
�r | j�$dd� d| _| j� � | _!Y nX nt d��W n$ | j} d | _| �&| � � Y nX 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` ra rb re rc rd r r! Tr_ )r` ra rb r z"Mode must be 'r', 'w', 'x', or 'a')'r| r� � _allowZip64�
_didModify�debugr] r\ r� r� r r� �_comment�_strict_timestampsr� ry r� r� �str� _filePassedrB r3 rA r< r: �getattr� _fileRefCnt� threading�RLockr r$ r �_RealGetContentsrZ r[ 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 k rd| jr8|�d| j � n| jd k rR|�d| j � |�d| j � n
|�d� |�d� d�|�S )Nr. z file=%rz filename=%rz mode=%rr/ r� r� ) r� r r r< rm r1 rB r r2 r1 r r r r� " s �
zZipFile.__repr__c C s� | j }z t|�}W n tk
r. td��Y nX |s r3 �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 )r� r< rU Zsize_cdZ offset_cd�concatZinferredrV r| �centdirrB �flagsra �t�dr r r rr 0 s�
� �
� ��
zZipFile._RealGetContentsc C s dd� | j D �S )Nc S s g | ]
}|j �qS r r� )�.0rV r r r �
� s 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 s^ d}| j D ]N}z*| �|jd��}|�|�r,q W 5 Q R X W q
tk
rV |j Y S X 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 kr 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� � TypeErrorrw r r� )r� r� r r r � setpassword� s
zZipFile.setpasswordc C s | j S r� )rj 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� rw r r0 �ZIP_MAX_COMMENT�warnings�warnrj rh )r� rb r� r r r rb � s
�� c
C s. | � |d|��}|�� W 5 Q R � S Q R X d S �Nr_ �rA r> )r� r! r� r< r r r r> � s z ZipFile.readF�� force_zip64c s |dkrt 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 ��}�z|�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 @ }
|
�r�|�s҈ j$}|�s�t%d| ��nd }t&||||d�W S |�'� � Y nX 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- 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.z6File %r is encrypted, password required for extractionT)(r| r� r� r� rw r r< r
r� re r� rf r� �_open_to_writer ro r rn rs r r> �sizeFileHeaderr0 r rF r/ r� �
_FH_SIGNATUREr� �_FH_FILENAME_LENGTH�_FH_EXTRA_FIELD_LENGTHrj r� � _FH_GENERAL_PURPOSE_FLAG_BITSr� rc 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��t|d�s0d|_d|_d|_d|_|jt krZ| jdO _| j
sn| jdO _|j szd|_ | j o�|p�|jd t 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.rq r r r � �g�������?T)rg r| r r@ rq rp ro rj re r r$ rm r� r< rD r[ rZ rn � _writecheckrh r r� rO )r� r� r� r� r r r r� ( s8
�
�
zZipFile._open_to_writec C s* |d krt �� }n
t �|�}| �|||�S r� )ry �getcwdr� �_extract_member)r� �memberr� r� r r r �extractS s
zZipFile.extractc C sH |d kr| � � }|d kr"t�� }n
t�|�}|D ]}| �|||� q0d S r� )r� ry r� r� r� )r� r� �membersr� r, r r r �
extractall` 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� ra r r r � { s z1ZipFile._sanitize_windows_name..c s s | ]}|r|V qd S r� r r� r r r r� } s )�!_windows_illegal_name_trans_tablerl � maketransr0 � translate�splitr2 )r� r� �pathsep�tableZillegalr r r �_sanitize_windows_nameq s
z ZipFile._sanitize_windows_namec
sL 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��(}t|d��}t�||� W 5 Q R X W 5 Q R X |S ) Nrv r r� c 3 s | ]}|� kr|V qd S r� r r� �Zinvalid_path_partsr r r� � s �z*ZipFile._extract_member..�\)r� rf )r� r
r� rB r{ ry r� rz 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
&
� zZipFile._extract_memberc C s� |j | jkr(dd l}|jd|j dd� | jdkr: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` ra rb 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 rg r0 r\ �ZIP_FILECOUNT_LIMITrq r� rn r
)r� r� r� �requires_zip64r r r r� � s,
�
�zZipFile._writecheckc
C sP | j std��| jr td��tj||| jd�}|�� rDd|_d|_n0|d k rT||_ n| j
|_ |d k rl||_ n| j |_ |�� �r| j
�� | jr�| j �| j� | j �� |_|j tkr�| jdO _| �|� d| _| j�|� || j|j< | j �|� d�� | j �� | _W 5 Q R X nCan't write to ZIP archive while an open writing handle existsr� r r TFr? r` i ) r< r| r r
r� rk r� rp ro re r� rf r� r r$ rD r[ rZ rn r rj r� rh 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 k r�||_|d k r�||_ t|�|_| j�* | j|d d ��}|�|� W 5 Q R X W 5 Q R X d S )
Nr� r" )rB rd r� rv i �Ar) r� r� z?Can't write to ZIP archive while an open writing handle exists.r` )r )r� rl 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� rv r� r r r �__del__ s zZipFile.__del__c C s| | j d krd S | jr td��zB| jdkr\| jr\| j�" | jrJ| j �| j � | �
� W 5 Q R X W 5 | j }d | _ | �|� X 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| rs r rh r r$ rD r[ �_write_end_recordr r r r r s
z
ZipFile.closec C s� | j D �]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}
|�r t| d �} t j
d d
t |� ddt |� f|�� | } t }
|j
tk�r$tt|
�}
n|j
tk�r:tt|
�}
t|
|j�} t|
|j�} zZ|�� \}
}t �
tt| |j| |j||j
|||j||t |
�t | �t |j�d|j |j |�}W nl t k
�r t tt| |j| |j|j |j
|||j||t |j!�t | �t |j�d|j |j |ft"j#d� � Y nX | j$�%|� | j$�%|
� | j$�%| � | j$�%|j� q| j$�&� }t | j �}|| j' }| j'}d }|t(k�r�d}n |tk�r�d}n|tk�r�d}|�r$| j)�s�t*|d ��t �
t+t,ddddd||||� }| j$�%|� t �
t-t.d|d�}| j$�%|� t/|d�}t/|d
�}t/|d
�}t �
t0t1dd||||t | j2�� }| j$�%|� | j$�%| j2� | j3dk�rt| j$�4� | j$�5� d S )Nr ru r$ r r! r r r& r r� )r r, �Qr r� r� zCentral directory offsetzCentral directory sizer� �, r r rb )6r\ rd rq r� rp r1 rn r3 r8 rF r� r0 r� re r r_ r� r r� rh rW r� r� r� rg ri ro rb rl rm �DeprecationWarningr} rj rB r} �stderrr< r rZ r[ r� rg r
rJ rK rG rH rG r] r\ rj r �truncater� )r� r� r� r� r� r3 rq rp rn Z
extra_datar� rh rW rB rj r� Zpos2Z centDirCountZ centDirSizeZ
centDirOffsetr� Z zip64endrecZ zip64locrecrU r r r r� 5 s $$
�
���
� ��
� � �
�
zZipFile._write_end_recordc C s& | j d8 _ | j s"| js"|�� d S �Nr )ro rm r r r r r rs � s zZipFile._fpclose)N)N)r_ N)F)NN)NNN)NNN)NN)#r r r r< r� r r� ru rz r� rr r� r� r� r� r� r� r^ rb �setterr> rA r� r� r� r� r� r� r� r r� r� r r� rs r r r r r � sR �� ZN
b
+
* �
2 �
)hc @ s0 e Zd Zdeddfdd�Zd dd �Zd
d � ZdS )
r r_ Tr� c C s t j| ||||d� || _d S )N)r r� rt )r r� � _optimize)r� r r r� rt �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)ry r� ri r� r� r} r� r2 �isfile�
_get_codenamer �sorted�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
sf dd l }� jrtd| � z|j| d|d� W n4 |jk
r` } zt|j� W Y �dS d }~X Y nX dS )Nr Z CompilingT)�doraiser� F)�
py_compileri 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� ry 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 �
Rc C s t �t| �dd �S r� )� itertools�islice� _ancestry)r� r r r �_parentsR s r c c s4 | � tj�} | r0| tjkr0| V t�| �\} }q d S r� )r� � posixpathrz r� )r� �tailr r r r e 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 rz )r� r� r r r r� � s 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� )rI r
r� r^ r )r� r rJ 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 }||ko ||k}|r(|S |S �Nrv )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|jkr.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 rN r r rJ r r
� s
r
c s( e Zd Z� fdd�Z� fdd�Z� ZS )�
FastLookupc
s: t �t�� | jW 5 Q R � S Q R X tt| ��� | _| jS r� )�
contextlib�suppressr) Z_FastLookup__namesrI r r� r� rJ r r r� � s zFastLookup.namelistc
s: t �t�� | jW 5 Q R � S Q R X tt| ��� | _| jS r� )r r r) Z_FastLookup__lookuprI r r r� rJ r r r � s zFastLookup._name_set)r r r r� r rN r r rJ r r � s r c @ s� e Zd ZdZd"dd�Zedd� �Zedd� �Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd � 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__c C s t �| jj| j�S r� )� functools�partialr� rA r r� r 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 s6 | � � �$}tj|f|�|��� W 5 Q R � S Q R X d S r� )rA r3 �
TextIOWrapperr> )r� �args�kwargs�strmr r r � read_text s
zPath.read_textc
C s( | � � �}|�� W 5 Q R � S Q R X d S r� 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 �_next% s z
Path._nextc C s | j p| j �d�S r )r �endswithr� r r r r� ( s z Path.is_dirc C s
| � � S r� )r� r� r r r �is_file+ s z Path.is_filec C s | j | j�� kS 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 �iterdir1 s z Path.iterdirc C s t �| jj| j�S r� )r r2 r� rB r r� r r r �__str__7 s z Path.__str__c C s | j j| d�S )Nr� )� _Path__reprr r� r r r r� : s z
Path.__repr__c C s t �| j|�}| �| j�|��S r� )r r2 r r* r� r )r� �add�nextr r r �joinpath= s z
Path.joinpathc C s( t �| j�d��}|r |d7 }| �|�S r )r r� r r� r* )r� Z parent_atr r r �parentC s z Path.parentN)r� )r r r r0 r� r^ rA r! r' r( r) r* r� r, r� r. r/ r� r3 � __truediv__r4 r r r r r � s( A
c
s� dd l }d}|j|d�}|jdd�}|jdddd d
� |jd d d
ddd� |jdddddd� |jddddd
� |�| �} | jd k r�| j}t|d��}|�� }W 5 Q R X |r�td� |�� td� n�| j
d k r�| j
}t|d��}|� � W 5 Q R X n�| j d k �r,| j \}}t|d��}|�
|� W 5 Q R X n�| jd k �r�| j�d�} | j}
� fd d �� t| d ��\}|
D ]P} tj�| �} | �s�tj�tj�| ��} | d tjtjfk�r�d } � || | � �qfW 5 Q R X 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� ) ry r� r� r r r� r� r� r2 )rW r� �zippathZnm��addToZipr r r> s s �zmain..addToZipr` r� )�argparse�ArgumentParserZ add_mutually_exclusive_group� add_argument�
parse_argsZtestr r� r} r r^ r� r� r� Zcreate�popry r� r� r� r� r� )
r$ r? r6 �parser�groupr� rW Zbadfiler� Zzip_name�filesr� r< r r= r �mainK s\
�
�
�
�
rG �__main__)N)N)�Zbinasciir! �importlib.utilr� r3 r ry r r� r� rF r} rp 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_CENTDIRrZ ZStructr. r8 r= r rX r9 �objectr
r� r� r� r� r� r� r� r� r� r r r4 r rO r r r r �dict�fromkeysr r r
r r rG r r r r r � sj
�
+= q& �
& AN /2
=