1 files changed, 124 insertions, 0 deletions
diff --git a/src/mlia/utils/filesystem.py b/src/mlia/utils/filesystem.py
new file mode 100644
index 0000000..73a88d9
--- /dev/null
+++ b/src/mlia/utils/filesystem.py
@@ -0,0 +1,124 @@
+# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
+# SPDX-License-Identifier: Apache-2.0
+"""Utils related to file management."""
+import hashlib
+import importlib.resources as pkg_resources
+import json
+import os
+import shutil
+from contextlib import contextmanager
+from pathlib import Path
+from tempfile import mkstemp
+from tempfile import TemporaryDirectory
+from typing import Any
+from typing import Dict
+from typing import Generator
+from typing import Iterable
+from typing import List
+from typing import Optional
+from typing import Union
+
+
+def get_mlia_resources() -> Path:
+    """Get the path to the resources directory."""
+    with pkg_resources.path("mlia", "__init__.py") as init_path:
+        project_root = init_path.parent
+        return project_root / "resources"
+
+
+def get_vela_config() -> Path:
+    """Get the path to the default Vela config file."""
+    return get_mlia_resources() / "vela/vela.ini"
+
+
+def get_profiles_file() -> Path:
+    """Get the Ethos-U profiles file."""
+    return get_mlia_resources() / "profiles.json"
+
+
+def get_profiles_data() -> Dict[str, Dict[str, Any]]:
+    """Get the Ethos-U profile values as a dictionary."""
+    with open(get_profiles_file(), encoding="utf-8") as json_file:
+        profiles = json.load(json_file)
+
+        if not isinstance(profiles, dict):
+            raise Exception("Profiles data format is not valid")
+
+        return profiles
+
+
+def get_profile(target: str) -> Dict[str, Any]:
+    """Get settings for the provided target profile."""
+    profiles = get_profiles_data()
+
+    if target not in profiles:
+        raise Exception(f"Unable to find target profile {target}")
+
+    return profiles[target]
+
+
+def get_supported_profile_names() -> List[str]:
+    """Get the supported Ethos-U profile names."""
+    return list(get_profiles_data().keys())
+
+
+@contextmanager
+def temp_file(suffix: Optional[str] = None) -> Generator[Path, None, None]:
+    """Create temp file and remove it after."""
+    _, tmp_file = mkstemp(suffix=suffix)
+
+    try:
+        yield Path(tmp_file)
+    finally:
+        os.remove(tmp_file)
+
+
+@contextmanager
+def temp_directory(suffix: Optional[str] = None) -> Generator[Path, None, None]:
+    """Create temp directory and remove it after."""
+    with TemporaryDirectory(suffix=suffix) as tmpdir:
+        yield Path(tmpdir)
+
+
+def file_chunks(
+    filepath: Union[Path, str], chunk_size: int = 4096
+) -> Generator[bytes, None, None]:
+    """Return sequence of the file chunks."""
+    with open(filepath, "rb") as file:
+        while data := file.read(chunk_size):
+            yield data
+
+
+def hexdigest(filepath: Union[Path, str], hash_obj: "hashlib._Hash") -> str:
+    """Return hex digest of the file."""
+    for chunk in file_chunks(filepath):
+        hash_obj.update(chunk)
+
+    return hash_obj.hexdigest()
+
+
+def sha256(filepath: Path) -> str:
+    """Return SHA256 hash of the file."""
+    return hexdigest(filepath, hashlib.sha256())
+
+
+def all_files_exist(paths: Iterable[Path]) -> bool:
+    """Check if all files are exist."""
+    return all(item.is_file() for item in paths)
+
+
+def all_paths_valid(paths: Iterable[Path]) -> bool:
+    """Check if all paths are valid."""
+    return all(item.exists() for item in paths)
+
+
+def copy_all(*paths: Path, dest: Path) -> None:
+    """Copy files/directories into destination folder."""
+    dest.mkdir(exist_ok=True)
+
+    for path in paths:
+        if path.is_file():
+            shutil.copy2(path, dest)
+
+        if path.is_dir():
+            shutil.copytree(path, dest, dirs_exist_ok=True)