init commit

workflows/simbox/utils/task_config_parser.py

@@ -0,0 +1,41 @@
+from omegaconf import DictConfig, ListConfig, OmegaConf
+
+
+class AttrDict(dict):
+    """A dict subclass that supports both task['key'] and task.key access."""
+
+    def __getattr__(self, key):
+        try:
+            return self[key]
+        except KeyError:
+            raise AttributeError(key)
+
+    def __setattr__(self, key, value):
+        self[key] = value
+
+
+def _to_attr_dict(obj):
+    if isinstance(obj, dict):
+        return AttrDict({k: _to_attr_dict(v) for k, v in obj.items()})
+    if isinstance(obj, list):
+        return [_to_attr_dict(i) for i in obj]
+    return obj
+
+
+class TaskConfigParser:
+    """Shared utilities for workflow configuration parsing."""
+
+    def __init__(self, task_cfg_path: str):
+        self.task_cfg_path = task_cfg_path
+
+    def parse_tasks(self):
+        yaml_conf = OmegaConf.load(self.task_cfg_path)
+        task_cfgs = []
+        assert "tasks" in yaml_conf, f"Expected 'tasks' key in the task configuration file: {self.task_cfg_path}"
+        for task in yaml_conf["tasks"]:
+            if isinstance(task, (DictConfig, ListConfig)):
+                cfg = OmegaConf.to_container(task, resolve=True)
+            else:
+                cfg = task
+            task_cfgs.append(_to_attr_dict(cfg))
+        return task_cfgs