feat[config.py]: Add initialisation from dict

ntc/__init__.py

@@ -1,4 +1,4 @@
 from .config import *
 from .errors import *
 
-__version__ = "0.1.2"
+__version__ = "0.1.3"

ntc/config.py

@@ -32,7 +32,7 @@ class CfgNode(UserDict):
 
     def __init__(
         self,
-        base: CfgNode = None,
+        base: dict = None,
         leaf_spec: Union[CfgLeaf, _CfgLeafSpec] = None,
         schema_frozen: bool = False,
         frozen: bool = False,
@@ -239,10 +239,18 @@ def _set_existing(self, key: str, value: Any) -> None:
                 )
             super().__setitem__(key, value)
 
-    def _init_with_base(self, base: CfgNode) -> None:
-        setattr(self, CfgNode._LEAF_SPEC, base.leaf_spec)
-        self._set_attrs(base.attrs)
-        self.freeze_schema()
+    def _init_with_base(self, base: dict) -> None:
+        if isinstance(base, CfgNode):
+            setattr(self, CfgNode._LEAF_SPEC, base.leaf_spec)
+            self._set_attrs(base.attrs)
+            self.freeze_schema()
+        elif isinstance(base, dict):
+            for key, value in base.items():
+                if isinstance(value, dict):
+                    value = CfgNode(value)
+                self[key] = value
+        else:
+            raise ValueError(f"Unknown base format {base}")
 
 
 class CfgLeaf:

tests/test_config.py

@@ -164,3 +164,9 @@ def test_get():
         cfg["BAR"]
     assert cfg.get("BAR") is None
     assert cfg.get("BAR", "foo") == "foo"
+
+
+def test_init_from_dict():
+    cfg = CfgNode({"FOO": "bar", "BAR": {"BAZ": "foo"}})
+    assert cfg.FOO == "bar"
+    assert cfg.BAR.BAZ == "foo"