Add package-level set_error_handling() to fairgraph.openminds

Author: apdavison

builder/update_openminds.py

@@ -943,8 +943,13 @@ def main(openminds_root, ignore=[]):
         )
 
     init_file_path = os.path.join("..", "fairgraph", "openminds", "__init__.py")
+    sorted_modules = sorted(openminds_modules)
     with open(init_file_path, "w") as fp:
-        fp.write(f"from . import ({', '.join(sorted(openminds_modules))})\n")
+        fp.write(f"from . import ({', '.join(sorted_modules)})\n\n\n")
+        fp.write("def set_error_handling(value):\n")
+        fp.write('    """Set error handling for all openMINDS classes, across every submodule."""\n')
+        fp.write(f"    for module in ({', '.join(sorted_modules)}):\n")
+        fp.write("        module.set_error_handling(value)\n")
 
     # Format with Black
     subprocess.call([sys.executable, "-m", "black", "--quiet", target_path])

fairgraph/__init__.py

@@ -47,15 +47,4 @@
 
 def set_error_handling(value):
     """Set error handling globally for all modules"""
-    for module in (
-        openminds.chemicals,
-        openminds.computation,
-        openminds.controlled_terms,
-        openminds.core,
-        openminds.ephys,
-        openminds.publications,
-        openminds.sands,
-        openminds.specimen_prep,
-        openminds.stimulation,
-    ):
-        module.set_error_handling(value)
+    openminds.set_error_handling(value)

fairgraph/openminds/__init__.py

@@ -1 +1,17 @@
 from . import chemicals, computation, controlled_terms, core, ephys, publications, sands, specimen_prep, stimulation
+
+
+def set_error_handling(value):
+    """Set error handling for all openMINDS classes, across every submodule."""
+    for module in (
+        chemicals,
+        computation,
+        controlled_terms,
+        core,
+        ephys,
+        publications,
+        sands,
+        specimen_prep,
+        stimulation,
+    ):
+        module.set_error_handling(value)