FIX: rdf:ID formatting for CYME compatible export

Author: pseudocubic

ravens/xml/graph.py

@@ -1,4 +1,5 @@
 import pathlib
+import re
 
 from uuid import uuid4
 
@@ -32,6 +33,23 @@ def __init__(self, profile_path: str | pathlib.Path | None = None, cim_namespace
         nm.bind("cim", self.cim, override=True)
         self.graph.namespace_manager = nm
 
+    @staticmethod
+    def _transform_to_cyme_uri(match):
+        """
+        Transform URIs to CYME format: #uuid -> #_UUID
+        Match patterns like rdf:about="#uuid" or rdf:resource="#uuid"
+        """
+        prefix = match.group(1)  # rdf:about=" or rdf:resource="
+        hash_sign = match.group(2)  # # or empty
+        uuid = match.group(3)  # the UUID
+
+        # Transform: remove leading underscore if present, uppercase, add underscore
+        uuid_clean = uuid.lstrip('_')
+        uuid_transformed = f"_{uuid_clean.upper()}"
+
+        return f'{prefix}{hash_sign}{uuid_transformed}"'
+
+
     def mRID(self) -> str:
         return self.uuid_format(str(uuid4()))
 
@@ -82,8 +100,19 @@ def get(self, subject, predicate, default=None):
     def get_name(self, subject):
         return self.get(subject, self.cim["IdentifiedObject.name"], str(subject))
 
-    def save(self, path: pathlib.Path | str) -> None:
+    def save(self, path: pathlib.Path | str, make_cyme_compatible: bool = False) -> None:
         rdfxml = self.graph.serialize(max_depth=1, format="pretty-xml")
-        rdfxml = rdfxml.replace("rdf:about", "rdf:ID")
+
+        if make_cyme_compatible:
+            # Transform both rdf:about and rdf:resource attributes
+            rdfxml = re.sub(
+                r'(rdf:(?:about|resource)=")(#?)([a-fA-F0-9_-]+)"',
+                self._transform_to_cyme_uri,
+                rdfxml
+            )
+
+            # Change rdf:about to rdf:ID
+            rdfxml = rdfxml.replace("rdf:about", "rdf:ID")
+
         with open(path, "w") as f:
             f.write(rdfxml)