File: //usr/lib/python3/dist-packages/pip/_internal/utils/__pycache__/hashes.cpython-310.pyc
o
@%Ne� � @ s� d dl Z d dlmZmZmZmZmZ d dlmZm Z m
Z
d dl m Z er0d dl m
Z
d dlmZ dZg d�ZG d d
� d
�ZG d d � d e�ZdS )
� N)�
TYPE_CHECKING�BinaryIO�Dict�Iterator�List)� HashMismatch� HashMissing�InstallationError)� read_chunks)�_Hash)�NoReturn�sha256)r
�sha384�sha512c @ s� e Zd ZdZd$deeee f ddfdd�Zd%dd �Ze de
fd
d ��Z d ed
ede fdd�Z
dee ddfdd�Zdeedf ddfdd�Zdeddfdd�Zdeddfd d �Zde fd d �Zdede fd d!�Zde
fd"d#�ZdS )&�HasheszaA wrapper that builds multiple hashes at once and checks them against
known-good values
N�hashes�returnc C s4 i }|dur|� � D ]
\}}t|�||< q
|| _dS )zo
:param hashes: A dict of algorithm names pointing to lists of allowed
hex digests
N)�items�sorted�_allowed)�selfr �allowed�alg�keys� r �</usr/lib/python3/dist-packages/pip/_internal/utils/hashes.py�__init__ s
zHashes.__init__�otherc sb t |t�stS |s �S �s|S i }|j�� D ]\� }� �jvr q� �fdd�|D �|� < qt|�S )Nc s g | ] }|�j � v r|�qS r )r )�.0�v�r r r r �
<listcomp>; s z"Hashes.__and__.<locals>.<listcomp>)�
isinstancer �NotImplementedr r )r r �new�valuesr r r �__and__+ s
zHashes.__and__c C s t dd� | j�� D ��S )Nc s s � | ]}t |�V qd S �N)�len)r �digestsr r r � <genexpr>@ s � z&Hashes.digest_count.<locals>.<genexpr>)�sumr r% �r r r r � digest_count> s zHashes.digest_count� hash_name�
hex_digestc C s || j �|g �v S )z/Return whether the given hex digest is allowed.)r �get)r r. r/ r r r �is_hash_allowedB s zHashes.is_hash_allowed�chunksc
C s� i }| j �� D ]}z t�|�||< W q ttfy" td|� ���w |D ]}|�� D ]}|�|� q+q%|� � D ]\}}|�
� | j | v rH dS q8| � |� dS )z�Check good hashes against ones built from iterable of chunks of
data.
Raise HashMismatch if none match.
zUnknown hash name: N) r r �hashlibr$ �
ValueError� TypeErrorr r% �updater � hexdigest�_raise)r r2 �gotsr. �chunk�hash�gotr r r �check_against_chunksF s � ��zHashes.check_against_chunksr9 r r c C s t | j|��r' )r r �r r9 r r r r8 ] s z
Hashes._raise�filec C s | � t|��S )zaCheck good hashes against a file-like object
Raise HashMismatch if none match.
)r= r
)r r? r r r �check_against_file` s zHashes.check_against_file�pathc C s8 t |d��
}| �|�W d � S 1 sw Y d S )N�rb)�openr@ )r rA r? r r r �check_against_pathh s $�zHashes.check_against_pathc C s
t | j�S )z,Return whether I know any known-good hashes.)�boolr r, r r r �__bool__l s
zHashes.__bool__c C s t |t�stS | j|jkS r' )r" r r# r )r r r r r �__eq__p s
z
Hashes.__eq__c C s" t d�tdd� | j�� D ����S )N�,c s s, � | ]\}}|D ]
}d � ||f�V qqdS )�:N)�join)r r � digest_list�digestr r r r* x s � � � �z"Hashes.__hash__.<locals>.<genexpr>)r; rJ r r r r, r r r �__hash__u s ���zHashes.__hash__r' )r r r r )�__name__�
__module__� __qualname__�__doc__r �strr r r& �property�intr- rE r1 r �bytesr= r8 r r@ rD rF �objectrG rM r r r r r s
r c s< e Zd ZdZd � fdd� Zdeedf ddfd d
�Z� ZS ) �
MissingHashesz�A workalike for Hashes used when we're missing a hash for a requirement
It computes the actual hash of the requirement and raises a HashMissing
exception showing it to the user.
r Nc s t � jtg id� dS )z!Don't offer the ``hashes`` kwarg.)r N)�superr �
FAVORITE_HASHr, �� __class__r r r � s zMissingHashes.__init__r9 r r c C s t |t �� ��r' )r rY r7 r> r r r r8 � s zMissingHashes._raise)r N) rN rO rP rQ r r rR r8 �
__classcell__r r rZ r rW � s "rW )r3 �typingr r r r r �pip._internal.exceptionsr r r �pip._internal.utils.miscr
r r rY �
STRONG_HASHESr rW r r r r �<module> s h