dir_tree.py

import os
from os.path import isdir

indent_amount = 4
collapse_amount = 10
indent_char = ' '
# indent_amount = 1
# collapse_amount = 5
# indent_char = '-'

indent_char *= indent_amount


def print_dir(path, _indent=0, dir_color=False):
    assert isdir(path), path
    if path in ['.', '.\\', './', '']:
        path = os.getcwd()
    if dir_color:
        print('\u001b[31m', end='')
    print(f'{indent_char * _indent}{path.split("/")[-1]}')
    if dir_color:
        print('\u001b[0m', end='')
    _indent += 1
    try:
        listdir_result = list(os.listdir(path))
        length = len([x for x in listdir_result if not isdir(f'{path}/{x}')])
        print_item_amount = length > collapse_amount
        if print_item_amount:
            print(f'{indent_char * _indent}{length} items')
        for x in listdir_result:
            full_path = f'{path}/{x}'
            if not isdir(full_path):
                if not print_item_amount:
                    print(indent_char * _indent + x)
            else:
                print_dir(full_path, _indent=_indent, dir_color=dir_color)
    except PermissionError:
        print(f'{indent_char*3}PERMISSION DENIED')


print_dir('./', dir_color=True)