added save and load method to project and controls

Author: alexhroom

RATapi/controls.py

@@ -2,6 +2,7 @@
 import os
 import tempfile
 import warnings
+from pathlib import Path
 
 import prettytable
 from pydantic import (
@@ -220,3 +221,28 @@ def delete_IPC(self):
         with contextlib.suppress(FileNotFoundError):
             os.remove(self._IPCFilePath)
         return None
+
+    def save(self, path: str | Path, filename: str = "controls"):
+        """Save a controls object to a JSON file.
+
+        Parameters
+        ----------
+        path : str or Path
+            The directory in which the controls object will be written.
+
+        """
+        file = Path(path, f"{filename.removesuffix('.json')}.json")
+        file.write_text(self.model_dump_json())
+
+    @classmethod
+    def load(cls, path: str | Path) -> "Controls":
+        """Load a controls object from file.
+
+        Parameters
+        ----------
+        path : str or Path
+            The path to the controls object file.
+
+        """
+        file = Path(path)
+        return cls.model_validate_json(file.read_text())

RATapi/project.py

@@ -23,6 +23,7 @@
 
 import RATapi.models
 from RATapi.classlist import ClassList
+from RATapi.utils.convert import project_from_json, project_to_json
 from RATapi.utils.custom_errors import custom_pydantic_validation_error
 from RATapi.utils.enums import Calculations, Geometries, LayerModels, Priors, TypeOptions
 
@@ -834,6 +835,31 @@ def classlist_script(name, classlist):
             + "\n)"
         )
 
+    def save(self, path: str | Path, filename: str = "project"):
+        """Save a project to a JSON file.
+
+        Parameters
+        ----------
+        path : str or Path
+            The directory in which the project will be written.
+
+        """
+        file = Path(path, f"{filename.removesuffix('.json')}.json")
+        file.write_text(project_to_json(self))
+
+    @classmethod
+    def load(cls, path: str | Path) -> "Project":
+        """Load a project from file.
+
+        Parameters
+        ----------
+        path : str or Path
+            The path to the project file.
+
+        """
+        file = Path(path)
+        return project_from_json(file.read_text())
+
     def _classlist_wrapper(self, class_list: ClassList, func: Callable):
         """Defines the function used to wrap around ClassList routines to force revalidation.