Support namespaces root query/update APIs

Author: xxx7xxxx

.gitignore

@@ -178,4 +178,6 @@ models/**
 !models/pretrained/
 !models/pretrained/.gitkeep
 
-tb_logs/
+tb_logs/
+.namespaces_root.metadata.json
+easevoice_trainer_namespaces/

src/rest/rest.py

@@ -171,6 +171,18 @@ def _register_routes(self):
             status_code=204,  # No body in response
             summary="Delete a namespace",
         )
+        self.router.add_api_route(
+            path="/namespaces-root",
+            endpoint=self.get_namespaces_root,
+            methods=["GET"],
+            summary="Get namespaces root directory",
+        )
+        self.router.add_api_route(
+            path="/namespaces-root",
+            endpoint=self.set_namespaces_root,
+            methods=["POST"],
+            summary="Set namespaces root directory",
+        )
 
     async def list_namespaces(self):
         """List all namespaces."""
@@ -202,6 +214,17 @@ async def remove_namespace(self, name: str):
         except ValueError as e:
             raise HTTPException(status_code=404, detail=str(e))
 
+    async def get_namespaces_root(self):
+        """Get namespaces root directory."""
+        return self.namespace_service.get_namespaces_root_metadata()
+
+    async def set_namespaces_root(self, request: dict):
+        """Set namespaces root directory."""
+        try:
+            self.namespace_service.set_namespaces_root(request["namespaces-root"])
+        except ValueError as e:
+            raise HTTPException(status_code=400, detail=str(e))
+
 
 class FileAPI:
     """Encapsulated API logic for file operations."""

src/service/namespace.py

@@ -1,4 +1,5 @@
 import os
+import logging
 import json
 import shutil
 from datetime import datetime, timezone
@@ -8,15 +9,47 @@ class NamespaceService:
     """Manages namespaces on the filesystem."""
 
     def __init__(self, base_dir: Optional[str] = None):
-        self.base_dir = base_dir or os.path.join(os.path.expanduser("~"), "easevoice_trainer_namespaces")
-        os.makedirs(self.base_dir, exist_ok=True)
+        ns_root = os.getenv("EASEVOICE_TRAINER_NAMESPACES_ROOT", os.path.join(os.getcwd(), "easevoice_trainer_namespaces"))
+
+        # Fixed path to the metadata file for the namespaces root.
+        self.ns_root_metadata_path = os.path.join(os.getcwd(), ".namespaces_root.metadata.json")
+
+        # Create the default namespace root anyway in the beginning.
+        self.ns_root = ns_root
+        os.makedirs(self.ns_root, exist_ok=True)
+
+        # The first time saving does not count as a set-once operation.
+        self._save_ns_root_metadata(setOnce=False)
+
+    def _save_ns_root_metadata(self, setOnce: bool):
+        with open(self.ns_root_metadata_path, "w") as f:
+            json.dump({
+                "namespaces-root": self.ns_root,
+                "setOnce": setOnce,
+            }, f)
+
+    def get_namespaces_root_metadata(self) -> str:
+        with open(self.ns_root_metadata_path, "r") as f:
+            return json.load(f)
+
+    def set_namespaces_root(self, ns_root: str):
+        if os.path.exists(self.ns_root_metadata_path):
+            with open(self.ns_root_metadata_path, "r") as f:
+                metadata = json.load(f)
+                if metadata["setOnce"]:
+                    logging.warning("change namespaces root %s to %s while setOnce is true", self.ns_root, ns_root)
+
+        self.ns_root = ns_root
+        os.makedirs(self.ns_root, exist_ok=True)
+
+        self._save_ns_root_metadata(setOnce=True)
 
     def _namespace_metadata_path(self, name: str) -> str:
-        return os.path.join(self.base_dir, name, ".metadata.json")
+        return os.path.join(self.ns_root, name, ".metadata.json")
 
     def create_namespace(self, name: str) -> dict:
         """Create a new namespace, raising FileExistsError if it already exists."""
-        home_path = os.path.join(self.base_dir, name)
+        home_path = os.path.join(self.ns_root, name)
         if os.path.exists(home_path):
             raise FileExistsError("Namespace already exists")
 
@@ -33,8 +66,8 @@ def create_namespace(self, name: str) -> dict:
     def get_namespaces(self) -> List[dict]:
         """List all namespaces."""
         namespaces = []
-        for name in os.listdir(self.base_dir):
-            namespace_path = os.path.join(self.base_dir, name)
+        for name in os.listdir(self.ns_root):
+            namespace_path = os.path.join(self.ns_root, name)
             if os.path.isdir(namespace_path):
                 try:
                     namespaces.append(self._load_namespace_metadata(name))
@@ -44,8 +77,8 @@ def get_namespaces(self) -> List[dict]:
 
     def update_namespace(self, old_name: str, new_name: str) -> dict:
         """Rename a namespace, raising FileExistsError if the new name is taken."""
-        old_home_path = os.path.join(self.base_dir, old_name)
-        new_home_path = os.path.join(self.base_dir, new_name)
+        old_home_path = os.path.join(self.ns_root, old_name)
+        new_home_path = os.path.join(self.ns_root, new_name)
 
         if not os.path.exists(old_home_path):
             raise ValueError("Namespace not found")
@@ -64,7 +97,7 @@ def update_namespace(self, old_name: str, new_name: str) -> dict:
 
     def delete_namespace(self, name: str):
         """Delete a namespace, raising ValueError if it does not exist."""
-        home_path = os.path.join(self.base_dir, name)
+        home_path = os.path.join(self.ns_root, name)
         if not os.path.exists(home_path):
             raise ValueError("Namespace not found")