PythonScripts/PrintHierarchy.py at main · jonserr/PythonScripts · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
#!/usr/local/bin python3
## ---------------------------
## Script name: PrintHierarchy.py
## Purpose: Input file paths and print out file heirarchy tree
## Author: Jonathan Serrano
## Copyright (c) Jonathan Serrano, 2023
## ---------------------------

#!/usr/local/bin python3
## ---------------------------
## Script name: PrintHierarchy.py
## Purpose: Input file paths and print out file heirarchy tree
## Author: Jonathan Serrano
## Copyright (c) Jonathan Serrano, 2023
## ---------------------------

import sys
import os

def add_to_tree(tree, path):
    if len(path) == 1:
        tree[path[0]] = {}
    else:
        add_to_tree(tree.setdefault(path[0], {}), path[1:])

def print_tree(tree, level=0, is_last=False, output_file=None):
    prefix = "|   " * (level - 1) + "|-- " if level > 0 else ""
    for i, (k, v) in enumerate(tree.items()):
        is_last = i == len(tree) - 1
        line = f"{prefix}{k}\n"
        output_file.write(line)
        print(line, end="")
        if isinstance(v, dict):
            print_tree(v, level + 1, is_last, output_file)

def main(input_paths):
    tree = {}

    for path in input_paths:
        if os.path.exists(path):
            add_to_tree(tree, path.split("/"))
        else:
            print(f"Warning: File or directory '{path}' does not exist.")

    home_dir = os.path.expanduser("~")
    output_file_path = os.path.join(home_dir, "tree_output.txt")

    with open(output_file_path, "w") as output_file:
        print_tree(tree, output_file=output_file)

    print(f"Tree output saved to: {output_file_path}")

def test():
    input_paths = ["dir1/dir2/file1.txt", "dir1/dir3/file2.txt", "dir4/file3.txt"]
    main(input_paths)

if __name__ == "__main__":
    input_paths = sys.argv[1:]
    main(input_paths)

    # Uncomment the line below to enable the test
    # test()