File: //lib/python3/dist-packages/wheel/__pycache__/wheelfile.cpython-38.pyc
U
�a�c� � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dl m
Z
d dl m Z
d dlmZmZmZ d dlmZ d dlmZmZmZmZmZmZ e�dej �Z d
d d
�Z G d d � d e�Z dS )� )�print_functionN)� OrderedDict)�log)� ZIP_DEFLATED�ZipInfo�ZipFile)�
WheelError)�urlsafe_b64decode�
as_unicode�native�urlsafe_b64encode�as_bytes�StringIOz�^(?P<namever>(?P<name>[^-]+?)-(?P<ver>[^-]+?))(-(?P<build>\d[^-]*))?
-(?P<pyver>[^-]+?)-(?P<abi>[^-]+?)-(?P<plat>[^.]+?)\.whl$c C s, t tj�d| pt�� ��} t�| �dd� S )N�SOURCE_DATE_EPOCHr � )�int�os�environ�get�time�gmtime)Z timestamp� r �1/usr/lib/python3/dist-packages/wheel/wheelfile.py�get_zipinfo_datetime s r c @ sR e Zd ZdZejZdefdd�Zddd�Z dd � Z
dd
d �Z dd d
�Z dd� Z
dS )� WheelFilez�A ZipFile derivative class that also reads SHA-256 hashes from
.dist-info/RECORD and checks any read files against those.
�rc
C s� t j�|�}t�|�| _|�d�r,| jd kr:td�|���t j
| |||dd� d�| j� d��| _ | j d | _
t� | _i | _|dk�r�d | j| j
< d | j| j
d
< d | j| j
d < z| �| j
�}W n$ tk
r� td �| j
���Y nX |�� |D ]�}|�d
�}|�dd�\}}} |r�|�d�\}
}zt�|
� W n$ tk
�rV td�|
���Y nX |
�� dk�rttd�|
���|
t|�d��f| j|< q�W 5 Q R X d S )Nz.whlzBad wheel filename {!r}T)� compressionZ
allowZip64z {}.dist-infoZnameverz/RECORDr )NNz.jwsz.p7szMissing {} filezutf-8�,� �=z Unsupported hash algorithm: {}> �sha1�md5z4Weak hash algorithm ({}) is not permitted by PEP 427�ascii) r �path�basename�
WHEEL_INFO_RE�matchZparsed_filename�endswithr �formatr �__init__�groupZdist_info_path� record_pathr � _file_hashes� _file_sizes�open�KeyError�decode�rsplit�split�hashlib�new�
ValueError�lowerr �encode) �self�file�moder r$ �record�liner# Zhash_sum�size� algorithmr r r r) & sH
�� �zWheelFile.__init__Nc s� d� ����fdd� }t �| |||�� tt|t�r8|jn|��|dkr���d�s��| jkrltd� t
�����| j� \}��d k r�t � |��� j
| �� _
� S )Nc sR |d kr� j }�| � n
�| |� ��| � |rN��� �krNtd�t�����d S )NzHash mismatch for file '{}')Z_eof�update�digestr r( r )�newdata�eof�ZefZef_nameZ
expected_hashZ running_hashZupdate_crc_origr r � _update_crcR s
z#WheelFile.open.<locals>._update_crcr �/zNo hash found for file '{}')N)r r. r
�
isinstancer �filenamer' r, r r( r r3 r4 rD )r8 Z name_or_infor: �pwdrD r> r rC r r. Q s �
zWheelFile.openc C s� t �d| j|� g }t�|�D ]�\}}}|�� t|�D ]r}tj�tj� ||��}tj�
|�r8tj� ||�� tjj
d�}|| jkr�q8|�d�r�|�||f� q8| �||� q8q |�� |D ]\}}| �||� q�d S )Nz#creating '%s' and adding '%s' to itrE z
.dist-info)�logger�inforG r �walk�sort�sortedr# �normpath�join�isfile�relpath�replace�sepr+ r' �append�write) r8 Zbase_dirZdeferred�rootZdirnames� filenames�namer# �arcnamer r r � write_filesl s
zWheelFile.write_filesc C s~ t |d�� }t�|�� �}|�� }W 5 Q R X t|p4|t|j�d�}t� |j
�t� |j
�B d> |_ |ph| j
|_| �|||� d S )N�rb�Z date_time� )r. r �fstat�fileno�readr r �st_mtime�stat�S_IMODE�st_mode�S_IFMT�
external_attrr �
compress_type�writestr)r8 rG rY rg �f�st�data�zinfor r r rU � s zWheelFile.writec C sr t �| |||� t|t�r |jn|}t�d|� || jkrn| �|�}|j t
t |� � ��f| j
|< t|�| j|< d S )Nz adding '%s')r rh rF r rG rI rJ r+ �_default_algorithmrX r r r@ r, �lenr- )r8 Zzinfo_or_arcname�bytesrg �fname�hash_r r r rh � s �
zWheelFile.writestrc s� � j d k r�� jdkr�� jr�t� }tj|dddd�}|�� fdd�� j�� D �� |�t � j
�ddf� t t � j
�t
� d �}� j|_d
|_� �|t|�� �� t�� � d S ) N�wr �"�
)Z delimiterZ quotecharZlineterminatorc 3 s. | ]&\}\}}||d | � j | fV qdS )r N)r- )�.0rp r> rq �r8 r r � <genexpr>� s
�
�z"WheelFile.close.<locals>.<genexpr>� r\ i �)�fpr: r, r �csv�writerZ writerows�itemsZwriterowr( r+ r r r r rg rf rh r
�getvaluer �close)r8 rk r{ rl r rv r r~ � s �zWheelFile.close)r N)NN)N)�__name__�
__module__� __qualname__�__doc__r3 Zsha256rm r r) r. rZ rU rh r~ r r r r r s +
r )N) Z
__future__r rz r3 �os.pathr �rerb r � collectionsr Z distutilsr rI Zzipfiler r r Z wheel.clir Z
wheel.utilr r
r r r
r �compile�VERBOSEr% r r r r r r �<module> s" �