Add get nested trait

Author: danlooo

pathtraits/access.py

@@ -11,6 +11,32 @@
 logger = logging.getLogger(__name__)
 
 
+def nest_dict(flat_dict, delimiter="/"):
+    """
+    Transforms a flat dictionary with path-like keys into a nested dictionary.
+
+    :param flat_dict: The flat dictionary with path-like keys.
+    :param delimiter: The delimiter used in the keys (default is '/').
+    :return: A nested dictionary.
+    """
+    nested_dict = {}
+
+    for path, value in flat_dict.items():
+        keys = path.split(delimiter)
+        current = nested_dict
+
+        for key in keys[:-1]:
+            # If the key doesn't exist or is not a dictionary, create/overwrite it as a dictionary
+            if key not in current or not isinstance(current[key], dict):
+                current[key] = {}
+            current = current[key]
+
+        # Set the value at the final key
+        current[keys[-1]] = value
+
+    return nested_dict
+
+
 def get_dict(self, path):
     """
     Get traits for a path as a Python dictionary
@@ -40,6 +66,7 @@ def get_dict(self, path):
             if not (v and k != "path"):
                 continue
             res[k] = v
+    res = nest_dict(res)
     return res
 
 

pathtraits/db.py

@@ -54,10 +54,11 @@ def merge_rows(rows: list):
         for row in rows:
             for k, v in row.items():
                 # pylint: disable=C0201
-                if k in res.keys() and v not in res[k]:
+                if not k in res.keys():
+                    res[k] = []
+                if not v in res[k]:
                     res[k].append(v)
-                else:
-                    res[k] = [v]
+
         # simplify lists with just one element
         # ensure fixed order of list entries
         res = {k: sorted(v, key=str) if len(v) > 1 else v[0] for k, v in res.items()}

test/test.py

@@ -47,6 +47,7 @@ def test_eu(self):
             "is_example": True,
             "score": 3.5,
             "users": ["dloos", "fgans"],
+            "foo": {"bar": {"a": 1, "b": 2, "c": [1, 2, 3]}},
         }
         for k, v in target.items():
             self.assertEqual(source[k], v)