o ߥiB @sddlZddlZddlmZddlmZddlmZddlZddlm Z m Z m Z m Z m Z ddlmZddlmZddlmZeZd d Zdd eefd d ZddZddZddZddZdS)N)datetime)Path)Optional)DEFAULT_MODELSCOPE_DOMAINDEFAULT_MODELSCOPE_GROUPMODEL_ID_SEPARATORMODELSCOPE_SDK_DEBUGMODELSCOPE_URL_SCHEME)FileIntegrityError) get_default_modelscope_cache_dir) get_loggercCs<t|vr|td}|td}||fSt}|}||fS)Nr)rsplitr)model_idgroup_or_ownernamerN/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/hub/utils/utils.pymodel_id_to_group_owner_namesrrcCsDtdd}tdtj|d}|dur|Stj||dS)zcache dir precedence: function parameter > environment > ~/.cache/modelscope/hub Args: model_id (str, optional): The model id. Returns: str: the model_id dir if model_id not None, otherwise cache root dir. z.cache modelscopeMODELSCOPE_CACHEhubN/)rhomejoinpathosgetenvpathjoin)rdefault_cache_dir base_pathrrr get_cache_dir s r!cCsLttjvrttt}|Sddlm }ttt |j d}|S)Nr)versionz%Y-%m-%d %H:%M:%S) rrenvironintroundrnow timestamprr"strptime__release_datetime__)rtr"rrrget_release_datetime/s  r+cCstdt}t|S)NMODELSCOPE_DOMAIN)rrrr )modelscope_domainrrr get_endpoint;sr.cCsbd}t}t|d} ||}|sn||q Wd|S1s(wY|S)Nirb)hashlibsha256openreadupdate hexdigest) file_path BUFFER_SIZE sha256_hashfdatarrr compute_hashAs    r;cCs>t|}||kst|d|||f}t|t|dS)zValidate the file hash is expected, if not, delete the file Args: file_path (str): The file to validate expected_sha256 (str): The expected sha256 hash Raises: FileIntegrityError: If file_path hash is not expected. zFile %s integrity check failed, expected sha256 signature is %s, actual is %s, the download may be incomplete, please try again.N)r;rremoveloggererrorr )r6expected_sha256 file_sha256msgrrrfile_integrity_validationMs   rB)N)r0rrpathlibrtypingrrequestsmodelscope.hub.constantsrrrrr modelscope.hub.errorsr modelscope.utils.file_utilsr modelscope.utils.loggerr r=rstrr!r+r.r;rBrrrrs"