Add support for custom code generators

src/objdictgen/node.py

@@ -242,7 +242,10 @@ def DumpFile(self, filepath: TPath, filetype: str|None = "jsonc", custom_genfile
 
         if custom_genfile != None:
             log.debug("Invoking custom generator: %s", custom_genfile)
-            return executeCustomGenerator(custom_genfile, filepath, self)
+            try:
+                return executeCustomGenerator(custom_genfile, filepath, self)
+            except (ImportError, ModuleNotFoundError):
+                log.debug("Failed to load custom generator: %s", custom_genfile)
 
         if filetype == 'c':
             log.debug("Writing C files '%s'", filepath)

tests/test_node.py

@@ -1,11 +1,26 @@
 """Tests for node.py"""
 import pytest
+import os
 
-from objdictgen.node import Node
+from objdictgen.node import Node, NodeProtocol, executeCustomGenerator
 
 @pytest.mark.parametrize("file", ['master', 'slave'])
 def test_node_LoadFile(odpath, file):
 
     od = Node.LoadFile(odpath / (file + '.json'))
     assert isinstance(od, Node)
     assert od.Name == 'master' if file == 'master' else 'slave'
+
+@pytest.mark.parametrize("filename", ["this_file_does_not_exist.py", "bad_generator.py", "generator.py"])
+def test_node_execute_custom_generator(filename: str):
+    print(os.getcwd())
+    full_path = "tests/test_generators/"
+    mock_path = "/mock"
+    if filename == "bad_generator.py":
+        with pytest.raises(AttributeError):
+            executeCustomGenerator(full_path + filename, mock_path, NodeProtocol)
+    elif filename == "this_file_does_not_exist.py":
+        with pytest.raises(FileNotFoundError):
+            executeCustomGenerator(full_path + filename, mock_path, NodeProtocol)
+    elif filename == "generator.py":
+        assert executeCustomGenerator(full_path + filename, mock_path, NodeProtocol) == "success"