File: //opt/saltstack/salt/lib/python3.10/site-packages/wheel/__pycache__/wheelfile.cpython-310.pyc
o
�N�g� � @ s d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dl m
Z
m Z d dl m
Z
mZmZ d dlmZmZmZ d dlmZ d dlmZmZmZ erod dl mZmZm Z d d l m Z e e ej e f Z!G d
d � d ee e�Z"e�#d ej$�Z%d
Z&dddd�Z'G dd� de�Z(dS )� )� annotationsN)�StringIO�
TextIOWrapper)�IO�
TYPE_CHECKING�Literal)� ZIP_DEFLATED�ZipFile�ZipInfo)�
WheelError)�log�urlsafe_b64decode�urlsafe_b64encode)�Protocol�Sized�Union)�Bufferc @ s e Zd ZdS )� SizedBufferN)�__name__�
__module__� __qualname__� r r �C/opt/saltstack/salt/lib/python3.10/site-packages/wheel/wheelfile.pyr s r z�^(?P<namever>(?P<name>[^\s-]+?)-(?P<ver>[^\s-]+?))(-(?P<build>\d[^\s-]*))?
-(?P<pyver>[^\s-]+?)-(?P<abi>[^\s-]+?)-(?P<plat>\S+)\.whl$i ��� timestamp� float | Nonec C s6 t tj�d| p
t�� ��} t| t�} t�| �dd� S )N�SOURCE_DATE_EPOCHr � )�int�os�environ�get�time�max�MINIMUM_TIMESTAMP�gmtime)r r r r �get_zipinfo_datetime$ s
r% c @ sf e Zd ZdZejZdefd(d d
�Z d)d*dd�Z d+dd�Z
d,d-d d �Z d.d/d$d%�Z d&d'� Z
d S )0� WheelFilez�A ZipFile derivative class that also reads SHA-256 hashes from
.dist-info/RECORD and checks any read files against those.
�r�file�StrPath�mode�Literal['r', 'w', 'x', 'a']� compressionr c
C s� t j�|�}t�|�| _|�d�r| jd u r td|����tj | |||dd� d�
| j� d��| _ | j d | _
i | _i | _|dkr�d | j| j
< d | j| j
d
< d | j| j
d < z| �| j
�}W n typ td | j
� d
��d �w |�V t�t|ddd��D ]B}|\}}} |s�q~|�d�\}
}zt�|
� W n ty� td|
� ��d �w |
�� dv r�td|
� d���|
t|�d��f| j|< q~W d � d S 1 s�w Y d S d S )Nz.whlzBad wheel filename T)r, Z
allowZip64z {}.dist-infoZnameverz/RECORDr' �NNz.jwsz.p7szMissing z file� �utf-8)�newline�encoding�=z Unsupported hash algorithm: > �md5�sha1zWeak hash algorithm (z ) 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) �selfr( r* r, r7 �record�liner6 Zhash_sum�size� algorithmr r r r; 3 s^
� � �
���
� ��"��zWheelFile.__init__N� name_or_info�
str | ZipInfo�Literal['r', 'w']�pwd� bytes | None�return� IO[bytes]c s� d � ����fdd� }t |t�r|jn|�|dkr+��d�s+�| jvr+td �� d
���t�| |||�� |dkrS��d�sS| j� \}��d urSt� |��� j
|�� _
� S ) N�newdata�bytesrU �Nonec s@ � j }�| � ��| � |r ��� �kr td�� d���d S d S )NzHash mismatch for file '�')�_eof�update�digestr )rW �eof�ZefZef_nameZ
expected_hashZ running_hashZupdate_crc_origr r � _update_crcs s
�z#WheelFile.open.<locals>._update_crcr' �/zNo hash found for file 'rZ )rW rX rU rY ) �
isinstancer
�filenamer: r? r r rA rF rG r` )rK rP r* rS r` rO r r_ r rA m s ��
zWheelFile.open�base_dir�strc C s� t �d| j� d|� d�� g }t�|�D ]G\}}}|�� t|�D ]9}tj�tj� ||��}tj�
|�rZtj� ||�� tjj
d�}|| jkrGq!|�d�rT|�||f� q!| �||� q!q|�� |D ]
\}}| �||� qbd S )Nz
creating 'z' and adding 'z' to itra z
.dist-info)r �inforc r �walk�sort�sortedr6 �normpath�join�isfile�relpath�replace�sepr> r: �append�write) rK rd Zdeferred�root�dirnames� filenames�namer6 �arcnamer r r � write_files� s&
�� �zWheelFile.write_filesrc rv �
str | None�
compress_type�
int | NonerY c C s� t |d��}t�|�� �}|�� }W d � n1 sw Y t|p$|t|j�d�}t� |j
�t� |j
�B d> |_ |p>| j
|_| �|||� d S )N�rb�Z date_time� )rA r �fstat�fileno�readr
r% �st_mtime�stat�S_IMODE�st_mode�S_IFMT�
external_attrr, ry �writestr)rK rc rv ry �f�st�dataZzinfor r r rq � s
�� zWheelFile.write�zinfo_or_arcnamer� �SizedBuffer | strc C s� t |t�rt|t� d�}| j|_dtjB d> |_t |t�r"|� d�}t
� | |||� t |t�r2|j n|}t
�d|� d�� || jkr_| �|�}|jt|�� ��d�f| j|< t|�| j|< d S d S )Nr| i� r} r/ zadding 'rZ r5 )rb re r
r% r, ry r� �S_IFREGr� rJ r r� rc r rf r> �_default_algorithmru r r] �decoder? �lenr@ )rK r� r� ry �fname�hash_r r r r� � s*
�
��
��zWheelFile.writestrc s� � j d ur<� jdkr<� jr<t� }tj|dddd�}|�� fdd�� j�� D �� |�t � j
�ddf� � � � j
|� � � t
�� � d S ) N�w�,�"�
)� delimiterZ quotecharZlineterminatorc 3 s0 � | ]\}\}}||d | � j | fV qdS )r2 N)r@ )�.0r� rO r� �rK r r � <genexpr>� s
�
�
�z"WheelFile.close.<locals>.<genexpr>r. )�fpr* r? r rC �writerZ writerows�itemsZwriterowr< r> r� �getvaluer �close)rK r� r� r r� r r� � s
��zWheelFile.close)r( r) r* r+ r, r )r' N)rP rQ r* rR rS rT rU rV )rd re r- )rc re rv rx ry rz rU rY �N)r� rQ r� r� ry rz )r r r �__doc__rF �sha256r� r r; rA rw rq r� r� r r r r r& , s �= �
" � � r& r� )r r ))Z
__future__r rC rF Zos.pathr �rer� r! �ior r �typingr r r �zipfiler r r
Z wheel.clir Z
wheel.utilr r
r r r r Ztyping_extensionsr re �PathLiker) r �compile�VERBOSEr8 r# r% r& r r r r �<module> s0 �