view.py

from humanize import naturalsize
from directory import DirectoryInfo
from screen import Screen


def list_view(dir, all=False, summarize=False, measure=False, depth=None, fullpath=False, max_count=None, window=False):
    root = None if fullpath else dir.path
    if max_count is None:
        if summarize:
            walk = [dir]
        else:
            walk = list(dir.walk(all=all))[::-1]
    else:
        walk = dir.get_max_objects(max_count, files=all)
    if depth is not None:
        walk = filter(lambda x: x.depth <= depth, walk)

    lines = [obj.str(measure=measure, root_path=root) for obj in walk]
    if window:
        win = Screen(lines[::-1])
        win.run()
    else:
        for line in lines:
            print(line)


def tree_view(dir, all=False, summarize=False, measure=False, depth=None, fullpath=False, window=False):
    space = '    '
    branch = '│   '
    tee = '├── '
    last = '└── '

    def tree_lines(dir, prefix=''):
        contents = dir.subdirs
        if all:
            contents += dir.files
        pointers = [tee] * (len(contents) - 1) + [last]
        for pointer, content in zip(pointers, contents):
            size = str(naturalsize(content.size) if measure else content.size)
            content_name = content.path if fullpath else content.name
            yield f"{prefix}{pointer}[{size}] {content_name}"
            if isinstance(content, DirectoryInfo):
                extension = branch if pointer == tee else space
                if depth is None or content.depth < depth:
                    yield from tree_lines(content, prefix=prefix + extension)

    lines = [f"[{naturalsize(dir.size) if measure else dir.size}]"]
    if not summarize:
        lines += tree_lines(dir)

    if window:
        win = Screen(lines)
        win.run()
    else:
        for line in lines:
            print(line)