o rq i@shddlZddlZddlmZddlmZddlmZddlm Z ddl m Z ddZ Gd d d eZ dS) N)ArgumentParser)Path) CLICommand)TEMPORARY_FOLDER_NAME)get_model_masked_directorycCst|S)z> Function which will be called for a specific sub parser. ) ClearCacheCMD)argsr `/home/app/PaddleOCR-VL/.venv_paddleocr/lib/python3.10/site-packages/modelscope/cli/clearcache.pysubparser_func sr c@sBeZdZdZddZedefddZddZd d Z d d Z d S)rz clear-cachecCs$||_tdtdd|_dS)NZMODELSCOPE_CACHEz.cacheZ modelscope)rosgetenvrhomejoinpath cache_dir)selfrr r r __init__s  zClearCacheCMD.__init__parserscCsD|tj}|}|jdtdd|jdtdd|jtddS)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_parserrnameadd_mutually_exclusive_group add_argumentstr set_defaultsr )rparsergroupr r r define_argss zClearCacheCMD.define_argscCs |dS)N)_execute_with_confirmation)rr r r execute2s zClearCacheCMD.executec Cs@d}d}d}|jjs|jjr.|jjr!|jj}d}|d|d}n|jj}|d|d}n |d|jd}d}t|d }|d kr|rU||jtd dSt j |j|r^d nd |}t j |j|rkd 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.ZhubZdatasetszCache for Model z not found. Nothing to do.zCache for Dataset zOperation aborted.) rmodelZdatasetrinputstripupper_remove_directoryprintr pathjoinr) rallZ single_modelpromptidZ user_inputZentity_directoryZtemp_directoryZentity_removedZ temp_removedr r r r 5sV       z(ClearCacheCMD._execute_with_confirmationc Cstj|rRz.tj|r%tt|t|td|dWdSt|td|dWdSt yQ}ztd|d|WYd}~dSd}~wwdS)NzCache and link for z removed.z Cache folder Tz*An error occurred while clearing cache at z: F) r r)existsislinkshutilrmtreereadlinkremover( Exception)rr)er r r r'ds    zClearCacheCMD._remove_directoryN) __name__ __module__ __qualname__rr staticmethodrrr!r r'r r r r rs /r)r r0argparserpathlibrZmodelscope.cli.baserZmodelscope.hub.constantsrZmodelscope.hub.utils.utilsrr rr r r r s