o Ne"&@sddlZddlZddlZddlZddlZddlmZgZdZdZ dZ ej ej dddd Zd d Zd d ZddZddZddZd%ddZddZddZddZd&dd Zd&d!d"Zd&d#d$ZdS)'N)get_path_from_urlz~/.cache dependenciesz hubconf.py~z.cacheZpaddleZhubcCs8tj|rtj|rt|dSt|dSdS)N)ospathexistsisfileremoveshutilrmtree)rr ?D:\Projects\ConvertPro\env\Lib\site-packages\paddle/hapi/hub.py_remove_if_existss  rcCsZtjd|z t|}tj|Wnty$tj|tdwtj||S)NrzPPlease make sure config exists or repo error messages above fixed when importing) sysrinsert __import__modulespop ImportErrorr RuntimeError)namerepo_dir hub_moduler r r _import_module&s   rcCs0|dkr d|||S|dkrd|||SdS)Ngithubz'https://github.com/{}/{}/archive/{}.zipgiteez1https://gitee.com/{}/{}/repository/archive/{}.zip)format) repo_owner repo_namebranchsourcer r r _git_archive_link6sr!cCsD|dkrdnd}d|vr|d\}}n|}|d\}}|||fS)NrmainZmaster:/)split)repor rZ repo_inforrr r r _parse_repo_info?s  r'cCsDzddlm}Wntyddlm}Ynw||jdddS)Nr)PathT)exist_ok)pathlibr(rZpathlib2mkdir)dirnamer(r r r _make_dirsIs  r-TrcCs@t}t|t||\}}}|dd}tj|d|||g} | o)tj| } | r9|r7tj d | | Stj||d} t | t ||||d} t| || d|dkrZdnd d } t| | t| }|d j}tj||}t |||Wdn1swYt | t | t|| | S) Nr$_zUsing cache found in {} z.zipr FrZwgetget)Z check_exist decompressmethodr)HUB_DIRr-r'replacerrjoinrrstderrwriterrr!rr movezipfileZipFileinfolistfilename extractall)r& force_reloadverboser Zhub_dirrrrZ normalized_brrZ use_cacheZ cached_fileurlZfpathZcached_zipfileZextraced_repo_nameZextracted_repor r r _get_cache_or_reloadQsB     rAcCs@t|ts tdt||d}|dust|std||S)zload entry from hubconf z5Invalid input: model should be a str of function nameNz"Cannot find callable {} in hubconf) isinstancestr ValueErrorgetattrcallablerr)mrfuncr r r _load_entry_from_hubconfs  rIcCs$zt|WdStyYdSw)NTF)rr)rr r r _check_module_existss  rJcCsFt|td}|durdd|D}t|r!tdd|dSdS)NcSsg|]}t|s|qSr )rJ).0pkgr r r s z'_check_dependencies..zMissing dependencies: {}z, )rEVAR_DEPENDENCYlenrrr5)rGr missing_depsr r r _check_dependenciess rQFcs\|dvr td||dvrt||d|d}ttdd|fdd tD}|S) a List all entrypoints available in `github` hubconf. Args: repo_dir(str): github or local path. github path (str): a str with format "repo_owner/repo_name[:tag_name]" with an optional tag/branch. The default branch is `main` if not specified. local path (str): local repo path source (str): `github` | `gitee` | `local`, default is `github`. force_reload (bool, optional): whether to discard the existing cache and force a fresh download, default is `False`. Returns: entrypoints: a list of available entrypoint names Example: .. code-block:: python import paddle paddle.hub.list('lyuwenyu/paddlehub_demo:main', source='github', force_reload=False) rrlocalCUnknown source: "{}". Allowed values: "github" | "gitee" | "local".rrTr/.rcs(g|]}tt|r|ds|qS)r.)rFrE startswith)rKfrr r rMs zlist..)rDrrArMODULE_HUBCONFr%dir)rr r> entrypointsr rYr lists  r]cCsR|dvr td||dvrt||d|d}ttdd|}t||}|jS)a Show help information of model Args: repo_dir(str): github or local path. github path (str): a str with format "repo_owner/repo_name[:tag_name]" with an optional tag/branch. The default branch is `main` if not specified. local path (str): local repo path. model (str): model name. source (str): `github` | `gitee` | `local`, default is `github`. force_reload (bool, optional): default is `False`. Return: docs Example: .. code-block:: python import paddle paddle.hub.help('lyuwenyu/paddlehub_demo:main', model='MM', source='github') rRrTrUTr/rVr)rDrrArrZr%rI__doc__)rmodelr r>rentryr r r helps racKsb|dvr td||dvrt||d|d}ttdd|}t|t||}|d i|S) a Load model Args: repo_dir(str): github or local path. github path (str): a str with format "repo_owner/repo_name[:tag_name]" with an optional tag/branch. The default branch is `main` if not specified. local path (str): local repo path. model (str): model name. source (str): `github` | `gitee` | `local`, default is `github`. force_reload (bool, optional): default is `False`. **kwargs: parameters using for model Return: paddle model Example: .. code-block:: python import paddle paddle.hub.load('lyuwenyu/paddlehub_demo:main', model='MM', source='github') rRrTrUTr/rVrNr )rDrrArrZr%rQrI)rr_r r>kwargsrr`r r r loads rc)Tr)rF)rrerr r9Zpaddle.utils.downloadr__all__ZDEFAULT_CACHE_DIRrNrZr expanduserr5r3rrr!r'r-rArIrJrQr]rarcr r r r s.   6 .,