File: //usr/local/lib/python3.7/site-packages/wheel/__pycache__/wheelfile.cpython-37.pyc
B
O��g � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dl m
Z
m Z d dl m
Z
mZmZ d dlmZ d dlmZmZmZ e�dej�ZdZd
d d
�ZG d d � d e�ZdS )� )� annotationsN)�StringIO�
TextIOWrapper)� ZIP_DEFLATED�ZipFile�ZipInfo)�
WheelError)�log�urlsafe_b64decode�urlsafe_b64encodez�^(?P<namever>(?P<name>[^\s-]+?)-(?P<ver>[^\s-]+?))(-(?P<build>\d[^\s-]*))?
-(?P<pyver>[^\s-]+?)-(?P<abi>[^\s-]+?)-(?P<plat>\S+)\.whl$i ��c C s6 t tj�d| pt�� ��} t| t�} t�| �dd� S )N�SOURCE_DATE_EPOCHr � )�int�os�environ�get�time�max�MINIMUM_TIMESTAMP�gmtime)� timestamp� r �3/tmp/pip-unpacked-wheel-f1gdtpe6/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|����tj | |||dd� d�
| j� d��| _ | j d | _
i | _i | _|dk�r�d | j| j
< d | j| j
d
< d | j| j
d < y| �| j
�}W n( tk
r� td | j
� d
��d �Y nX |�� x�t�t|ddd��D ]�}|\}}} |�s �q|�d�\}
}yt�|
� W n& tk
�r` td|
� ��d �Y nX |
�� dk�r~td�
|
���|
t|�d��f| j|< �qW W d Q R X d S )Nz.whlzBad wheel filename T)� compression�
allowZip64z {}.dist-infoZnameverz/RECORDr )NNz.jwsz.p7szMissing z file� zutf-8)�newline�encoding�=z Unsupported hash algorithm: > �md5�sha1z4Weak hash algorithm ({}) is not permitted by PEP 427�ascii) r �path�basename�
WHEEL_INFO_RE�matchZparsed_filename�endswithr r �__init__�format�groupZdist_info_path� record_path� _file_hashes� _file_sizes�open�KeyError�csv�readerr �split�hashlib�new�
ValueError�lowerr
�encode) �self�file�moder r& �record�liner% Zhash_sum�size� algorithmr r r r* ( sJ
zWheelFile.__init__Nc s� � ����fdd�}t |t�r$|jn|�|dkrT��d�sT�| jkrTtd�� d���t�| |||�� |dkr���d�s�| j� \}��d k r�t� |��� j
| �� _
� S )Nc s< � j }�| � ��| � |r8��� �kr8td�� d���d S )NzHash mismatch for file '�')�_eof�update�digestr )�newdata�eof)�ef�ef_name�
expected_hash� running_hash�update_crc_origr r � _update_crc^ s
z#WheelFile.open.<locals>._update_crcr �/zNo hash found for file 'rA ) �
isinstancer �filenamer) r. r r r0 r5 r6 rL )r: Z name_or_infor< �pwdrL r@ r )rG rH rI rJ rK r r0 ] s
zWheelFile.openc C s� t �d| j� d|� d�� g }x�t�|�D ]�\}}}|�� x~t|�D ]r}tj�tj� ||��}tj�
|�rFtj� ||�� tjj
d�}|| jkr�qF|�d�r�|�||f� qF| �||� qFW q*W |�� x |D ]\}}| �||� q�W d S )Nz
creating 'z' and adding 'z' to itrM z
.dist-info)r �inforO r �walk�sort�sortedr% �normpath�join�isfile�relpath�replace�sepr- r) �append�write) r: �base_dir�deferred�root�dirnames� filenames�namer% �arcnamer r r � write_filesz s
zWheelFile.write_filesc C s~ t |d�� }t�|�� �}|�� }W d Q R X t|p4|t|j�d�}t� |j
�t� |j
�B d> |_ |ph| j
|_| �|||� d S )N�rb)� date_time� )r0 r �fstat�fileno�readr r �st_mtime�stat�S_IMODE�st_mode�S_IFMT�
external_attrr �
compress_type�writestr)r: rO rc rq �f�st�data�zinfor r r r\ � s zWheelFile.writec C s� t |t�r0t|t� d�}| j|_dtjB d> |_t |t�rD|� d�}t
� | |||� t |t�rd|j n|}t
�d|� d�� || jkr�| �|�}|jt|�� ��d�f| j|< t|�| j|< d S )N)rf i� rg zutf-8zadding 'rA r$ )rN �strr r r rq rl �S_IFREGrp r9 r rr rO r rQ r- �_default_algorithmrb r rD �decoder. �lenr/ )r: �zinfo_or_arcnameru rq �fname�hash_r r r rr � s
zWheelFile.writestrc s� � j d k rx� jdkrx� jrxt� }tj|dddd�}|�� fdd�� j�� D �� |�t � j
�ddf� � � � j
|� � � t
�� � d S ) N�w�,�"�
)� delimiter� quotechar�lineterminatorc 3 s. | ]&\}\}}||d | � j | fV qdS )r! N)r/ )�.0r} r@ r~ )r: r r � <genexpr>� s z"WheelFile.close.<locals>.<genexpr>r )�fpr< r. r r2 �writer� writerows�items�writerowr+ r- rr �getvaluer �close)r: ru r� r )r: r r� � s
zWheelFile.close)r N)NN)N)�__name__�
__module__� __qualname__�__doc__r5 �sha256ry r r* r0 rd r\ rr r� r r r r r ! s 5
r )N) �
__future__r r2 r5 �os.pathr �rerl r �ior r �zipfiler r r Z wheel.clir Z
wheel.utilr r
r �compile�VERBOSEr' r r r r r r r �<module> s