obtain node resource by model name (#70)

ergo · Mar 31, 2021 · 85cb03e · 85cb03e
1 parent a7405bf
commit 85cb03e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/ziggurat_foundations/models/services/resource_tree_postgres.py b/ziggurat_foundations/models/services/resource_tree_postgres.py
@@ -136,7 +136,7 @@ def from_parent_deeper(
         return query
 
     @classmethod
-    def build_subtree_strut(self, result, *args, **kwargs):
+    def build_subtree_strut(cls, result, *args, **kwargs):
         """
         Returns a dictionary in form of
         {node:Resource, children:{node_id: Resource}}
@@ -149,7 +149,8 @@ def build_subtree_strut(self, result, *args, **kwargs):
         if len(items) == 0:
             return root_elem
         for _, node in enumerate(items):
-            new_elem = {"node": node.Resource, "children": OrderedDict()}
+            node_res = getattr(node, cls.model.__name__)
+            new_elem = {"node": node_res, "children": OrderedDict()}
             path = list(map(int, node.path.split("/")))
             parent_node = root_elem
             normalized_path = path[:-1]