← Back
Editing: zipstream.cpython-38.pyc
U W[ & � @ s� d Z ddlZddlZddlZddlZe�ej�ZG dd� dej �Z G dd� de �Z G dd� de �Z G d d � d e �Zd Zd d � Zdd� Zddd�ZdS )z� An incremental approach to unzipping files. This allows you to unzip a little bit of a file at a time, which means you can report progress as a file unzips. � Nc @ s e Zd ZdZdd� ZdS )�ChunkingZipFilez~ A L{zipfile.ZipFile} object which, with L{readfile}, also gives you access to a file-like object for each entry. c C s& | j dkrtd��| js td��| �|�}| j�|jd� | j�t�}|dd� tj krbt� d��t � tj |�}| j�|tj �}|tj r�| j�|tj � |jd@ r�|�d�}n |�d �}||jkr�t� d |j|f ��|jtjkr�t| |j�S |jtjk�rt| |j�S t� d |j|f ��d S ) z3 Return file-like object for name. )�r�az read() requires mode "r" or "a"z3Attempt to read ZIP archive that was already closedr � z Bad magic number for file headeri zutf-8�cp437z3File name in directory "%s" and header "%s" differ.z-Unsupported compression method %d for file %sN)�mode� RuntimeError�fpZgetinfo�seek� header_offset�read�_fileHeaderSize�zipfileZstringFileHeaderZ BadZipfile�structZunpack�structFileHeaderZ_FH_FILENAME_LENGTHZ_FH_EXTRA_FIELD_LENGTHZ flag_bits�decodeZ orig_filenameZ compress_typeZ ZIP_STORED� ZipFileEntryZ compress_sizeZ ZIP_DEFLATED�DeflatedZipFileEntry)�self�nameZzinfoZfheader�fnameZ fname_str� r �:/usr/lib/python3/dist-packages/twisted/python/zipstream.py�readfile sD � �� ��zChunkingZipFile.readfileN)�__name__� __module__� __qualname__�__doc__r r r r r r s r c @ sd e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � ZeZ d d � Z dd� Z dd� Z dd� Z dd� ZdS )� _FileEntrya! Abstract superclass of both compressed and uncompressed variants of file-like objects within a zip archive. @ivar chunkingZipFile: a chunking zip file. @type chunkingZipFile: L{ChunkingZipFile} @ivar length: The number of bytes within the zip file that represent this file. (This is the size on disk, not the number of decompressed bytes which will result from reading it.) @ivar fp: the underlying file object (that contains pkzip data). Do not touch this, please. It will quite likely move or go away. @ivar closed: File-like 'closed' attribute; True before this file has been closed, False after. @type closed: L{bool} @ivar finished: An older, broken synonym for 'closed'. Do not touch this, please. @type finished: L{int} c C s& || _ | j j| _|| _d| _d| _dS )zC Create a L{_FileEntry} from a L{ChunkingZipFile}. r FN)�chunkingZipFiler �length�finished�closed�r r r r r r �__init__] s z_FileEntry.__init__c C s dS )zD Returns false because zip files should not be ttys Fr �r r r r �isattyh s z_FileEntry.isattyc C s d| _ d| _| `dS )z/ Close self (file-like object) T� N)r" r! r r% r r r �closeo s z_FileEntry.closec s4 d}t � fdd�d�D ]}||7 }|dkr q0q|S )z Read a line. � c s � � d�S )Nr' )r r r% r r �<lambda>} r) z%_FileEntry.readline.<locals>.<lambda>� )�iter)r �lineZbyter r% r �readlinex s z_FileEntry.readlinec C s | � � }|r|S t� �dS )zi Implement next as file does (like readline, except raises StopIteration at EOF) N)r. � StopIteration)r Znextliner r r �__next__� s z_FileEntry.__next__c C s t | �S )z1 Returns a list of all the lines )�listr% r r r � readlines� s z_FileEntry.readlinesc C s | S �z/ Returns an iterator (so self) r r% r r r � xreadlines� s z_FileEntry.xreadlinesc C s | S r3 r r% r r r �__iter__� s z_FileEntry.__iter__c C s | S �Nr r% r r r � __enter__� s z_FileEntry.__enter__c C s | � � d S r6 )r( )r �exc_type� exc_value� tracebackr r r �__exit__� s z_FileEntry.__exit__N)r r r r r$ r& r( r. r0 �nextr2 r4 r5 r7 r; r r r r r F s r c @ s* e Zd ZdZdd� Zdd� Zd dd�ZdS ) r zJ File-like object used to read an uncompressed entry in a ZipFile c C s t �| ||� d| _d S )Nr )r r$ � readBytesr# r r r r$ � s zZipFileEntry.__init__c C s | j S r6 )r= r% r r r �tell� s zZipFileEntry.tellNc C sv |d kr| j | j }|dks"| jr&dS | jj�t|| j | j ��}| jt|�7 _| j| j kslt|�|k rrd| _|S )Nr r) r' )r r= r! r r r �min�len)r �n�datar r r r � s �zZipFileEntry.read)N�r r r r r$ r> r r r r r r � s r c @ s* e Zd ZdZdd� Zdd� Zd dd�ZdS ) r zE File-like object used to read a deflated entry in a ZipFile c C s0 t �| ||� d| _d| _t�d�| _d| _d S )Nr i�r) )r r$ � returnedBytesr= �zlibZ decompressobj�decomp�bufferr# r r r r$ � s z DeflatedZipFileEntry.__init__c C s | j S r6 )rD r% r r r r> � s zDeflatedZipFileEntry.tellNc C s` | j r dS |d kr�| jg}|�| j�| jj�| j| j ��� |�| j�d�� |�| j� � � d| _d| _ d� |�}| j t |�7 _ |S t | j�|k �r(| jj�t|d| j| j ��}| j t |�7 _ |�s| j| j�d� | j� � }d| _ d| _| j t |�7 _ |S | j| j�|�7 _q�| jd |� }| j|d � | _| j t |�7 _ |S d S )Nr) � Zr' i )r! rG �appendrF � decompressr r r r r= �flush�joinrD r@ r? )r rA �resultrB r r r r � sL ��� � ��zDeflatedZipFileEntry.read)NrC r r r r r � s r � c C s, d}t | �}|�� D ]}|t||�7 }q|S )zr Predict the number of chunks that will be extracted from the entire zipfile, given chunksize blocks. r )r �infolist�countFileChunks)�filename� chunksizeZ totalchunksZzf�infor r r �countZipFileChunks s rT c C s( t | j|�\}}|dkr |d7 }|p&dS )a9 Count the number of chunks that will result from the given C{ZipInfo}. @param zipinfo: a C{zipfile.ZipInfo} instance describing an entry in a zip archive to be counted. @return: the number of chunks present in the zip file. (Even an empty file counts as one chunk.) @rtype: L{int} r r' )�divmod� file_size)ZzipinforR �countZextrar r r rP s rP �.� c c sP t | d�}tj�|�s t�|� t| |�}|�� }|�� }t||�D �]\}} | j t @ } tj� ||�} | r�tj�| �s�t�| � |d8 }|V qDtj� | �d } tj�| �s�t�| � |s�tj�| ��s6|� |�} | jdkr�|d8 }|V t| d��:}| �� | jk �r*| �|�}|�|� |d8 }|V q�W 5 Q R X qD|t| |�8 }|V qDdS )z� Return a generator for the zipfile. This implementation will yield after every chunksize uncompressed bytes, or at the end of a file, whichever comes first. The value it yields is the number of chunks left to unzip. r r' r �wbN)r �os�path�exists�makedirsrT ZnamelistrO �zipZ external_attr�DIR_BITrL �splitr rV �openr> r �writerP )rQ Z directoryZ overwriterR ZczfZ remaining�namesZinfos�entryrS �isdir�fZfdirr ZoutfileZhunkr r r �unzipIterChunky$ s< rh )rX r rY )r r �os.pathr[ rE r Zcalcsizer r ZZipFiler �objectr r r r` rT rP rh r r r r �<module> s 4j 7 �
Save File
Cancel