ujtddlZddlZddlmZddlmZddlmZddlm Z ddl m Z dZ Gdd eZ dS) N)ArgumentParser)Path) CLICommand)TEMPORARY_FOLDER_NAME)get_model_masked_directoryc t|S)z> Function which will be called for a specific sub parser. ) ClearCacheCMD)argss i/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/modelscope/cli/clearcache.pysubparser_funcr s   cFeZdZdZdZedefdZdZdZ dZ dS) r z clear-cachec||_tjdtjdd|_dS)NMODELSCOPE_CACHEz.cache modelscope)r osgetenvrhomejoinpath cache_dir)selfr s r __init__zClearCacheCMD.__init__s<   IKK < 8 8::r parsersc|tj}|}|dt d|dt d|tdS)z. define args for clear-cache command. z--modelzThe id of the model whose cache will be cleared. For clear-cache, if neither model or dataset id is provided, entire cache will be cleared.)typehelpz --datasetzThe id of the dataset whose cache will be cleared. For clear-cache, if neither model or dataset id is provided, entire cache will be cleared.)funcN) add_parserr nameadd_mutually_exclusive_group add_argumentstr set_defaultsr )rparsergroups r define_argszClearCacheCMD.define_argss##M$6773355   X       X     00000r c.|dS)N)_execute_with_confirmation)rs r executezClearCacheCMD.execute2s '')))))r cPd}d}d}|jjs |jjr:|jjr|jj}d}|d|dz}n&|jj}|d|dz}n|d|jdz}d}t |d z}|d kr|r+||jtd dStj |j|rd nd |}tj |j|rd nd t|}||}||} |s.| s,|rtd|ddStd|ddStd dStddS)NFz You are about to delete Tzlocal cache for model z. zlocal cache for dataset zentire ModelScope cache at z$, including ALL models and dataset. z9 Please press Y or y to proceed, any other key to abort. YzCache cleared.hubdatasetszCache for Model z not found. Nothing to do.zCache for Dataset zOperation aborted.) r modeldatasetrinputstripupper_remove_directoryprintrpathjoinr) rall single_modelpromptid user_inputentity_directorytemp_directoryentity_removed temp_removeds r r(z(ClearCacheCMD._execute_with_confirmation5s& - 9? di/ y DY_# "A2"A"A"AAY&"CR"C"C"CCqDNqqqqFC K L   %''%%''     ,&&t~666&'''''#%7<<N\$IEEz2$O$O !#N\$IEEz)2"/"/"&!7!78H!I!I#55nEE & , ,#MrMMMOOOOOOOOO*+++++ & ' ' ' Fr ctj|r tj|rNt jtj|tj|td|dn't j|td|ddS#t$r}td|d|Yd}~nd}~wwxYwdSdS)NzCache and link for z removed.z Cache folder Tz*An error occurred while clearing cache at z: F) rr5existsislinkshutilrmtreereadlinkremover4 Exception)rr5es r r3zClearCacheCMD._remove_directoryds 7>>$    P7>>$'';M"+d"3"3444IdOOO????@@@@M$'''9$999:::t P P PN4NN1NNOOOOOOOO P5  sBB77 C CC N) __name__ __module__ __qualname__rr staticmethodrr&r)r(r3r r r r s} D::: 1^111\1,***---^     r r )rrCargparserpathlibrmodelscope.cli.basermodelscope.hub.constantsrmodelscope.hub.utils.utilsrr r rMr r rSs ######******::::::AAAAAA _____J_____r