-PhdZddlZddlZddlmZdejDZ ddlZeddDde DZn #e $rYnwxYwdd Z d Z dd Z ddZdS)z' Calculating and checking file hashes. N)Pathc vi|]6}|tt|tjtj|7S)getattrhashlib functoolspartialnew.0algs L/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pooch/hashes.py rsE #y0cBB C Cc<i|]}|tt|dSN)rxxhashr s rrrs6    d++   r)xxh128xxh64xxh32xxh3_128xxh3_64ci|] \}}||| Srr)r r funcs rrr$s*c44CSTCSCSCSrsha256c |tvr:td|dttdd}t|}t |d5}||}|r,||||}|,dddn #1swxYwY|S)ad Calculate the hash of a given file. Useful for checking if a file has changed or been corrupted. Parameters ---------- fname : str The name of the file. alg : str The type of the hashing algorithm Returns ------- hash : str The hash of the file. Examples -------- >>> fname = "test-file-for-hash.txt" >>> with open(fname, "w") as f: ... __ = f.write("content of the file") >>> print(file_hash(fname)) 0fc74468e6a9a829f103d069aeb2bb4f8646bad58bf146bb0e3379b759ec4a00 >>> import os >>> os.remove(fname) z Algorithm 'zR' not available to the pooch library. Only the following algorithms are available .irbN)ALGORITHMS_AVAILABLE ValueErrorlistkeysopenreadupdate hexdigest)fnamer chunksizehasherfinbuffs r file_hashr,+s-< &&& 4# 4 4(--//00 4 4 4   I !# & ( (F eT  'cxx "" ' MM$   88I&&D ''''''''''''''''     s*AB::B>B>c|d}||}n"d|vr|}n|dd}|S)a Parse the name of the hash method from the hash string. The hash string should have the following form ``algorithm:hash``, where algorithm can be the name of any algorithm known to :mod:`hashlib`. If the algorithm is omitted or the hash string is None, will default to ``"sha256"``. Parameters ---------- hash_string : str The hash string with optional algorithm prepended. Returns ------- hash_algorithm : str The name of the algorithm. Examples -------- >>> print(hash_algorithm("qouuwhwd2j192y1lb1iwgowdj2898wd2d9")) sha256 >>> print(hash_algorithm("md5:qouuwhwd2j192y1lb1iwgowdj2898wd2d9")) md5 >>> print(hash_algorithm("sha256:qouuwhwd2j192y1lb1iwgowdj2898wd2d9")) sha256 >>> print(hash_algorithm("SHA256:qouuwhwd2j192y1lb1iwgowdj2898wd2d9")) sha256 >>> print(hash_algorithm("xxh3_64:qouuwhwd2j192y1lb1iwgowdj2898wd2d9")) xxh3_64 >>> print(hash_algorithm(None)) sha256 rN:r)splitlower) hash_stringdefault algorithms rhash_algorithmr4ZsQJG K   %%c**1- ??  rFc X|dSt|}t||}||ddk}|r@|s>|t |}t |d|d|d|d |S) a Check if the hash of a file matches a known hash. If the *known_hash* is None, will always return True. Coverts hashes to lowercase before comparison to avoid system specific mismatches between hashes in the registry and computed hashes. Parameters ---------- fname : str or PathLike The path to the file. known_hash : str The known hash. Optionally, prepend ``alg:`` to the hash to specify the hashing algorithm. Default is SHA256. strict : bool If True, will raise a :class:`ValueError` if the hash does not match informing the user that the file may be corrupted. source : str The source of the downloaded file (name or URL, for example). Will be used in the error message if *strict* is True. Has no other use other than reporting to the user where the file came from in case of hash mismatch. If None, will default to *fname*. Returns ------- is_same : bool True if the hash matches, False otherwise. NT)r r.z hash of downloaded file (z*) does not match the known hash: expected z but got zm. Deleted download for safety. The downloaded file may have been corrupted or the known hash may be outdated.)r4r,r0r/strr upper)r' known_hashstrictsourcer3new_hashmatchess r hash_matchesr>s>tz**II...Hnn*"2"23"7"7";"A"A"C"CCG  g >ZZF   / /F / /)3 / />F / / /   NrTc t|rd}nd}tfd|D}fd|D}t|dd5}t ||D]B\}}|d|d d |C d d d d S#1swxYwYd S) a. Make a registry of files and hashes for the given directory. This is helpful if you have many files in your test dataset as it keeps you from needing to manually update the registry. Parameters ---------- directory : str Directory of the test data to put in the registry. All file names in the registry will be relative to this directory. output : str Name of the output registry file. recursive : bool If True, will recursively look for files in subdirectories of *directory*. z**/**c3K|]:}|t|V;dSr)is_filer7 relative_to)r path directorys r z make_registry..s[  <<>> D  Y ' '((rcNg|]!}tt|z "Sr)r,r7)r r'rEs r z!make_registry..s. C C CEiI-..// C C Crwzutf-8)encodingz{} {} \/N)rsortedglobr#zipwriteformatreplace) rEoutput recursivepatternfileshashesoutfiler'fhashs ` r make_registryrZsd&YI NN7++  E D C C CU C C CF fcG , , ,Mv.. M MLE5 MM)**5==s+C+CUKK L L L L MMMMMMMMMMMMMMMMMMMs&ACC C )r)FN)T)__doc__rrpathlibralgorithms_availablerrr%items ImportErrorr,r4r>rZrrrr`sL +  MMM  J   #7#=#=#?#?   D ,,,,^,,,^----`%M%M%M%M%M%MsAA%%A-,A-