U
��.e� � @ s� d dl mZ d dlZd dlmZmZmZ d dlmZm Z m
Z
d dl m Z d dl
mZ er�d dlmZmZmZmZmZ d dlmZ er�d d lmZ n d d
lmZ d Zd d d
gZG dd� de�ZG dd� de�Z dS )� )�absolute_importN)� iteritems�iterkeys�
itervalues)� HashMismatch� HashMissing�InstallationError)� read_chunks)�MYPY_CHECK_RUNNING)�Dict�List�BinaryIO�NoReturn�Iterator)�PY3)�_Hash)�_hashZsha256Zsha384Zsha512c @ s^ e Zd ZdZddd�Zedd� �Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dS )�HasheszaA wrapper that builds multiple hashes at once and checks them against
known-good values
Nc C s |dkr i n|| _ dS )zo
:param hashes: A dict of algorithm names pointing to lists of allowed
hex digests
N)�_allowed)�self�hashes� r �>/usr/lib/python3.8/site-packages/pip/_internal/utils/hashes.py�__init__, s zHashes.__init__c C s t dd� | j�� D ��S )Nc s s | ]}t |�V qd S �N)�len)�.0Zdigestsr r r � 7 s z&Hashes.digest_count..)�sumr �values�r r r r � digest_count4 s zHashes.digest_countc C s || j �|g �kS )z/Return whether the given hex digest is allowed.)r �get)r � hash_nameZ
hex_digestr r r �is_hash_allowed9 s zHashes.is_hash_allowedc
C s� i }t | j�D ]<}zt�|�||< W q ttfk
rH td| ��Y qX q|D ] }t|�D ]}|�|� q\qPt |�D ] \}}|�
� | j| krv dS qv| � |� dS )z�Check good hashes against ones built from iterable of chunks of
data.
Raise HashMismatch if none match.
zUnknown hash name: %sN) r r �hashlib�new�
ValueError� TypeErrorr r �updater � hexdigest�_raise)r Zchunks�gotsr# �chunk�hashZgotr r r �check_against_chunksA s zHashes.check_against_chunksc C s t | j|��d S r )r r �r r, r r r r+ Y s z
Hashes._raisec C s | � t|��S )zaCheck good hashes against a file-like object
Raise HashMismatch if none match.
)r/ r )r �filer r r �check_against_file] s zHashes.check_against_filec
C s, t |d��}| �|�W 5 Q R � S Q R X d S )N�rb)�openr2 )r �pathr1 r r r �check_against_pathf s zHashes.check_against_pathc C s
t | j�S )z,Return whether I know any known-good hashes.)�boolr r r r r � __nonzero__k s zHashes.__nonzero__c C s | � � S r )r8 r r r r �__bool__p s zHashes.__bool__)N)�__name__�
__module__� __qualname__�__doc__r �propertyr! r$ r/ r+ r2 r6 r8 r9 r r r r r ' s
r c s( e Zd ZdZ� fdd�Zdd� 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.
c s t t| �jtg id� dS )z!Don't offer the ``hashes`` kwarg.)r N)�superr? r �
FAVORITE_HASHr �� __class__r r r | s zMissingHashes.__init__c C s t |t �� ��d S r )r rA r* r0 r r r r+ � s zMissingHashes._raise)r: r; r< r= r r+ �
__classcell__r r rB r r? u s r? ) Z
__future__r r% Zpip._vendor.sixr r r Zpip._internal.exceptionsr r r Zpip._internal.utils.miscr Zpip._internal.utils.typingr
�typingr r r
r r r r r rA Z
STRONG_HASHES�objectr r? r r r r � s
N