
    ,h!                         S r SSKrSSKJrJr  SSKJr  SSKJrJ	r	J
r
  SSKJr  SS	KJrJrJr   " S
 S\5      rSS.S\	S\S\4S jjrg)zContains command to scan the HF cache directory.

Usage:
    huggingface-cli scan-cache
    huggingface-cli scan-cache -v
    huggingface-cli scan-cache -vvv
    huggingface-cli scan-cache --dir ~/.cache/huggingface/hub
    N)	Namespace_SubParsersAction)Optional   )CacheNotFoundHFCacheInfoscan_cache_dir   )BaseHuggingfaceCLICommand)ANSIshow_deprecation_warningtabulatec                   V    \ rS rSr\S\4S j5       rS\SS4S jrS r	S	\
SS4S
 jrSrg)ScanCacheCommand!   parserc                     U R                  SSS9nUR                  S[        S SS9  UR                  SSS	S
SS9  UR                  [        S9  g )Nz
scan-cachezScan cache directory.)helpz--dirzMcache directory to scan (optional). Default to the default HuggingFace cache.)typedefaultr   z-vz	--verbosecountr   zshow a more verbose output)actionr   r   )func)
add_parseradd_argumentstrset_defaultsr   )r   scan_cache_parsers     [/var/www/html/shao/venv/lib/python3.13/site-packages/huggingface_hub/commands/scan_cache.pyregister_subcommand$ScanCacheCommand.register_subcommand"   sp    "--lAX-Y&&`	 	' 	
 	&&- 	' 	
 	&&,<&=    argsreturnNc                 H    UR                   U l        UR                  U l        g N)verbose	verbositydir	cache_dir)selfr#   s     r   __init__ScanCacheCommand.__init__5   s    "ll(,r"   c           
      2   [        SS5         [        R                  " 5       n[        U R                  5      n[        R                  " 5       nU R                  U5        [        S[        X1-
  S5       S[        UR                  5       S[        R                  " UR                  5       S35        [        UR                  5      S	:  a  S
[        UR                  5       S3nU R                  S:  aR  [        [        R                  " U5      5        UR                   H"  n[        [        R                  " U5      5        M$     g [        [        R                  " US-   5      5        g g ! [         a$  nUR                  n[        SU 35         S nAg S nAff = f)Nzhuggingface-cli scan-cachezhf cache scanzCache directory not found: z	
Done in r
   zs. Scanned z repo(s) for a total of .r   zGot z warning(s) while scanning.   z Use -vvv to print details.)r   timer	   r*   r   print_print_hf_cache_info_as_tableroundlenreposr   redsize_on_disk_strwarningsr(   gray)r+   t0hf_cache_infot1excr*   messagewarnings           r   runScanCacheCommand.run9   sN    !=O	B*4>>:MB 	**=9rw*+;s=;N;N7O6P Q#xx(F(FGHK	
 }%%&*S!7!7899TUG~~"dii(),55G$))G,-  6 dii*G GHI +  	I/	{;<	s   ?E( (
F2FFr<   c                 <    [        [        XR                  S95        g )Nr(   )r2   	get_tabler(   )r+   r<   s     r   r3   .ScanCacheCommand._print_hf_cache_info_as_tableT   s    i@Ar"   )r*   r(   )__name__
__module____qualname____firstlineno__staticmethodr   r    r   r,   rA   r   r3   __static_attributes__ r"   r   r   r   !   sK    >$5 > >$1Y 14 1J6B; B4 Br"   r   rD   r<   r(   r$   c                <   US:X  a  [        [        U R                  S S9 Vs/ sH  nUR                  UR                  SR                  UR                  5      UR                  UR                  UR                  SR                  [        UR                  5      5      [        UR                  5      /PM     sn/ SQS9$ [        [        U R                  S S9 VVs/ sH  n[        UR                  S	 S9 H  nUR                  UR                  UR                  SR                  UR                  5      UR                  UR                  SR                  [        UR                  5      5      [        UR                   5      /PM     M     snn/ S
QS9$ s  snf s  snnf )a  Generate a table from the [`HFCacheInfo`] object.

Pass `verbosity=0` to get a table with a single row per repo, with columns
"repo_id", "repo_type", "size_on_disk", "nb_files", "last_accessed", "last_modified", "refs", "local_path".

Pass `verbosity=1` to get a table with a row per repo and revision (thus multiple rows can appear for a single repo), with columns
"repo_id", "repo_type", "revision", "size_on_disk", "nb_files", "last_modified", "refs", "local_path".

Example:
```py
>>> from huggingface_hub.utils import scan_cache_dir
>>> from huggingface_hub.commands.scan_cache import get_table

>>> hf_cache_info = scan_cache_dir()
HFCacheInfo(...)

>>> print(get_table(hf_cache_info, verbosity=0))
REPO ID                                             REPO TYPE SIZE ON DISK NB FILES LAST_ACCESSED LAST_MODIFIED REFS LOCAL PATH
--------------------------------------------------- --------- ------------ -------- ------------- ------------- ---- --------------------------------------------------------------------------------------------------
roberta-base                                        model             2.7M        5 1 day ago     1 week ago    main C:\Users\admin\.cache\huggingface\hub\models--roberta-base
suno/bark                                           model             8.8K        1 1 week ago    1 week ago    main C:\Users\admin\.cache\huggingface\hub\models--suno--bark
t5-base                                             model           893.8M        4 4 days ago    7 months ago  main C:\Users\admin\.cache\huggingface\hub\models--t5-base
t5-large                                            model             3.0G        4 5 weeks ago   5 months ago  main C:\Users\admin\.cache\huggingface\hub\models--t5-large

>>> print(get_table(hf_cache_info, verbosity=1))
REPO ID                                             REPO TYPE REVISION                                 SIZE ON DISK NB FILES LAST_MODIFIED REFS LOCAL PATH
--------------------------------------------------- --------- ---------------------------------------- ------------ -------- ------------- ---- -----------------------------------------------------------------------------------------------------------------------------------------------------
roberta-base                                        model     e2da8e2f811d1448a5b465c236feacd80ffbac7b         2.7M        5 1 week ago    main C:\Users\admin\.cache\huggingface\hub\models--roberta-base\snapshots\e2da8e2f811d1448a5b465c236feacd80ffbac7b
suno/bark                                           model     70a8a7d34168586dc5d028fa9666aceade177992         8.8K        1 1 week ago    main C:\Users\admin\.cache\huggingface\hub\models--suno--bark\snapshots\70a8a7d34168586dc5d028fa9666aceade177992
t5-base                                             model     a9723ea7f1b39c1eae772870f3b547bf6ef7e6c1       893.8M        4 7 months ago  main C:\Users\admin\.cache\huggingface\hub\models--t5-base\snapshots\a9723ea7f1b39c1eae772870f3b547bf6ef7e6c1
t5-large                                            model     150ebc2c4b72291e770f58e6057481c8d2ed331a         3.0G        4 5 months ago  main C:\Users\admin\.cache\huggingface\hub\models--t5-large\snapshots\150ebc2c4b72291e770f58e6057481c8d2ed331a                                                 ```
```

Args:
    hf_cache_info ([`HFCacheInfo`]):
        The HFCacheInfo object to print.
    verbosity (`int`, *optional*):
        The verbosity level. Defaults to 0.

Returns:
    `str`: The table as a string.
r   c                     U R                   $ r&   	repo_pathrepos    r   <lambda>get_table.<locals>.<lambda>       r"   )keyz{:>12}z, )REPO ID	REPO TYPESIZE ON DISKNB FILESLAST_ACCESSEDLAST_MODIFIEDREFS
LOCAL PATH)rowsheadersc                     U R                   $ r&   rP   rR   s    r   rT   rU      rV   r"   c                     U R                   $ r&   )commit_hash)revisions    r   rT   rU      s
    HL`L`r"   )rX   rY   REVISIONrZ   r[   r]   r^   r_   )r   sortedr6   repo_id	repo_typeformatr8   nb_fileslast_accessed_strlast_modified_strjoinrefsr   rQ   	revisionsrd   snapshot_path)r<   r(   rS   re   s       r   rE   rE   X   sv   V A~ #=#6#6<WX YD LLNNOOD$9$9:MM****IIfTYY/0'	 Y	
 	
4  #=#6#6<WX YD &t~~;` aH LLNN((OOH$=$=>%%..IIfX]]34../	 !b	 Y	
 	
34s   BFB/F
)__doc__r1   argparser   r   typingr   utilsr   r   r	    r   
_cli_utilsr   r   r   r   intr   rE   rM   r"   r   <module>ry      sX     1  > > ' @ @4B0 4Bn ?@ _
[ _
 _
C _
r"   