Treescriptify:

*u tree_reader.py
*u tree_runner.py

Author: DK96-OS

treescriptify/tree_reader.py

@@ -7,12 +7,21 @@
 
 
 def generate_from_json(json_string: str) -> Generator[TreeNodeData, None, None]:
+    """Read the JSON string and generate TreeNodeData for all elements.
     """
-    """
-    tree_son, report = json.loads(json_string)
-    tree_son = tree_son['contents']
-    depth = 0
-    # todo:
-    for i in tree_son:
+    full_json = json.loads(json_string)[0]
+    for i in full_json['contents']:
+        for node in _process_node(i, 0):
+            yield node
+
 
-    pass
+def _process_node(node: dict, depth: int) -> Generator[TreeNodeData, None, None]:
+    """
+    """
+    if node['type'] == 'file':
+        yield TreeNodeData(depth, False, node['name'])
+    else:
+        yield TreeNodeData(depth, True, node['name'])
+        for nested_nodes in node.get('contents', []):
+            for node in _process_node(nested_nodes, depth + 1):
+                yield node

treescriptify/tree_runner.py

@@ -9,26 +9,27 @@ def get_tree_json(data: InputData) -> str:
     """Obtain the Tree Information as a JSON string.
     """
     result = subprocess.run(
-        args=['tree', '-J', '-i', '-x', '--noreport'] + _check_arguments(data),
+        args='tree -Jix --noreport ' + _check_arguments(data),
         capture_output=True,
         text=True,
-        shell=False,
+        shell=True,
+        timeout=3
     )
     output = result.stdout
     #error = result.stderr
     return output
 
 
-def _check_arguments(data: InputData) -> list[str]:
+def _check_arguments(data: InputData) -> str:
     """Check the Input Data and map flags to tree command.
     """
     extras = []
     if data.directories_only:
-        extras += '-d'
+        extras.append('-d ')
     if data.prune_dirs:
-        extras += '--prune'
+        extras += ['--prune ']
     if data.include_hidden:
-        extras += '-a'
+        extras += ['-a ']
     if data.git_ignore:
-        extras += '--gitignore'
-    return extras
+        extras.append('--gitignore ')
+    return ' '.join(extras)