Print filesystem tree, similar to command line tool `tree` in UNIX.

tree.py

#!/usr/bin/env python
import argparse
import os

def tree(directory=".", indent = "    "):
    """
    Helper function that prints the filesystem tree.
    """
    ignorables = ["__pycache__", ".gitignore", ".DS_Store", ".ipynb_checkpoints", ".git", "venv"]
    for root, dirs, files in os.walk(directory):
        path = root.split(os.sep)
        if any(ign in path for ign in ignorables):
            continue
        print((len(path) - 1) * indent, os.path.basename(root) + "/")
        for file in files:
            if file in ignorables:
                continue
            print(len(path) * indent, file)
            

if __name__ == "__main__":
    parser = argparse.ArgumentParser(prog = 'tree', description = 'list contents of directories in a tree-like format.')
    parser.add_argument('directory', nargs='?', default=".")
    args = parser.parse_args()
    tree(args.directory)