fix: Reading from file on <scxml.datamodel.data.src> tag

Author: fgmacedo

statemachine/io/scxml/parser.py

@@ -2,6 +2,7 @@
 import xml.etree.ElementTree as ET
 from typing import Iterable
 from typing import Set
+from urllib.parse import urlparse
 
 from .schema import Action
 from .schema import AssignAction
@@ -106,12 +107,19 @@ def parse_datamodel(root: ET.Element) -> "DataModel | None":
 
     for datamodel_elem in root.findall(".//datamodel"):
         for data_elem in datamodel_elem.findall("data"):
+            content = data_elem.text and re.sub(r"\s+", " ", data_elem.text).strip() or None
+            src = data_elem.attrib.get("src")
+            src_parsed = urlparse(src) if src else None
+            if src_parsed and src_parsed.scheme == "file" and content is None:
+                with open(src_parsed.path) as f:
+                    content = f.read()
+
             data_model.data.append(
                 DataItem(
                     id=data_elem.attrib["id"],
-                    src=data_elem.attrib.get("src"),
+                    src=src_parsed,
                     expr=data_elem.attrib.get("expr"),
-                    content=data_elem.text and re.sub(r"\s+", " ", data_elem.text).strip() or None,
+                    content=content,
                 )
             )
 

statemachine/io/scxml/processor.py

@@ -1,3 +1,5 @@
+import os
+from contextlib import contextmanager
 from dataclasses import dataclass
 from pathlib import Path
 from typing import Any
@@ -19,6 +21,16 @@
 from .schema import Transition
 
 
+@contextmanager
+def temporary_directory(new_current_dir):
+    original_dir = os.getcwd()
+    try:
+        os.chdir(new_current_dir)
+        yield
+    finally:
+        os.chdir(original_dir)
+
+
 class IOProcessor:
     def __init__(self, processor: "SCXMLProcessor", machine: StateMachine):
         self.scxml_processor = processor
@@ -52,7 +64,8 @@ def __init__(self):
 
     def parse_scxml_file(self, path: Path):
         scxml_content = path.read_text()
-        return self.parse_scxml(path.stem, scxml_content)
+        with temporary_directory(path.parent):
+            return self.parse_scxml(path.stem, scxml_content)
 
     def parse_scxml(self, sm_name: str, scxml_content: str):
         definition = parse_scxml(scxml_content)

statemachine/io/scxml/schema.py

@@ -3,6 +3,7 @@
 from typing import Dict
 from typing import List
 from typing import Set
+from urllib.parse import ParseResult
 
 
 @dataclass
@@ -125,7 +126,7 @@ class State:
 @dataclass
 class DataItem:
     id: str
-    src: "str | None"
+    src: "ParseResult | None"
     expr: "str | None"
     content: "str | None"