a
lj @ s d dl Z d dlmZmZmZmZmZ d dlmZm Z m
Z
d dl m Z er`d dl m
Z
d dlmZ dZg dZG d d
d
ZG d d d eZdS )
N)
TYPE_CHECKINGBinaryIODictIteratorList) HashMismatch HashMissingInstallationError) read_chunks)_Hash)NoReturnsha256)r
Zsha384Zsha512c @ s e Zd ZdZd#eeee f ddddZd d dddZe e
d d
d Z eee d d
dZ
ee ddddZeedf ddddZeddddZedddd Ze d d d Zee dd d Ze
d d!d"ZdS )$HasheszaA wrapper that builds multiple hashes at once and checks them against
known-good values
N)hashesreturnc C s4 i }|dur*| D ]\}}t|||< q|| _dS )zo
:param hashes: A dict of algorithm names pointing to lists of allowed
hex digests
N)itemssorted_allowed)selfr allowedalgkeys r >/usr/lib/python3.9/site-packages/pip/_internal/utils/hashes.py__init__ s
zHashes.__init__)otherr c sb t |tstS |sS s |S i }|j D ],\ } jvr@q, fdd|D | < q,t|S )Nc s g | ]}|j v r|qS r )r ).0vr r r r
; z"Hashes.__and__..)
isinstancer NotImplementedr r )r r newvaluesr r r __and__+ s
zHashes.__and__r c C s t dd | j D S )Nc s s | ]}t |V qd S N)len)r Zdigestsr r r @ r z&Hashes.digest_count..)sumr r$ r r r r digest_count> s zHashes.digest_count) hash_name
hex_digestr c 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)chunksr c
C s i }| j D ]<}zt|||< W q ttfyH td| Y q0 q|D ] }| D ]}|| q\qP| D ] \}}|
| j | v rv dS qv| | dS )zCheck 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 r1 gotsr- chunkhashZgotr r r check_against_chunksF s zHashes.check_against_chunksr r r8 r c C s t | j|d S r' )r r r r8 r r r r7 ] s z
Hashes._raise)filer c 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)pathr c C s8 t |d}| |W d S 1 s*0 Y d S )Nrb)openr? )r r@ r> r r r check_against_pathh s zHashes.check_against_pathc C s
t | jS )z,Return whether I know any known-good hashes.)boolr r+ r r r __bool__l s zHashes.__bool__c C s t |tstS | j|jkS r' )r! r r" r )r r r r r __eq__p s
z
Hashes.__eq__c C s" t dtdd | j D S )N,c s s* | ]"\}}|D ]}d ||fV qqdS ):N)join)r r Z digest_listZdigestr r r r) x s z"Hashes.__hash__..)r: rI r r r r+ r r r __hash__u s zHashes.__hash__)N)__name__
__module__ __qualname____doc__r strr r r% propertyintr, rD r0 r bytesr; r7 r r? rC rE objectrF rJ r r r r r s r c s> e Zd ZdZdd fdd Zeedf ddd d
Z ZS )
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.
Nr& c s t jtg id dS )z!Don't offer the ``hashes`` kwarg.)r N)superr
FAVORITE_HASHr+ __class__r r r s zMissingHashes.__init__r r r< c C s t |t d S r' )r rV r6 r= r r r r7 s zMissingHashes._raise) rK rL rM rN r r rO r7
__classcell__r r rW r rT s rT )r2 typingr r r r r Zpip._internal.exceptionsr r r Zpip._internal.utils.miscr
r r rV Z
STRONG_HASHESr rT r r r r s h