Misc refactor (#396)

* feat: construct trees with nested_dict_key_to_tree

* refactor: test clean up

* feat: export trees with tree_to_nested_dict_key

* feat: add docs

* docs: update CHANGELOG

* docs: enhance docstring

* refactor: abstract out export tree logic

* docs: update CHANGELOG

Author: kayjan

CHANGELOG.md

@@ -5,6 +5,8 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
 ## [Unreleased]
+### Changed
+- Misc: Some code refactoring
 
 ## [0.30.0] - 2025-09-05
 ### Added:

bigtree/tree/export/dictionaries.py

@@ -13,6 +13,34 @@
 T = TypeVar("T", bound=node.Node)
 
 
+def _assemble_attributes(
+    _node: T,
+    attr_dict: Optional[Dict[str, str]],
+    all_attrs: bool,
+    data_child: Dict[str, Any] = None,
+) -> Dict[str, Any]:
+    """Assemble attributes of node into a dictionary.
+
+    Args:
+        _node: node
+        attr_dict: node attributes mapped to dictionary key, key: node attributes, value: corresponding dictionary key
+        all_attrs: indicator whether to retrieve all ``Node`` attributes, overrides `attr_dict`
+        data_child: existing attributes, if any
+
+    Returns:
+        node attributes
+    """
+    data_child = data_child or {}
+    if all_attrs:
+        data_child.update(
+            dict(_node.describe(exclude_attributes=["name"], exclude_prefix="_"))
+        )
+    elif attr_dict:
+        for k, v in attr_dict.items():
+            data_child[v] = _node.get_attr(k)
+    return data_child
+
+
 def tree_to_dict(
     tree: T,
     name_key: Optional[str] = "name",
@@ -79,17 +107,9 @@ def _recursive_append(_node: T) -> None:
                     if _node.parent:
                         parent_name = _node.parent.node_name
                     data_child[parent_key] = parent_name
-                if all_attrs:
-                    data_child.update(
-                        dict(
-                            _node.describe(
-                                exclude_attributes=["name"], exclude_prefix="_"
-                            )
-                        )
-                    )
-                elif attr_dict:
-                    for k, v in attr_dict.items():
-                        data_child[v] = _node.get_attr(k)
+                data_child = _assemble_attributes(
+                    _node, attr_dict, all_attrs, data_child
+                )
                 data_dict[_node.path_name] = data_child
             for _child in _node.children:
                 _recursive_append(_child)
@@ -145,17 +165,9 @@ def _recursive_append(_node: T, parent_dict: Dict[str, Any]) -> None:
         if _node:
             if not max_depth or _node.depth <= max_depth:
                 data_child = {name_key: _node.node_name}
-                if all_attrs:
-                    data_child.update(
-                        dict(
-                            _node.describe(
-                                exclude_attributes=["name"], exclude_prefix="_"
-                            )
-                        )
-                    )
-                elif attr_dict:
-                    for k, v in attr_dict.items():
-                        data_child[v] = _node.get_attr(k)
+                data_child = _assemble_attributes(
+                    _node, attr_dict, all_attrs, data_child
+                )
                 if child_key in parent_dict:
                     parent_dict[child_key].append(data_child)
                 else:
@@ -212,18 +224,7 @@ def _recursive_append(_node: T, parent_dict: Dict[str, Any]) -> None:
         """
         if _node:
             if not max_depth or _node.depth <= max_depth:
-                data_child = {}
-                if all_attrs:
-                    data_child.update(
-                        dict(
-                            _node.describe(
-                                exclude_attributes=["name"], exclude_prefix="_"
-                            )
-                        )
-                    )
-                elif attr_dict:
-                    for k, v in attr_dict.items():
-                        data_child[v] = _node.get_attr(k)
+                data_child = _assemble_attributes(_node, attr_dict, all_attrs)
                 if child_key in parent_dict:
                     parent_dict[child_key][_node.node_name] = data_child
                 else: