def unpack_local_file(self,
name: str = None
) -> Dict[str, Union[str, List[str]]]:
if not Path(name).exists():
raise FileNotFoundError(f"No such file or directory: {name}")
warnings.warn("Expanding a local model to the cachedir. "
"If you'll move the file to another path, "
"it's treated as a different model.")
name = Path(name).absolute()
outdir = self.cachedir / str_to_hash(name)
filename = outdir / name.name
outdir.mkdir(parents=True, exist_ok=True)
if not filename.exists():
if filename.is_symlink():
filename.unlink()
filename.symlink_to(name)
# Skip unpacking if the cache exists
meta_yaml = outdir / "meta.yaml"
if meta_yaml.exists():
info = get_dict_from_cache(meta_yaml)
if info is not None:
return info
# Extract files from archived file
return unpack(filename, outdir)
def download_and_unpack(self,
name: str = None,
version: int = -1,
**kwargs: str) -> Dict[str, Union[str, List[str]]]:
url = self.get_url(name=name, version=version, **kwargs)
if not is_url(url) and Path(url).exists():
return self.unpack_local_file(url)
# Unpack to <cachedir>/<hash> in order to give an unique name
outdir = self.cachedir / str_to_hash(url)
# Skip downloading and unpacking if the cache exists
meta_yaml = outdir / "meta.yaml"
if meta_yaml.exists():
info = get_dict_from_cache(meta_yaml)
if info is not None:
return info
# Download the file to an unique path
filename = self.download(url)
# Extract files from archived file
return unpack(filename, outdir)