o id@srddlZddlmZddlmZmZddlZddlmZe fdee de fddZ d eee fde fd d Z dS) N)Path)IterableUnion)msgexcludereturncsBt|trfdd|D}tj|ddd}t|S)zGet the hash for a JSON-serializable object. data: The data to hash. exclude (Iterable[str]): Top-level keys to exclude if data is a dict. RETURNS (str): The hash. csi|] \}}|vr||qSr).0kvrrG/home/ubuntu/.local/lib/python3.10/site-packages/weasel/util/hashing.py szget_hash..T) sort_keysutf8) isinstancedictitemssrsly json_dumpsencodehashlibmd5 hexdigest)datardata_strrr r get_hash s rpathcCst|}|s|stjd|ddd|r&tt|St}t dd| dDD] }| |q6|S)zGet the checksum for a file or directory given its file path. If a directory path is provided, this uses all files in that directory. path (Union[Path, str]): The file or directory path. RETURNS (str): The checksum. zCan't get checksum for z: not a file or directory)exitscss|] }|r|VqdS)N)is_file)r fprrr %szget_checksum..*) rr is_dirrfailrr read_bytesrsortedrglobupdate)r dir_checksumsub_filerrr get_checksumsr,) rpathlibrtypingrrrwasabirtuplestrrr,rrrr s