Merge pull request #214 from DedSecInside/fix_tree_features

Fix the basic functionality of tree features
DedSecInside · Jul 9, 2021 · 41a2d9f · 41a2d9f
2 parents f95eacc + 663523d
commit 41a2d9f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/src/modules/analyzer.py b/src/modules/analyzer.py
@@ -50,7 +50,7 @@ def save(self, file_name, tree_style=default_style):
             tree_style (TreeStyle): Styling of downloaded tree
         """
         self._tree.layout_fn = default_layout
-        self._tree.render(file_name, tree_style)
+        self._tree.render(file_name, tree_style=tree_style)
 
     def show(self, tree_style=default_style):
         """

diff --git a/src/modules/link.py b/src/modules/link.py
@@ -85,6 +85,10 @@ def __init__(self, link):
         self._loaded = False
         self._name = link
         self._link = link
+        self._emails = []
+        self._images = []
+        self._children = []
+        self._json_data = {}
 
     def load_data(self):
         response = requests.get(self._link)