JezK
Edit File: zipfile.cpython-39.pyc
a �iC[ � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZzddlZejZW n ey� dZejZY n0 z ddlZW n ey� dZY n0 z ddlZW n ey� dZY n0 g d�ZG dd� de�ZG dd� de�Ze ZZdZd Zd Z dZ d Z d Z d Z d Z!dZ"dZ#dZ$dZ%dZ&dZ'e �(e&�Z)dZ*dZ+dZ,dZ-dZ.dZ/dZ0dZ1d Z2dZ3dZ4d Z5e �(e4�Z6dZ7dZ8dZ9dZ:dZ;dZ<dZ=dZ>d Z?dZ@d ZAd ZBd ZCd ZDd ZEd ZFd!ZGd"ZHd#ZId$ZJd%ZKe �(eJ�ZLdZMdZNdZOdZPdZQdZRdZSdZTd ZUdZVd ZWd ZXd&ZYd'ZZe �(eY�Z[d(Z\d)Z]e �(e\�Z^dZ_dZ`dZadZbdZcdZddZedZfd ZgdZhd*Zie �jd+�Zkd,d-� Zld.d/� Zmd0d1� Znd2d3� Zod4d5� ZpG d6d7� d7eq�Zrdasd8d9� Ztd:d;� ZuG d<d=� d=�ZvG d>d?� d?�Zwd@dAdBdBdBdBdCdDdEdFdCdGdHdIdJdKdLdM�ZxdNdO� ZydodPdQ�ZzdRdS� Z{G dTdU� dU�Z|G dVdW� dW�Z}G dXdY� dYej~�ZG dZd[� d[ej~�Z�G d\d]� d]�Z�G d^d_� d_e��Z�d`da� Z�dbdc� Z�e�j�Z�ddde� Z�G dfdg� dge��Z�G dhdi� die��Z�G djdk� dk�Z�dpdldm�Z�e�dnk�r�e�� dS )qzP Read and write ZIP files. XXX references to utf-8 need further investigation. � 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.9/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 . s r i���� � � � � �- �. �? s <4s4H2LHs PK� � � � � � � � z<4s4B4HL2L5H2Ls PK� � � � � � � z <4s2B4HL2L2Hs PKz<4sLQLs PKz <4sQ2H2L4Qs PKiPK�<HHc C s� t j}d}g }d }}|d t| �krz|| ||d � �\}}|d | } ||v rt||krl|�| ||� � | }d}| }q|s�| S d�|�S )NFr r T� )�_EXTRA_FIELD_STRUCT�unpack�len�append�join) �extraZxidsr/ Zmodified�buffer�start�iZxidZxlen�jr r r � _strip_extra� s r8 c C s* zt | �rW dS W n ty$ Y n0 dS )NTF)� _EndRecData�OSError��fpr r r �_check_zipfile� s r= c C sl d}zLt | d�r t| d�}n2t| d��}t|�}W d � n1 sD0 Y W n ttfyf Y n0 |S )z�Quickly see if a file is a ZIP file by checking the magic number. The filename argument may be a file or file-like object too. F�readr; �rbN)�hasattrr= �openr: r )�filename�resultr< r r r r � s *r c C s� |t 8 }|dk r|S | �|� | �t �}t|�t kr<td��t�t|�\}}}}|tkr\|S |dksl|dkrtt d��|t 8 }||kr�t d��| �|� || }| �t �}t|�t kr�td��|� t �s�||kr�| �|� d}| �t �}t|�t kr�td��|� t ��st d��t�t |�\ }} } } } } }}}}|| |k�sN| d t | k�rVt d��||t< | |t< | |t< ||t< ||t< ||t< ||t< || |t< |S ) zM Read the ZIP64 end-of-archive records and use that to update endrec r zUnknown I/O errorr z3zipfiles that span multiple disks are not supportedz.Corrupt zip64 end of central directory locatorz/Zip64 end of central directory record not foundr z-Corrupt zip64 end of central directory record)�sizeEndCentDir64Locator�seekr> r0 r: �structr/ �structEndArchive64Locator�stringEndArchive64Locatorr �sizeEndCentDir64� startswith�stringEndArchive64�structEndArchive64�_ECD_SIGNATURE�_ECD_DISK_NUMBER�_ECD_DISK_START�_ECD_ENTRIES_THIS_DISK�_ECD_ENTRIES_TOTAL� _ECD_SIZE� _ECD_OFFSET� _ECD_LOCATION)�fpin�offset�endrec�data�sigZdisknoZreloffZdisksZextraszZsz�create_versionZ read_versionZdisk_numZdisk_dirZdircountZ dircount2ZdirsizeZ diroffsetr r r � _EndRecData64� sX �� r[ c C sd | � dd� | �� }z| � t d� W n ty: Y dS 0 | �� }t|�tkr�|dd� tkr�|dd� dkr�t�t |�}t |�}|� d� |� |t � t | |t |�S t |d t d�}| � |d� | �� }|�t�}|dk�r`|||t � }t|�tk�rdS t t�t |��}|t }||t |t | � }|� |� |� || � t | || |�S dS ) z�Return 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 �tell�sizeEndCentDirr: r> r0 �stringEndArchiverF r/ �structEndArchive�listr1 r[ �max�rfind�_ECD_COMMENT_SIZE) rU ZfilesizerX rW ZmaxCommentStartr5 ZrecDataZ commentSize�commentr r r r9 s@ �� r9 c @ s\ e Zd ZdZdZddd�Zdd� Zdd d �Zd d � Zdd� Z e ddd�dd��Z dd� Z d S )r z>Class with attributes describing each file in the ZIP archive.)� orig_filenamerB � date_time� compress_type�_compresslevelre r3 � create_systemrZ �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 �/ry z+ZIP does not support timestamps before 1980r- �win32r ) rf �find�chr�os�sep�replacerB rg � ValueErrorr rh ri re r3 �sys�platformrj �DEFAULT_VERSIONrZ rk rl rm rn ro rp rs rt rv )�selfrB rg Z null_byter r r �__init__h 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 rh r r1 �compressor_names�getrp �stat�filemode�is_dirrt rs 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 )z-Return the per-file header as a bytes object.r ry r$ r r! r r r& r r Nz<HHQQz'Filesize would require ZIP64 extensions� �� )rg rm rr rs rt r3 � ZIP64_LIMITrF �pack�calcsizer � ZIP64_VERSIONrh r rb � BZIP2_VERSIONr � LZMA_VERSIONrk rZ �_encodeFilenameFlags�structFileHeader�stringFileHeaderrl r0 )r� �zip64�dt�dosdate�dostimerr rs rt r3 � min_version�fmtrB rm �headerr r r � FileHeader� sJ $$ � �zZipInfo.FileHeaderc C sB z| j �d�| jfW S ty< | j �d�| jdB f Y S 0 d S )N�ascii�utf-8� )rB �encoderm �UnicodeEncodeError�r� r r r r� � s z ZipInfo._encodeFilenameFlagsc C s2 | j }tj}t|�dk�r.|d|d d� �\}}|d t|�krPtd||f ��|dk�r |d|d � }z�| jdv r�d}|d|d d� �\| _|dd � }| jd kr�d }|d|d d� �\| _|dd � }| jd kr�d }|d|d d� �\| _W n( tj�y td |� d ��d �Y n0 ||d d � }q d S )Nr r, z"Corrupt extra field %04x (size=%d)r )l ���� r� z File sizez<Qr r� z Compress sizez Header offsetzCorrupt zip64 extra field. z not found.) r3 rF r/ r0 r rt rs rq r )r� r3 r/ �tpZlnrX Zfieldr r r � _decodeExtra� s6 ��zZipInfo._decodeExtraT��strict_timestampsc C s t |tj�rt�|�}t�|�}t�|j�}t�|j �}|dd� }|sZ|d dk rZd}n|sn|d dkrnd}|du rz|}tj � tj � |�d �}|d tj tjfv r�|dd� }q�|r�|d 7 }| ||�}|jd @ d > |_|r�d|_| jd O _n|j|_|S ) a_ Construct an appropriate ZipInfo for a file on the filesystem. filename should be the path to a file or directory on the filesystem. arcname is the name which it will have within the archive (by default, this will be the same as filename, but without a drive letter and with leading path separators removed). r r" ry rx �; )r� r � � �; r� Nr rz r r) )� isinstancer~ �PathLike�fspathr� �S_ISDIR�st_mode�time� localtime�st_mtime�path�normpath� splitdriver �altseprp rt �st_size) �clsrB �arcnamer� �str� �mtimerg �zinfor r r � from_file� s0 zZipInfo.from_filec C s | j d dkS )z2Return True if this archive member is a directory.���rz �rB r� r r r r� ! s zZipInfo.is_dir)rw rx )N)N) r r r r � __slots__r� r� r� r� r� � classmethodr� r� r r r r r M 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 )z(Compute the CRC32 primitive on one byte.r � r )Zchr� )�crctabler r �crc32E s z _ZipDecrypter.<locals>.crc32c s< � | �����d@ d@ ��d d d@ �� �d? ���d S )Nr� r� i�r � r )�c)r� �key0�key1�key2r r � update_keysI s z"_ZipDecrypter.<locals>.update_keysc sN t � }|j}| D ]4}� 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 � decrypterS s z _ZipDecrypter.<locals>.decrypter)� _crctablera �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� c s zLZMACompressor.__init__c C sF t �dt ji�}t jt jt �t j|�gd�| _t�dddt |��| S )N�id��filtersz<BBHr$ r ) �lzma�_encode_filter_properties� FILTER_LZMA1r� � FORMAT_RAW�_decode_filter_propertiesr� rF r� r0 )r� Zpropsr r r �_initf s � zLZMACompressor._initc C s* | j d u r | �� | j �|� S | j �|�S r� )r� r� �compress)r� rX r r r r� m s zLZMACompressor.compressc C s& | j d u r | �� | j �� S | j �� S r� )r� r� �flushr� r r r r� r s zLZMACompressor.flushN)r r r r� r� r� r� r r r r r� a s r� c @ s e Zd Zdd� Zdd� ZdS )�LZMADecompressorc C s d | _ d| _d| _d S )Nr- F)�_decomp� _unconsumed�eofr� r r r r� z s zLZMADecompressor.__init__c C s� | j d u r�| j|7 _t| j�dkr*dS t�d| jdd� �\}t| j�d| krXdS tjtjt�tj | jdd| � �gd�| _ | jd| d � }| `| j � |�}| j j | _ |S )Nr r- z<Hr r� ) r� r� r0 rF r/ r� r� r� r� r� � decompressr� )r� rX ZpsizerC r r r r� s" �� zLZMADecompressor.decompressN)r r r r� r� r r r r r� x s r� �storeZshrink�reduceZimplode�tokenizeZdeflateZ deflate64�bzip2r� ZterseZlz77ZwavpackZppmd)r r r r r r! r" r# r r$ r% r r r+ � �a �b c C sX | t kr nJ| tkr tsTtd��n4| tkr6tsTtd��n | tkrLtsTtd��ntd��d S )Nz.Compression requires the (missing) zlib modulez-Compression requires the (missing) bz2 modulez.Compression requires the (missing) lzma modulez(That compression method is not supported) r r �zlib� RuntimeErrorr �bz2r r� �NotImplementedError)� compressionr r r �_check_compression� s$ ���r c C sj | t kr2|d ur t�|tjd�S t�tjtjd�S | tkrT|d urLt�|�S t�� S | tkrbt � S d S d S )N��) r r� � compressobj�DEFLATED�Z_DEFAULT_COMPRESSIONr r� � BZ2Compressorr r� )rh � compresslevelr r r �_get_compressor� s r c C sv t | � | tkrd S | tkr&t�d�S | tkr6t�� S | tkrDt � S t � | �}|rdt d| |f ��nt d| f ��d S )Nr zcompression type %d (%s)zcompression type %d) r r r r� � decompressobjr r� �BZ2Decompressorr r� r� r� r� )rh Zdescrr r r �_get_decompressor� s r c @ s8 e Zd Zdd� Zdd� Zddd�Zdd d �Zd d � Zd S )� _SharedFilec C s* || _ || _|| _|| _|| _|j| _d S r� )�_file�_pos�_close�_lock�_writing�seekable)r� �file�pos�close�lock�writingr r r r� � s z_SharedFile.__init__c C s | j S r� )r r� r r r r] � s z_SharedFile.tellr c C sZ | j �@ | �� rtd��| j�||� | j�� | _| jW d � S 1 sL0 Y 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 rE r] r )r� rV �whencer r r rE � s z_SharedFile.seekr� c C sd | j �J | �� rtd��| j�| j� | j�|�}| j�� | _|W d � S 1 sV0 Y 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 rE r r> r] �r� �nrX 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� 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 s | j S r� )rV r� r r r r] 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 r] r� r r r r r r s r c s� e Zd ZdZdZdZdZd(dd�Zd d � Zd d � Z d)dd�Z d*dd�Z dd� Z d+dd�Z 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|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 rr Tr r� r� zBad password for file %r)%�_fileobj�_pwd�_close_fileobjrh �_compress_typers �_compress_leftrt �_leftr � _decompressor�_eof� _readbuffer�_offset�newlines�moderB �namer@ rr � _expected_crcr� � _running_crc� _seekabler r] �_orig_compress_start�_orig_compress_size�_orig_file_size�_orig_start_crc�AttributeError� _decrypterrm ru �_init_decrypterr� rf )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" r6 r! r> r% )r� r� r r r r7 Z 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� e s � �� zZipExtFile.__repr__r� c C sL |dk r>| j �d| j�d }|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) r| r* �io�BufferedIOBase�readline)r� �limitr6 �liner r r rA s s zZipExtFile.readliner c C sr |t | j�| j kr\| �|�}t |�| jkrJ|| 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 rtd��dS )N� I/O operation on closed file.T)r< r� r� r r r �readable� s zZipExtFile.readablec C s | j rtd��|du 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 )z�Read 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- )r< r� r) r* r( �_read1�MAX_Nr0 )r� r �buf�endrX r r r r> � s8 zZipExtFile.readc C s@ | j d u rd S t|| j�| _| jr<| j| j kr<td| j ��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 rR| j | jd� }d| _ d| _| jsN| �| j�}|r,||7 }qNq,|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|r�||7 }�qq�|S )z7Read up to n bytes with at most one read() system call.Nr r- )r) r* r( rH rI r0 )r� r rJ rX rK r r r �read1� s> 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' �unconsumed_tailr0 �_read2r r% rb � MIN_READ_SIZEr� r� r� r& rM r r r r rH � s4 � zZipExtFile._read1c C sd | j dkrdS t|| j�}t|| j �}| j�|�}| j t|�8 _ |sLt�| jd ur`| �|�}|S )Nr r- ) r% rb rQ �minr! r> r0 �EOFErrorr6 r r r r rP 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 �NrF )r< r� r0 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� r0 r? �UnsupportedOperationr] r3 r* r0 r) r! rE r1 r4 r/ r2 r% r& r r$ r' r( r6 r7 rR � MAX_SEEK_READr> )r� rV r Zcurr_posZnew_posZ read_offsetZ buff_offsetZread_lenr r r rE ' sL zZipExtFile.seekc C s> | j rtd��| js t�d��| j| j t| j� | j }|S )Nztell on closed file.rW ) r< r� r0 r? rX r3 r& r0 r) r* )r� Zfileposr r r r] Y s zZipExtFile.tell)NF)r� )r )r� )r )r r r r rI rQ rY r� r7 r� rA rE rG r> rM rN rH rP r r rE r] � __classcell__r r rU 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 rh ri � _compressor� _file_size�_compress_size�_crc)r� �zfr� r� r r r r� c s �z_ZipWriteFile.__init__c C s | j jS r� )r^ r< r� r r r r! m s z_ZipWriteFile._fileobjc C s dS )NTr r� r r r �writableq 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 rV )r< r� r� r� r� r0 � memoryview�nbytesr` r� rb r_ r� ra r! r )r� rX rf r r r r t 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 z<LLQQz<LLLLz+File size unexpectedly exceeded ZIP64 limitz1Compressed size unexpectedly exceeded ZIP64 limitF)!r< rT r r_ r� ra r0 r! r r\ rs r` rb rr rt rm r] rF r� � _DD_SIGNATUREr] r^ � start_dirr� r� rE rq r� �filelistr1 � NameToInforB r )r� rJ r� rU r r r � sD � � �z_ZipWriteFile.close) r r r r� �propertyr! rd r r rZ r r rU r r[ b s r[ c @ s$ e Zd ZdZdZdZdeddfdd�dd�Zdd � Zd d � Z d d � Z dd� Z dd� Z dd� Z d=dd�Zdd� Zdd� Zdd� Zed d � �Zejd d � �Zd>d d �Zd?d!d"�d#d$�Zd@d%d&�ZdAd'd(�ZdBd)d*�Zed+d,� �Zd-d.� Z d/d0� Z dCd1d2�Z dDd3d4�Z d5d6� Z d7d8� Z!d9d:� Z"d;d<� Z#dS )Er ai Class with methods to open, read, write, close, list zip files. z = ZipFile(file, mode="r", compression=ZIP_STORED, allowZip64=True, compresslevel=None) file: Either the path to the file, or a file-like object. If it is a path, the file will be opened and closed by ZipFile. mode: The mode can be either read 'r', write 'w', exclusive create 'x', or append 'a'. compression: ZIP_STORED (no compression), ZIP_DEFLATED (requires zlib), ZIP_BZIP2 (requires bz2) or ZIP_LZMA (requires lzma). allowZip64: if True ZipFile will create files with ZIP64 extensions when needed, otherwise it will raise an exception when this would be necessary. compresslevel: None (default for the given compression type) or an integer specifying the level to pass to the compressor. When using ZIP_STORED or ZIP_LZMA this keyword has no effect. When using ZIP_DEFLATED integers 0 through 9 are accepted. When using ZIP_BZIP2 integers 1 through 9 are accepted. N�rTr� c C sH |dvrt d��t|� || _d| _d| _i | _g | _|| _|| _|| _ d| _ d| _ || _ t |tj�rpt�|�}t |t�r�d| _|| _ddd d 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 dS )z]Open the ZIP file with mode read 'r', write 'w', exclusive create 'x', or append 'a'.)rl �w�x�az+ZipFile requires mode 'r', 'w', 'x', or 'a'Fr Nr- r? �w+b�x+b�r+b�wb�xb)rl rm rn ro rr rp rq r r- Trl )rm rn ro r z"Mode must be 'r', 'w', 'x', or 'a')'r� r � _allowZip64� _didModify�debugrj ri r� r r, r� �_comment�_strict_timestampsr� r~ r� r� �str� _filePassedrB r? rA r< r: �getattr� _fileRefCnt� threading�RLockr r0 r �_RealGetContentsr] rh r5 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__+ 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< r{ r1 rB r, r2 r= r r r r� 1 s � zZipFile.__repr__c C s� | j }z t|�}W n ty, td��Y n0 |s:td��| jdkrLt|� |t }|t }|t | _ |t | | }| jdkr�|| }td|||� || | _ |� | j d� |� |�}t�|�}d}||k �r�|� t�} t| �tkr�td��t�t| �} | t tk�rtd��| jdk�r$t| � |� | t �} | d } | d @ �rP| �d �} n | �d �} t| �} |� | t �| _|� | t �| _ | t | _ | dd � \ | _ | _!| _"| _#| _$| _%} }| _&| _'| _(| j"t)k�r�t*d | j"d ��| dd� \| _+| _,| _-| | _.|d? d |d? d@ |d@ | d? | d? d@ | d@ d f| _/| �0� | j | | _ | j1�2| � | | j3| j4< |t | t | t | t }| jdkr�td|� q�| j }t5| j1dd� dd�D ]}||_6|j }�q�dS ) z/Read in the table of contents for the ZIP file.zFile 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$ ry r� r& r �totalc S s | j S r� )rq )r� r r r �<lambda>� r- z*ZipFile._RealGetContents.<locals>.<lambda>T)�key�reverseN)7r< r9 r: r rw �printrR rS � _ECD_COMMENTrx rT rh rE r> r? �BytesIO�sizeCentralDirr0 rF r/ �structCentralDir� _CD_SIGNATURE�stringCentralDir�_CD_FILENAME_LENGTH�decoder �_CD_EXTRA_FIELD_LENGTHr3 �_CD_COMMENT_LENGTHre �_CD_LOCAL_HEADER_OFFSETrq rZ rj rk rl rm rh rr rs rt �MAX_EXTRACT_VERSIONr� rn ro rp ru rg r� ri r1 rj rB �sortedrv )r� r< rW Zsize_cdZ offset_cd�concatZinferredrX r� �centdirrB �flagsrn �t�dZ end_offsetr� r r r r� ? 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 � <listcomp>� r- z$ZipFile.namelist.<locals>.<listcomp>�ri r� r r r �namelist� s zZipFile.namelistc C s | j S )zJReturn a list of class ZipInfo instances for files in the archive.r� r� r r r �infolist� s zZipFile.infolistc C sL t dd |d� | jD ]0}d|jdd� }t d|j||jf |d� qdS )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� ri rg rB rt )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 dS )z%Read all the files and check the CRC.i rl N)ri rA rB r> r )r� Z chunk_sizer� �fr r r �testzip� s $ zZipFile.testzipc C s$ | j �|�}|du r td| ��|S )z,Return the instance of ZipInfo given 'name'.Nz(There is no item named %r in the archive)rj r� �KeyError)r� r- �infor r r �getinfo� s �zZipFile.getinfoc C s6 |r t |t�s tdt|�j ��|r,|| _nd| _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 s | j S )z.The comment text associated with the ZIP file.)rx r� r r r re � 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� r� r r0 �ZIP_MAX_COMMENT�warnings�warnrx rv )r� re r� r r r re � s �� c C s: | � |d|��}|�� W d � S 1 s,0 Y dS )zReturn file bytes for name.rl N�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$du�r�|�%� |j& |j$k�r�td|j#�d���|j d @ } | �r|�s� j'}|�s t(d| ��nd}t)||||d�W S |�*� � Y n0 dS )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. > rm rl z open() requires mode "r" or "w"r� rm 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.<locals>.<lambda>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� rh r ri r� �_open_to_writer r} r rq r� r r> �sizeFileHeaderr0 r rF r/ r� � _FH_SIGNATUREr� �_FH_FILENAME_LENGTH�_FH_EXTRA_FIELD_LENGTHrm r� � _FH_GENERAL_PURPOSE_FLAG_BITSr� rf rv r] rs 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)ru r� r rs rr rm rh r r0 rp rt r� r< rE rh r] rq � _writecheckrv r r� r[ )r� r� r� r� r r r r� > s4 � � zZipFile._open_to_writec C s* |du rt �� }n t �|�}| �|||�S )a# Extract a member from the archive to the current working directory, using its full name. Its file information is extracted as accurately as possible. `member' may be a filename or a ZipInfo object. You can specify a different directory using `path'. N)r~ �getcwdr� �_extract_member)r� �memberr� r� r r r �extractg s zZipFile.extractc C sH |du r| � � }|du r"t�� }n t�|�}|D ]}| �|||� q0dS )z�Extract 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� r8 r r r � extractallt s zZipFile.extractallc C s^ | j }|s(d}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)�rstrip�r� rn r r r � <genexpr>� r- z1ZipFile._sanitize_windows_name.<locals>.<genexpr>c s s | ]}|r|V qd S r� r r� r r r r� � r- )�!_windows_illegal_name_trans_tablerz � maketransr0 � translate�splitr2 )r� r� �pathsep�tableZillegalr r r �_sanitize_windows_name� 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 ) zbExtract the ZipInfo object 'member' to a physical file on the path targetpath. rz r r� c 3 s | ]}|� vr|V qd S r� r r� �Zinvalid_path_partsr r r� � s �z*ZipFile._extract_member.<locals>.<genexpr>�\�r� rs N)r� r r� rB r� r~ r� r r� r� �curdir�pardirr2 r� r� r� �dirname�exists�makedirsr� r� �mkdirrA �shutil� copyfileobj)r� r� � targetpathr� r� Z upperdirs�source�targetr r� r r� � s2 & �LzZipFile._extract_memberc C s� |j | jv r(ddl}|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 ��dS ) z6Check for errors before writing a file to the archive.r NzDuplicate name: %rr r� �rm rn ro 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 rj r� r� r, r� r< r rh ru r0 ri �ZIP_FILECOUNT_LIMITrt r� rq r )r� r� r� �requires_zip64r r r r� � s, � �zZipFile._writecheckc C s� | j std��| jr td��tj||| jd�}|�� rDd|_d|_n0|durT||_ n| j |_ |durl||_ 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 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? rm i ) r< r� r r r� ry r� rs rr rh r� ri r r r0 rE rh r] rq r rm r� rv ri r1 rj rB r r� rA r� r� )r� rB r� rh r r� �src�destr r r r � sF ��� .z ZipFile.writec C s" t |t�r|�d�}t |t�sxt|t�t�� �dd� d�}| j|_| j|_ |j d dkrpd|_ | j dO _ q|d |_ n|}| j s�t d ��| jr�t d ��|dur�||_|dur�||_ t|�|_| j�H | j|d d ��}|�|� W d � n1 s�0 Y W d � n1 �s0 Y dS )a Write a file into the archive. The contents is 'data', which may be either a 'str' or a 'bytes' instance; if it is a 'str', it is encoded as UTF-8 first. 'zinfo_or_arcname' is either a ZipInfo instance or the name of the file in the archive.r� Nr" )rB rg r� rz i �Ar) r� r� z?Can't write to ZIP archive while an open writing handle exists.rm )r, )r� rz r� r r� r� r� rh r ri rB rp r< r� r r0 rt r rA r )r� Zzinfo_or_arcnamerX rh r r� r� r r r �writestr s: ��� zZipFile.writestrc C s | � � dS )z2Call the "close()" method in case the user forgot.Nr� r� r r r �__del__/ s zZipFile.__del__c C s� | j du rdS | jr td��zl| jdv rp| jrp| j�, | jrJ| j �| j� | � � W d � n1 sf0 Y W | j }d| _ | � |� n| j }d| _ | � |� 0 dS )zOClose the file, and for mode 'w', 'x' and 'a' write the ending records.NzvCan't close the ZIP file while there is an open writing handle on it. Close the writing handle before closing the zip.r� ) r< r r� r, rv r r0 rE rh �_write_end_recordr� r r r r r 3 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<tt| �} t| |j�} t| |j�} |�� \} }t � tt| |j| |j||j |||j||t | �t | �t |j�d|j |j |�}| j � |� | j � | � | j � | � | j � |j� q| j � � }t | j �}|| j! }| j!}d }|t"k�rd}n |tk�r&d}n|tk�r4d}|�r�| j#�sNt$|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.dk�r| j �/� | j �0� d S )Nr ry r$ r r! r r r& r r� )r r, �Qr r� zCentral directory offsetzCentral directory sizer� r r r ro )1ri rg rt r� rs r1 rq r3 r8 rF r� r0 r� rh r rb r� r r� rk rZ r� r� r� rj rl rr re ro rp r< r r] rh r� ru r rL rK rI rG rH rR r` r_ rx r, �truncater� )r� r� r� r� r� r3 rt rs rq Z extra_datar� rk rZ rB rm r� Zpos2Z centDirCountZ centDirSizeZ centDirOffsetr� Z zip64endrecZ zip64locrecrW r r r r� I s� $$ � ��� � �� � � zZipFile._write_end_recordc C s4 | j dksJ �| j d8 _ | j s0| js0|�� d S )Nr r )r} r{ r r r r r r� � s zZipFile._fpclose)N)N)rl N)F)NN)NNN)NNN)NN)$r r r r r< r� r r� r� r� r� r� r� r� r� r� r� r� rk re �setterr> rA r� r� r� r� r� r� r� r r� r� r r� r� r r r r r � sP �� ZQ f ) * � 2 � )^r c @ s4 e Zd ZdZdeddfdd�Zd d d �Zd d � ZdS )r zDClass to create ZIP archives with Python library files and packages.rl 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 �|�}|rD||�sD| jr@t j�|�r,dnd}td||f � dS t j�|�\}}t j�|��rht j�|d�}t j�|��r�|r�d||f }n|}| jr�td|d|� | � |d d � |�\}} | 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�| � | d d � |�\}} | j�r�td | � | � || � q�n�| j�r�td|� t t � |��D ]�} t j�|| �} t j�| �\} }|d k�r�|�r,|| ��s,| j�r�td| � �q�| � | d d � |�\}} | j�rVtd | � | � || � �q�nP|d d� d k�r�td��| � |d d � |�\}} | j�r�td| � | � || � 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 in�asr ���ZAdding)� filterfunc�.pyz file %r skipped by filterfunczAdding files from directoryz.Files added with writepy() must end with ".py"z Adding file)r~ r� rw r� r� r� r� r2 �isfile� _get_codenamer r� �listdir�remove�splitext�writepyr� )r� �pathname�basenamer� Zlabel�dirr- Zinitnamer� r� �dirlistrB r� �root�extr 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 )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 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_compilerw r� �compile�PyCompileError�msg)r r� r �errr� r r �_compile s z)PyZipFile._get_codename.<locals>._compiler� z.pycr� )� optimizationr r r z"invalid value for 'optimize': {!r})r� r� )r� )� importlib�util�cache_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� sj � � �� � 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| �dd�S )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)� itertools�islice� _ancestry)r� r r r �_parents\ s r c c s2 | � tj�} | � tj�r.| V t�| �\} }q 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 o s r c C s t �t|�j| �S )zZ Return items in minuend not in subtrahend, retaining order with O(1) lookup. )r � filterfalse�set� __contains__)ZminuendZ subtrahendr r r � _difference� s r c sH e Zd ZdZedd� �Z� fdd�Zdd� Zdd � Ze d d � �Z � Z S ) � CompleteDirszk A ZipFile subclass that ensures that implied directories are always included in the namelist. c 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.<locals>.<genexpr>)r �chain� from_iterabler� r �_deduper )�names�parentsZas_dirsr r r � _implied_dirs� s zCompleteDirs._implied_dirsc s t t| ��� }|t| �|�� S r� )rT r r� ra r )r� r rU 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 )zx If the name represents a directory, return that name as a directory (with the trailing slash). rz )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 )zl Given a source (filename or zipfile), return an appropriate CompleteDirs subclass. rl )r� r r r, �__new__�vars�update)r� r� �resr r r �make� s zCompleteDirs.make) r r r r � staticmethodr r� r! r" r� r' rZ r r rU r r � s r c s, e Zd ZdZ� fdd�Z� fdd�Z� ZS )� FastLookupzV ZipFile subclass to ensure implicit dirs exist and are resolved rapidly. c sF t �t�� | jW d � S 1 s&0 Y tt| ��� | _| jS r� )� contextlib�suppressr5 Z_FastLookup__namesrT r) r� r� rU r r r� � s $zFastLookup.namelistc sF t �t�� | jW d � S 1 s&0 Y tt| ��� | _| jS r� )r* r+ r5 Z_FastLookup__lookuprT r) r! r� rU r r r! � s $zFastLookup._name_set)r r r r r� r! rZ r r rU r r) � s r) c @ s� e Zd ZdZdZd$dd�Zd%dd�Zed d � �Zd d � Z d d� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd � Zd d � Zd d � ZeZed!d"� �Zd#S )&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 = 'abcde.zip' Path accepts the zipfile object itself or a filename >>> root = Path(zf) From there, several path operations are available. Directory iteration (including the zip file itself): >>> a, b = root.iterdir() >>> a Path('abcde.zip', 'a.txt') >>> b Path('abcde.zip', 'b/') name property: >>> b.name 'b' join with divide operator: >>> c = b / 'c.txt' >>> c Path('abcde.zip', 'b/c.txt') >>> c.name 'c.txt' Read text: >>> c.read_text() 'content of c' existence: >>> c.exists() True >>> (b / 'missing.txt').exists() False Coercion to string: >>> str(c) 'abcde.zip/b/c.txt' z>{self.__class__.__name__}({self.root.filename!r}, {self.at!r})r� c C s t �|�| _|| _d S r� )r) r' r �at)r� r r, r r r r� s z Path.__init__rl c O s\ |� dd�}|d }| jj| j||d�}d|v rD|s8|r@td��|S tj|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� Nr r� �bz*encoding args invalid for binary operation)�popr rA r, r� r? � TextIOWrapper)r� r, �args�kwargsr� Zzip_mode�streamr r r rA ! s z Path.openc C s t �| j�d��S �Nrz )r r� r, r� r� r r r r- 0 s z Path.namec O sD | j dg|�R i |���}|�� W d � S 1 s60 Y d S )Nrl r� )r� r0 r1 �strmr r r � read_text4 s zPath.read_textc C s6 | � d��}|�� W d � S 1 s(0 Y d S )Nr? r� )r� r4 r r r � read_bytes8 s zPath.read_bytesc C s t �|j�d��| j�d�kS r3 )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 r3 )r, �endswithr� r r r r� B s z Path.is_dirc C s | � � S r� )r� r� r r r �is_fileE s z Path.is_filec C s | j | j�� v S r� )r, r r! r� r r r r� H s z Path.existsc C s. | � � std��t| j| j�� �}t| j|�S )NzCan't listdir a file)r� r� r� r8 r r� �filterr7 )r� Zsubsr r r �iterdirK s z Path.iterdirc C s t �| jj| j�S r� )r r2 r rB r, r� r r r �__str__Q s z Path.__str__c C s | j j| d�S )Nr� )� _Path__reprr r� r r r r� T s z Path.__repr__c C s t �| j|�}| �| j�|��S r� )r r2 r, r8 r r" )r� �add�nextr r r �joinpathW s z Path.joinpathc C s( t �| j�d��}|r |d7 }| �|�S r3 )r r� r, r� r8 )r� Z parent_atr r r �parent] s z Path.parentN)r� )rl )r r r r r> r� rA rk r- r5 r6 r7 r8 r� r: r� r< r= r� rA � __truediv__rB r r r r r � s( @ 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)�requiredz-lz--list� <zipfile>zShow listing of a zipfile)�metavar�helpz-ez --extractr )rF z <output_dir>z Extract zipfile into target dir)�nargsrG rH z-cz--create�+)z<name>z<file>zCreate zipfile from sourcesz-tz--testzTest if a zipfile is validrl 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� ) r~ r� r� r r r� r� r� r2 )rc r� �zippathZnm��addToZipr r rM � s �zmain.<locals>.addToZiprm r� )�argparse�ArgumentParser� add_mutually_exclusive_group� add_argument� parse_args�testr r� r� r ra r� r� r� Zcreater. r~ r� r� r� r� r� ) r0 rN rD �parser�groupr� rc Zbadfiler� Zzip_name�filesr� rK r rL r �maine s\ � � � � & * , rW �__main__)N)N)�r Zbinascii�importlib.utilr r? r r~ r r� r� rF r� r~ 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^ rM rN rO rP rQ rR rS rd 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_CENTDIRrg �Structr. r8 r= r r[ r9 �objectr r� r� r� r� r� r� r r r r r r@ r r[ r r r r �dict�fromkeysr r r r) r rW r r r r r �<module> sR << ^& � ( KU | /2 =