pantsbuild / pants / 18538208066

from __future__ import annotations

import os

from collections import defaultdict

from collections.abc import Iterable

def partition_files_by_directory(filepaths: Iterable[str]) -> dict[str, list[str]]:

    directories = defaultdict(list)

    for filepath in filepaths:

        directory = os.path.dirname(filepath) or "."

        directories[directory].append(filepath)

    return directories