Improve parse tree

src/parse_tree/parse_tree.py

@@ -1,5 +1,5 @@
 import warnings
-from typing import List
+from typing import List, Dict
 
 from anytree import Node, RenderTree
 
@@ -23,10 +23,21 @@ class ParseTree:
         return nodes
 
     def __create_tree(self, nodes: List[ParseItem]) -> Node:
-        root: Node
+        tree_nodes_count: int = 0
+        parents: Dict[int, Node] = {}
+        root = None
+        while tree_nodes_count < len(nodes):
             for node in nodes:
+                if parents.get(node.index()) is not None:
+                    continue
+                parent: Node = parents.get(node.parent_index())
+                if parent is not None or node.parent_index() == 0:
+                    new_node = Node(node, parent)
+                    parents[node.index()] = new_node
+                    tree_nodes_count = tree_nodes_count + 1
                     if node.parent_index() == 0:
-                root = Node(node)
+                        root = new_node
+                    break
         return root
 
     def __repr__(self) -> str:

src/parse_tree/parse_tree_item.py

@@ -9,6 +9,9 @@ class ParseItem:
         self._parent_index: int = int(parsed_str[6])
         self._relation: str = parsed_str[7]
 
+    def index(self) -> int:
+        return self._index
+
     def parent_index(self) -> int:
         return self._parent_index