Add the capability to upgrade a .ttl file version using AdaptiveGraph

Author: Oleksandr Muzyka (EPAM)

core/esmf-aspect-meta-model-python/esmf_aspect_meta_model_python/adaptive_graph.py

@@ -0,0 +1,169 @@
+import pathlib
+import subprocess
+import tempfile
+
+from typing import Optional, Union
+
+from rdflib import Graph
+
+from esmf_aspect_meta_model_python.samm_cli import SammCli
+
+
+class AdaptiveGraph(Graph):  # TODO: avoid double parse when upgrading
+    """An RDF graph that can adaptively upgrade SAMM files using the SAMM CLI."""
+
+    _samm_namespace_prefix = "samm"  # TODO: take out to a common place
+    _samm_org_identifier = "org.eclipse.esmf.samm"
+
+    def __init__(self, samm_version: str | None = None, *args, **kwargs) -> None:
+        super().__init__(*args, **kwargs)
+
+        self._samm_version = samm_version
+        self._samm_cli = SammCli()
+
+    def _upgrade_ttl_file(self, file_path: pathlib.Path) -> str:
+        """Run SAMM CLI prettyprint to upgrade a TTL file to the latest version."""
+        try:
+            return self._samm_cli.prettyprint(str(file_path), capture=True)
+        except subprocess.CalledProcessError as e:
+            raise RuntimeError(f"SAMM CLI failed for {file_path}:\n{e.stdout}\n{e.stderr}") from e
+
+    def _has_version_mismatch_in_graph(self, input_source: str | bytes | pathlib.PurePath) -> bool:
+        """Parse into a temporary graph to detect mismatch BEFORE loading into self."""
+        temp_graph = Graph()
+
+        if isinstance(input_source, pathlib.Path):
+            temp_graph.parse(input_source)
+        else:
+            temp_graph.parse(data=input_source, format="ttl")  # type: ignore[arg-type]
+
+        for prefix, namespace in temp_graph.namespace_manager.namespaces():
+            if prefix.startswith(self._samm_namespace_prefix):
+                parts = namespace.strip("#").split(":")
+                if len(parts) >= 5 and parts[2] == self._samm_org_identifier:
+                    version = parts[-1]
+                    if version != self._samm_version:
+                        return True
+        return False
+
+    def _upgrade_source(self, source: str | pathlib.PurePath) -> str:
+        source_path = pathlib.Path(source)
+
+        print(f"[INFO] SAMM version mismatch detected in {source_path}. Upgrading...")
+
+        return self._upgrade_ttl_file(source_path)
+
+    def _upgrade_data(self, data: str | bytes) -> str:
+        print(  # TODO: improve logging
+            f"[INFO] SAMM version mismatch detected in provided data (target v{self._samm_version}) Upgrading..."
+        )
+
+        with tempfile.NamedTemporaryFile("wb", suffix=".ttl", delete=False) as tmp:
+            tmp.write(data.encode("utf-8") if isinstance(data, str) else data)
+            tmp_path = pathlib.Path(tmp.name)
+
+        try:
+            return self._upgrade_ttl_file(tmp_path)
+        finally:
+            tmp_path.unlink(missing_ok=True)
+
+    @classmethod
+    def get_samm_version_from_input(cls, stream_input: pathlib.PurePath | str) -> str:
+        """Extract the SAMM version from the given TTL input (file path or string)."""
+        temp_graph = Graph()
+        parse_kwargs: dict[str, str | pathlib.PurePath] = {}
+        version = ""
+
+        if isinstance(stream_input, pathlib.PurePath):
+            parse_kwargs.update(source=stream_input, format="turtle")
+        else:
+            parse_kwargs.update(data=stream_input)
+
+        temp_graph.parse(**parse_kwargs)  # type: ignore[arg-type]
+
+        for prefix, namespace in temp_graph.namespace_manager.namespaces():
+            if prefix.startswith(cls._samm_namespace_prefix):
+                urn_parts = namespace.split(":")
+                current_version = urn_parts[-1].replace("#", "")
+
+                version = max(version, current_version)
+
+        return version
+
+    def set_samm_version(self, samm_version: str | None) -> None:
+        """Set the SAMM version for this graph."""
+        self._samm_version = samm_version
+
+    def parse(  # type: ignore[override]
+        self,
+        *,
+        source: Optional[pathlib.PurePath] = None,
+        data: Optional[str | bytes] = None,
+        **kwargs,
+    ) -> "AdaptiveGraph":
+        """
+        Parse a TTL file into this graph, upgrading via SAMM CLI if version mismatch detected.
+
+        If a SAMM version mismatch is detected, the TTL file will be upgraded using the SAMM CLI prettyprint
+        before parsing into this graph.
+
+        Args:
+            source: Path to the TTL file as pathlib.Path.
+            data: RDF content as string or bytes.
+            *args, **kwargs: Additional arguments passed to rdflib.Graph.parse().
+
+        Returns:
+            self (AdaptiveGraph): The current graph instance with parsed data.
+
+        Raises:
+            RuntimeError: If the SAMM CLI fails during the upgrade process.
+            ValueError: If neither 'source' nor 'data' is provided, or if SAMM version is not set.
+        """
+        if (source is None) == (data is None):
+            raise ValueError("Either 'source' or 'data' must be provided.")
+        if not self._samm_version:
+            raise ValueError("SAMM version is not set.")
+
+        if self._has_version_mismatch_in_graph(input_source=data or pathlib.Path(source)):  # type: ignore[arg-type]
+            data = self._upgrade_data(data) if data else self._upgrade_source(source)  # type: ignore[arg-type]
+            source = None
+
+        super().parse(source=source, data=data, **kwargs)
+
+        return self
+
+    def __add__(self, other: Union["Graph", "AdaptiveGraph"]) -> "AdaptiveGraph":
+        """Override addition to propagate SAMM version to the resulting graph."""
+        retval = super().__add__(other)
+
+        if isinstance(retval, AdaptiveGraph):
+            if isinstance(other, AdaptiveGraph) and other._samm_version != self._samm_version:
+                raise ValueError("SAMM version mismatch during addition.")
+
+            retval.set_samm_version(self._samm_version)
+
+        return retval  # type: ignore[return-value]
+
+    def __sub__(self, other: Union["Graph", "AdaptiveGraph"]) -> "AdaptiveGraph":
+        """Override subtraction to propagate SAMM version to the resulting graph."""
+        retval = super().__sub__(other)
+
+        if isinstance(retval, AdaptiveGraph):
+            if isinstance(other, AdaptiveGraph) and other._samm_version != self._samm_version:
+                raise ValueError("SAMM version mismatch during subtraction.")
+
+            retval.set_samm_version(self._samm_version)
+
+        return retval  # type: ignore[return-value]
+
+    def __mul__(self, other: Union["Graph", "AdaptiveGraph"]) -> "AdaptiveGraph":
+        """Override multiplication to propagate SAMM version to the resulting graph."""
+        retval = super().__mul__(other)
+
+        if isinstance(retval, AdaptiveGraph):
+            if isinstance(other, AdaptiveGraph) and other._samm_version != self._samm_version:
+                raise ValueError("SAMM version mismatch during multiplication.")
+
+            retval.set_samm_version(self._samm_version)
+
+        return retval  # type: ignore[return-value]

core/esmf-aspect-meta-model-python/esmf_aspect_meta_model_python/loader/samm_graph.py

@@ -14,6 +14,7 @@
 
 from rdflib import RDF, Graph, Node
 
+from esmf_aspect_meta_model_python.adaptive_graph import AdaptiveGraph
 from esmf_aspect_meta_model_python.base.aspect import Aspect
 from esmf_aspect_meta_model_python.base.base import Base
 from esmf_aspect_meta_model_python.base.property import Property
@@ -29,9 +30,10 @@ class SAMMGraph:
     """Class representing the SAMM graph and its operations."""
 
     samm_namespace_prefix = "samm"
+    samm_org_identifier = "org.eclipse.esmf.samm"
 
     def __init__(self):
-        self.rdf_graph = Graph()
+        self.rdf_graph = AdaptiveGraph()
         self.samm_graph = Graph()
         self._cache = DefaultElementCache()
 
@@ -59,7 +61,7 @@ def _get_rdf_graph(self, input_data: Union[str, Path], input_type: Optional[str]
         """Read the RDF graph from the given input data.
 
         This method initializes the `InputHandler` with the provided input data and type,
-        retrieves the reader, and reads the RDF graph into `self.rdf_graph`.
+        retrieves the reader, sets the SAMM version, and reads the RDF graph into `self.rdf_graph`.
 
         Args:
             input_data (Union[str, Path]): The input data to read the RDF graph from. This can be a file path or a str.
@@ -69,25 +71,26 @@ def _get_rdf_graph(self, input_data: Union[str, Path], input_type: Optional[str]
             None
         """
         self._reader = InputHandler(input_data, input_type).get_reader()
+        self._reader.set_samm_version(input_data)
         self.rdf_graph = self._reader.read(input_data)
 
-    def _get_samm_version_from_rdf_graph(self) -> str:
-        """Extracts the SAMM version from the RDF graph.
-
-        This method searches through the RDF graph namespaces to find a prefix that indicates the SAMM version.
-
-        Returns:
-            str: The SAMM version as a string extracted from the graph. Returns an empty string if no version
-                 can be conclusively identified.
-        """
-        version = ""
-
-        for prefix, namespace in self.rdf_graph.namespace_manager.namespaces():
-            if prefix == self.samm_namespace_prefix:
-                urn_parts = namespace.split(":")
-                version = urn_parts[-1].replace("#", "")
-
-        return version
+    # def _get_samm_version_from_rdf_graph(self) -> str:
+    #     """Extracts the SAMM version from the RDF graph.
+    #
+    #     This method searches through the RDF graph namespaces to find a prefix that indicates the SAMM version.
+    #
+    #     Returns:
+    #         str: The SAMM version as a string extracted from the graph. Returns an empty string if no version
+    #              can be conclusively identified.
+    #     """
+    #     version = ""
+    #
+    #     for prefix, namespace in self.rdf_graph.namespace_manager.namespaces():
+    #         if prefix == self.samm_namespace_prefix:
+    #             urn_parts = namespace.split(":")
+    #             version = urn_parts[-1].replace("#", "")
+    #
+    #     return version
 
     def _get_samm_version(self):
         """Retrieve and set the SAMM version from the RDF graph.
@@ -98,7 +101,7 @@ def _get_samm_version(self):
         Raises:
             ValueError: If the SAMM version is not found in the RDF graph.
         """
-        self.samm_version = self._get_samm_version_from_rdf_graph()
+        self.samm_version = self._reader.samm_version
 
         if not self.samm_version:
             raise ValueError(
@@ -239,7 +242,7 @@ def _validate_samm_namespace_version(self, graph: Graph) -> None:
             if prefix.startswith(self.samm_namespace_prefix):
                 namespace_info = namespace.split(":")  # [urn, namespace_id, namespace_specific_str, entity, version#]
 
-                if len(namespace_info) == 5 and namespace_info[2] == "org.eclipse.esmf.samm":
+                if len(namespace_info) == 5 and namespace_info[2] == self.samm_org_identifier:
                     version = namespace_info[-1].replace("#", "")
 
                     if version != self.samm_version:

core/esmf-aspect-meta-model-python/esmf_aspect_meta_model_python/resolver/base.py

@@ -13,8 +13,7 @@
 from pathlib import Path
 from typing import Union
 
-from rdflib import Graph
-
+from esmf_aspect_meta_model_python.adaptive_graph import AdaptiveGraph
 from esmf_aspect_meta_model_python.samm_meta_model import SammUnitsGraph
 
 
@@ -33,7 +32,7 @@ class ResolverInterface(ABC):
     """
 
     def __init__(self):
-        self.graph = Graph()
+        self.graph = AdaptiveGraph()
         self.samm_graph = None
         self.samm_version = ""
 
@@ -70,46 +69,63 @@ def _validate_samm_version(samm_version: str):
         elif samm_version > SammUnitsGraph.SAMM_VERSION:
             raise ValueError(f"{samm_version} is not supported SAMM version.")
 
-    def _get_samm_version_from_graph(self) -> str:
-        """
-        Extracts the SAMM version from the RDF graph.
-
-        This method searches through the RDF graph namespaces to find a prefix that indicate the SAMM version.
-
-        Returns:
-            str: The SAMM version as a string extracted from the graph. Returns an empty string if no version
-                 can be conclusively identified.
-        """
-        version = ""
-
-        for prefix, namespace in self.graph.namespace_manager.namespaces():
-            if prefix == "samm":
-                urn_parts = namespace.split(":")
-                version = urn_parts[-1].replace("#", "")
-
-        return version
-
-    def get_samm_version(self) -> str:
+    # def _get_samm_version_from_graph(self) -> str:
+    #     """
+    #     Extracts the SAMM version from the RDF graph.
+    #
+    #     This method searches through the RDF graph namespaces to find a prefix that indicate the SAMM version.
+    #
+    #     Returns:
+    #         str: The SAMM version as a string extracted from the graph. Returns an empty string if no version
+    #              can be conclusively identified.
+    #     """
+    #     version = ""
+    #
+    #     for prefix, namespace in self.graph.namespace_manager.namespaces():
+    #         if prefix == "samm":
+    #             urn_parts = namespace.split(":")
+    #             version = urn_parts[-1].replace("#", "")
+    #
+    #     return version
+
+    # def get_samm_version(self) -> str:
+    #     """
+    #     Retrieves and validates the specified SAMM version from the provided Aspect model graph.
+    #
+    #     This method attempts to extract the version information of the SAMM from a graph. There is also a validation
+    #     against known SAMM versions to ensure the version is supported and recognized.
+    #
+    #
+    #     Returns:
+    #         str: The validated version of SAMM if it is recognized and supported. If the version is not valid,
+    #              an appropriate message or value indicating non-recognition is returned.
+    #
+    #     Raises:
+    #         ValueError: If the extracted version is not supported or if it is not found in the Graph.
+    #
+    #     """
+    #     version = self._get_samm_version_from_graph()
+    #     self._validate_samm_version(version)
+    #     self.samm_version = version
+    #
+    #     return version
+
+    def set_samm_version(self, steam_input: Union[str, Path]) -> None:
         """
-        Retrieves and validates the specified SAMM version from the provided Aspect model graph.
+        Sets the SAMM version by extracting it from the specified file.
 
-        This method attempts to extract the version information of the SAMM from a graph. There is also a validation
-        against known SAMM versions to ensure the version is supported and recognized.
+        This method uses the AdaptiveGraph class to extract the SAMM version from the given file.
+        There is also a validation against known SAMM versions to ensure the version is supported and recognized.
 
-
-        Returns:
-            str: The validated version of SAMM if it is recognized and supported. If the version is not valid,
-                 an appropriate message or value indicating non-recognition is returned.
+        Args:
+            steam_input (Union[str, Path]): The path to the file from which the SAMM version is to be extracted.
 
         Raises:
-            ValueError: If the extracted version is not supported or if it is not found in the Graph.
-
+            ValueError: If the extracted version is not supported or if it is not found in the file.
         """
-        version = self._get_samm_version_from_graph()
+        version = AdaptiveGraph.get_samm_version_from_input(steam_input)
         self._validate_samm_version(version)
         self.samm_version = version
 
-        return version
-
-    def prepare_aspect_model(self, graph: Graph):
+    def prepare_aspect_model(self, graph: AdaptiveGraph):
         """Resolve all additional graph elements if needed."""

core/esmf-aspect-meta-model-python/esmf_aspect_meta_model_python/resolver/data_string.py

@@ -12,8 +12,7 @@
 from pathlib import Path
 from typing import Union
 
-from rdflib import Graph
-
+from esmf_aspect_meta_model_python.adaptive_graph import AdaptiveGraph
 from esmf_aspect_meta_model_python.resolver.base import ResolverInterface
 
 
@@ -33,7 +32,7 @@ def read(self, data_string: Union[str, Path]):
         Returns:
             RDFGraph: An object representing the RDF graph constructed from the input data.
         """
-        self.graph = Graph()
+        self.graph = AdaptiveGraph(samm_version=self.samm_version)
         self.graph.parse(data=str(data_string) if isinstance(data_string, Path) else data_string)
 
         return self.graph