Quickly get a flat or nested directory structure, and optionally save to yaml

map_dir_structure.py

import os
import yaml
from pprint import pprint

def dir_structure(dir_path, nested=False):
    result = []

    if nested:
        # return directory structure as nested dict
        for item in os.listdir(dir_path):
            item_path = os.path.join(dir_path, item)
            if os.path.isdir(item_path):
                result.append({item: list_dir(item_path, nested=nested)})
            else:
                result.append(item)
    else:       
        # return directory structure as flat file list
        for root, _, files in os.walk(dir_path):
            for file in files:
                full_path = os.path.join(root, file)
                result.append(full_path)

    return result

flat_dir_structure = dir_structure('..')
nested_dir_structure = dir_structure('..', nested=True)

print(flat_dir_structure)
pprint(nested_dir_structure)

with open("directory_structure.yaml", "w") as f:
    yaml.dump(nested_dir_structure, f, default_flow_style=False)