U
    Mff                     @   s   d dl mZ d dlZd dlmZmZmZ d dlmZm	Z	m
Z
 d dlmZ d dlmZ erd dlmZmZmZmZmZ d dlmZ erd d	lmZ nd d
lmZ dZdddgZG dd deZG dd deZdS )    )absolute_importN)	iteritemsiterkeys
itervalues)HashMismatchHashMissingInstallationError)read_chunks)MYPY_CHECK_RUNNING)DictListBinaryIONoReturnIterator)PY3)_Hash)_hashsha256sha384sha512c                   @   s^   e Zd ZdZdddZedd Zdd Zd	d
 Zdd Z	dd Z
dd Zdd Zdd ZdS )HasheszaA wrapper that builds multiple hashes at once and checks them against
    known-good values

    Nc                 C   s   |dkri n|| _ dS )zo
        :param hashes: A dict of algorithm names pointing to lists of allowed
            hex digests
        N)_allowed)selfhashes r   >/tmp/pip-unpacked-wheel-o7_vsxcn/pip/_internal/utils/hashes.py__init__)   s    zHashes.__init__c                 C   s   t dd | j D S )Nc                 s   s   | ]}t |V  qd S N)len).0digestsr   r   r   	<genexpr>4   s     z&Hashes.digest_count.<locals>.<genexpr>)sumr   valuesr   r   r   r   digest_count1   s    zHashes.digest_countc                 C   s   || j |g kS )z/Return whether the given hex digest is allowed.)r   get)r   	hash_name
hex_digestr   r   r   is_hash_allowed6   s    zHashes.is_hash_allowedc              
   C   s   i }t | jD ]<}zt|||< W q ttfk
rH   td| Y qX q|D ]}t|D ]}|| q\qPt	|D ] \}}|
 | j| krv dS qv| | dS )zCheck good hashes against ones built from iterable of chunks of
        data.

        Raise HashMismatch if none match.

        zUnknown hash name: %sN)r   r   hashlibnew
ValueError	TypeErrorr   r   updater   	hexdigest_raise)r   chunksgotsr'   chunkhashgotr   r   r   check_against_chunks?   s    zHashes.check_against_chunksc                 C   s   t | j|d S r   )r   r   r   r2   r   r   r   r0   W   s    zHashes._raisec                 C   s   |  t|S )zaCheck good hashes against a file-like object

        Raise HashMismatch if none match.

        )r6   r	   )r   filer   r   r   check_against_file[   s    zHashes.check_against_filec              
   C   s,   t |d}| |W  5 Q R  S Q R X d S )Nrb)openr9   )r   pathr8   r   r   r   check_against_pathd   s    zHashes.check_against_pathc                 C   s
   t | jS )z,Return whether I know any known-good hashes.)boolr   r$   r   r   r   __nonzero__i   s    zHashes.__nonzero__c                 C   s   |   S r   )r?   r$   r   r   r   __bool__n   s    zHashes.__bool__)N)__name__
__module____qualname____doc__r   propertyr%   r)   r6   r0   r9   r=   r?   r@   r   r   r   r   r   $   s   

		r   c                       s(   e Zd ZdZ fddZdd Z  ZS )MissingHasheszA 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| jtg id dS )z!Don't offer the ``hashes`` kwarg.)r   N)superrF   r   FAVORITE_HASHr$   	__class__r   r   r   z   s    zMissingHashes.__init__c                 C   s   t |t  d S r   )r   rH   r/   r7   r   r   r   r0      s    zMissingHashes._raise)rA   rB   rC   rD   r   r0   __classcell__r   r   rI   r   rF   s   s   rF   )
__future__r   r*   Zpip._vendor.sixr   r   r   pip._internal.exceptionsr   r   r   pip._internal.utils.miscr	   pip._internal.utils.typingr
   typingr   r   r   r   r   r   r   r   rH   STRONG_HASHESobjectr   rF   r   r   r   r   <module>   s   
O