File: //opt/saltstack/salt/lib/python3.10/site-packages/wheel/__pycache__/wheelfile.cpython-310.pyc
o
;jH$ � @ s d dl mZ g d�Zd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl m
Z
mZ d dlmZmZmZ d dlmZmZmZ erPd dlmZmZ e �de j�Z d Z e� d
�Z G d d � d e �Z!ddd�Z"ddd�Z# d d dd�Z$G dd� de�Z%dS ) � )� annotations)�
WHEEL_INFO_RE� WheelFile�
WheelErrorN)�StringIO�
TextIOWrapper)�IO�
TYPE_CHECKING�Literal)� ZIP_DEFLATED�ZipFile�ZipInfo)� SizedBuffer�StrPathz�^(?P<namever>(?P<name>[^\s-]+?)-(?P<ver>[^\s-]+?))(-(?P<build>\d[^\s-]*))?
-(?P<pyver>[^\s-]+?)-(?P<abi>[^\s-]+?)-(?P<plat>\S+)\.whl$i ��Zwheelc @ s e Zd ZdS )r N)�__name__�
__module__� __qualname__� r r �C/opt/saltstack/salt/lib/python3.10/site-packages/wheel/wheelfile.pyr ! s r �data�bytes�returnc C s t �| ��d�S )z!urlsafe_b64encode without padding� =)�base64�urlsafe_b64encode�rstrip)r r r r r % s r c C s" ddt | �d@ }t�| | �S )z!urlsafe_b64decode without paddingr � � )�lenr �urlsafe_b64decode)r �padr r r r * s r � timestamp� float | None� tuple[int, int, int, int, int]c 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_datetime0 s
r. c @ sh e Zd ZdZejZdefd(d d
�Z d)d*dd�Z d+dd�Z
d,d-d d �Z d.d/d$d%�Z d0d&d'�Z
d S )1r z�A ZipFile derivative class that also reads SHA-256 hashes from
.dist-info/RECORD and checks any read files against those.
�r�filer �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�| j
| �� vrj| j �� d }| �� D ]}|�� |kri|�dd
�d | _ || _
nqSd | j| j
< d | j| j
d
< d | j| j
d < z| �| j
�}W n ty� 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)r3 Z
allowZip64z {}.dist-infoZnameverz/RECORDr/ z/record� r �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�basenamer �matchZparsed_filename�endswithr r �__init__�format�groupZdist_info_path� record_path� _file_hashes� _file_sizesZnamelist�lower�rsplit�open�KeyError�csv�readerr �split�hashlib�new�
ValueErrorr �encode)
�selfr0 r1 r3 r? Zlowered�name�record�liner> Zhash_sum�size� algorithmr r r rB A sn
� � � �
���
� ��"��zWheelFile.__init__N� name_or_info�
str | ZipInfo�Literal['r', 'w']�pwd� bytes | Noner � 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�newdatar r �Nonec s@ � j }�| � ��| � |r ��� �kr td�� d���d S d S )NzHash mismatch for file '�')�_eof�update�digestr )r_ �eof�ZefZef_nameZ
expected_hashZ running_hashZupdate_crc_origr r � _update_crc� s
�z#WheelFile.open.<locals>._update_crcr/ �/zNo hash found for file 'ra )r_ r r r` ) �
isinstancer
�filenamerA rF r r rJ rO rP rg )rS rY r1 r\ rg rX r rf r rJ � s ��
zWheelFile.open�base_dir�strr` c C s� t �d| j|� g }t�|�D ]G\}}}|�� t|�D ]9}tj�tj� ||��}tj�
|�rUtj� ||�� tjj
d�}|| jkrBq |�d�rO|�||f� q | �||� q q|�� |D ]
\}}| �||� q]d S )Nz creating %r and adding %r to itrh z
.dist-info)�log�inforj r' �walk�sort�sortedr> �normpath�join�isfile�relpath�replace�seprE rA �append�write) rS rk Zdeferred�root�dirnames� filenamesrT r> �arcnamer r r � write_files� s&
�� �zWheelFile.write_filesrj r} �
str | None�
compress_type�
int | Nonec 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� )rJ r' �fstat�fileno�readr
r. �st_mtime�stat�S_IMODE�st_mode�S_IFMT�
external_attrr3 r� �writestr)rS rj r} r� �f�str Zzinfor r r ry � 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|� || jkr\| �|�}|jt|�� ��d�f| j|< t|�| j|< d S d S )Nr� i� r� r7 z adding %rr= )ri rl r
r. r3 r� r� �S_IFREGr� rR r r� rj rm rn rE �_default_algorithmrT r rd �decoderF r rG )rS r� r r� �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 )r: N)rG )�.0r� rX r� �rS r r � <genexpr>� s
�
�
�z"WheelFile.close.<locals>.<genexpr>r6 )�fpr1 rF r rL �writerZ writerows�itemsZwriterowrC rE r� �getvaluer �close)rS r r� r r� r r� � s
��zWheelFile.close)r0 r r1 r2 r3 r& )r/ N)rY rZ r1 r[ r\ r] r r^ )rk rl r r` r5 )rj rl r} r r� r� r r` �N)r� rZ r r� r� r� r r` )r r` )r r r �__doc__rO �sha256r� r rB rJ r~ ry r� r� r r r r r : s �H �
" � � r )r r r r r� )r! r" r r# )&�
__future__r �__all__r rL rO �loggingZos.pathr' �rer� r* �ior r �typingr r r
�zipfiler r r
Z _typeshedr r �compile�VERBOSEr r, � getLoggerrm � Exceptionr r r r. r r r r r �<module> s6 �
�