From af101d57fe608880e595a47206375e9cc262c270 Mon Sep 17 00:00:00 2001 From: Wu Clan Date: Tue, 2 Jul 2024 08:49:40 +0800 Subject: [PATCH] Update the tree structure traversal algorithm (#338) --- backend/utils/build_tree.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/backend/utils/build_tree.py b/backend/utils/build_tree.py index 5037beff..8fb2291a 100644 --- a/backend/utils/build_tree.py +++ b/backend/utils/build_tree.py @@ -35,7 +35,11 @@ def traversal_to_tree(nodes: list[dict[str, Any]]) -> list[dict[str, Any]]: if parent_node is not None: if 'children' not in parent_node: parent_node['children'] = [] - parent_node['children'].append(node) + if node not in parent_node['children']: + parent_node['children'].append(node) + else: + if node not in tree: + tree.append(node) return tree