using .txt file for saving and fixing flake8

Author: KingAkeem

torbot/main.py

@@ -77,7 +77,7 @@ def handle_tree_args(self, args):
 
         # -d/--download
         if args.download:
-            file_name = str(input("File Name (.pdf/.png/.svg): "))
+            file_name = str(input("File Name (.txt): "))
             tree.save(file_name)
 
     def perform_action(self):

torbot/modules/config.py

@@ -22,5 +22,6 @@ def get_log_level():
         if log_level_str == str_input:
             return log_level
 
+
 if not os.path.exists(data_directory):
     os.mkdir(data_directory)

torbot/modules/linktree.py

@@ -1,36 +1,41 @@
 """
 Module is used for analyzing link relationships
 """
-from treelib import Node, Tree, exceptions
+from treelib import Tree, exceptions
 
 from .api import get_node
 from .utils import join_local_path
-from .log import info, debug
+from .log import debug
+
 
 def formatNode(n):
     return f"{n['url']} {n['status_code']} {n['status']}"
 
+
 def build_tree_recursive(t, n):
 
+    current_id = n["url"]
     # this will only be ran on the root node since others will exist before being passed
-    if not t.contains(n["url"]):
-        debug(f"adding root node {n}")
-        t.create_node(formatNode(n), n["url"])
+    if not t.contains(current_id):
+        debug(f"adding id {current_id}")
+        t.create_node(formatNode(n), current_id)
 
+    children = n["children"]
     # if there are no children, there's nothing to process
-    if not n["children"]:
+    if not children:
         return
 
-    for child in n["children"]: 
-        id = child["url"]
+    for child in children:
+
+        current_id = child["url"]
         parent_id = n["url"]
+
         try:
-            debug(f"adding node {child}")
-            debug(f"parent_id {parent_id}")
-            t.create_node(formatNode(child), child["url"], parent=n["url"])
+            debug(f"adding child_id {current_id} to parent_id {parent_id}")
+            t.create_node(formatNode(child), current_id, parent=parent_id)
         except exceptions.DuplicatedNodeIdError:
-            debug(f"found a duplicate url {child['url']}")
-            continue # this node has already been processed somewhere else 
+            debug(f"found a duplicate url {current_id}")
+            continue  # this node has already been processed somewhere else
 
         build_tree_recursive(t, child)
 
@@ -53,21 +58,26 @@ def __build_tree(self, url: str, depth: int = 1):
         Returns:
             tree (ete3.Tree): Built tree.
         """
-        debug(f"building tree for {root} at {depth} depth")
+        debug(f"building tree for {url} at {depth} depth")
         n = get_node(url, depth)
         t = Tree()
         build_tree_recursive(t, n)
-        self._tree = t; 
+        self._tree = t
         debug("tree built successfully")
 
     def save(self, file_name: str):
         """
         Saves LinkTree to file with given file_name
-        Current file types supported are .png, .pdf, .svg
+        Current file types supported are .txt
         """
         debug(f"saving link tree as {file_name}")
         file_path = join_local_path(file_name)
-        self._tree.save2file(file_path)
+        try:
+            self._tree.save2file(file_path)
+        except Exception as e:
+            debug(f"failed to save link tree to {file_path}")
+            raise e
+
         debug(f"file saved successfully to {file_path}")
 
     def show(self):