File: //usr/local/lib/python3.10/dist-packages/wheel/__pycache__/wheelfile.cpython-310.pyc
o
��h� � @ 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 �:/usr/local/lib/python3.10/dist-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, �
allowZip64z {}.dist-info�nameverz/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�match�parsed_filename�endswithr r �__init__�format�group�dist_info_path� record_path� _file_hashes� _file_sizes�open�KeyError�csv�readerr �split�hashlib�new�
ValueError�lowerr
�encode) �selfr( r* r, r9 �record�liner8 �hash_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�bytesrZ �Nonec s@ � j }�| � ��| � |r ��� �kr td�� d���d S d S )NzHash mismatch for file '�')�_eof�update�digestr )r\ �eof��ef�ef_name�
expected_hash� running_hash�update_crc_origr r � _update_crcs s
�z#WheelFile.open.<locals>._update_crcr' �/zNo hash found for file 'r_ )r\ r] rZ r^ ) �
isinstancer
�filenamer= rC r r rE rJ rK rj )rO rU r* rX rj rT r rd r rE 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 itrk z
.dist-info)r �inform r �walk�sort�sortedr8 �normpath�join�isfile�relpath�replace�seprB r= �append�write) rO rn �deferred�root�dirnames� filenames�namer8 �arcnamer r r � write_files� s&
�� �zWheelFile.write_filesrm r� �
str | None�
compress_type�
int | Noner^ 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�� date_time� )rE r �fstat�fileno�readr
r% �st_mtime�stat�S_IMODE�st_mode�S_IFMT�
external_attrr, r� �writestr)rO rm r� r� �f�st�data�zinfor r r r{ � 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� r1 zadding 'r_ r7 )rl ro r
r% r, r� r� �S_IFREGr� rN r r� rm r rp rB �_default_algorithmr� r rb �decoderC �lenrD )rO 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�,�"�
)� delimiter� quotechar�lineterminatorc 3 s0 � | ]\}\}}||d | � j | fV qdS )r4 N)rD )�.0r� rT r� �rO r r � <genexpr>� s
�
�
�z"WheelFile.close.<locals>.<genexpr>r0 )�fpr* rC r rG �writer� writerows�items�writerowr? rB r� �getvaluer �close)rO r� r� r r� r r� � s
��zWheelFile.close)r( r) r* r+ r, r )r' N)rU rV r* rW rX rY rZ r[ )rn ro r/ )rm ro r� r� r� r� rZ r^ �N)r� rV r� r� r� r� )r r r �__doc__rJ �sha256r� r r> rE r� r{ r� r� r r r r r& , s �= �
" � � r& r� )r r ))�
__future__r rG rJ �os.pathr �rer� r! �ior r �typingr r r �zipfiler r r
� wheel.clir �
wheel.utilr r
r r r r �typing_extensionsr ro �PathLiker) r �compile�VERBOSEr: r# r% r& r r r r �<module> s0 �