\jZUdZddlZddlZddlZddlZddlmZddlmZeje Z dZ iZ eedfed<ej"Zed Gd d Zd ed edefdZdedefdZd ed edeeefdzfdZdedeeefdzfdZd ed edeeefddfdZy)a5On-disk cache for repository tree listings. A tree listing is the set of files (with their download metadata) contained in a repo at a given commit. Because a commit hash is immutable, its tree listing never changes and can be cached forever without any invalidation logic. The listing is stored as a human-readable JSON file under `/trees/.json`. The folder depends on the download target: the per-repo `storage_folder` for `cache_dir` downloads, or `local_dir/.cache/huggingface/` for `local_dir` downloads (see `tree_cache_folder_for_local_dir`). ```json { "format_version": 1, "files": { "config.json": {"size": 519, "blob_id": ""}, "model.safetensors": {"size": 1234, "blob_id": "...", "lfs_sha256": "", "lfs_size": 1234, "xet_hash": "..."} } } ``` N) dataclass)loggingzdict[str, TreeCacheEntry]_IN_MEMORY_TREE_CACHET)frozenceZdZUdZeed<eed<dZedzed<dZedzed<dZ edzed<de fd Z e d e ddfd Z y) TreeCacheEntryz^Raw metadata of a single file in a cached tree listing, mirroring the `/tree` endpoint fields.sizeblob_idN lfs_sha256lfs_sizexet_hashreturnc|j|jd}|j|j|d<|j|d<|j|j|d<|S)N)r r r r rr r r r r)selfinfos N/root/.hermes/venv/lib/python3.12/site-packages/huggingface_hub/_tree_cache.pyto_jsonzTreeCacheEntry.to_json>sW"iiDLLA ?? &!%D #}}D  == $#}}D  rc ||d|d|jd|jd|jdS)Nr r r r rr)get)clsrs r from_jsonzTreeCacheEntry.from_jsonGsAfOxx -XXj)XXj)   r)__name__ __module__ __qualname____doc__int__annotations__strr r rdictr classmethodrrrr r 4shh I L!Jd !HcDjHcDj T &6  rr tree_cache_folder commit_hashrcJtjj|d|dS)Ntreesz.jsonospathjoin)r%r&s r_tree_cache_pathr-Rs! 77<<)7{m54I JJr local_dircDtjj|ddS)zGFolder under which the `trees/` cache lives for a `local_dir` download.z.cache huggingfacer))r.s rtree_cache_folder_for_local_dirr1Vs 77<< 8] ;;rct||}t5|tvrt|cdddS dddt|}|t5|t|<ddd|S|S#1swY2xYw#1swY|SxYw)zZReturn the cached tree listing for a commit hash, or `None` if not cached (or unreadable).N)r-_IN_MEMORY_TREE_CACHE_LOCKr_read_tree_cache_from_disk)r%r&r+entriess rread_tree_cacher6[s -{ ;D #/ ( ((.// (/).G ' 2*1 !$ ' 2 N7N//  2 NsA  A, A),A6r+c t|d5}tj|}dddjdtk7ry|dj Dcic]\}}|t j|c}}S#1swY\xYwcc}}w#t$rYyttttf$r%}tjd|d|Yd}~yd}~wwxYw)Nutf-8encodingformat_versionfilesz#Ignoring corrupted tree cache file : )openjsonloadrTREE_CACHE_FORMAT_VERSIONitemsr rFileNotFoundErrorOSError ValueErrorKeyError TypeErrorloggerwarning)r+fdata file_pathres rr4r4hs  $ ) Q99Q33D99gg   h  Z9 5rds)( !   H %ACtC!<<=B+Y^^- $   :KK#K#KFYAZ]aAa  S T#~2E-F-M 44#4SR`M`Ha4fj4r