Function main

scripts/sort_class.py:220–246

(index_filename: str, class_names: list[str], dry_run: bool)

def main(index_filename: str, class_names: list[str], dry_run: bool):
    if len(class_names) > 1:
        print(f"Sorting {len(class_names)} Python files")

    with open(index_filename) as r:
        index = json.load(r)

    filenames = {}
    for class_name in class_names:
        full_class_name = class_name
        if "." not in class_name:
            cls = index.get("classes", {}).get(class_name)
            if not cls:
                raise ValueError(f"Class {class_name} does not exist in index")
            full_class_name = f'{cls.get("package")}.{cls.get("module")}.{cls.get("name")}'
        package, module, class_name = full_class_name.split(".", maxsplit=2)
        filename = f"{package}/{module}.py"
        filenames[class_name] = filename

    with multiprocessing.Manager() as manager:
        locks = {filename: manager.Lock() for filename in filenames}
        locks.update({"stdout": manager.Lock()})

        # sort files in parallel
        worker = SortFileWorker(dry_run, locks)
        with multiprocessing.Pool() as pool:
            pool.map(worker.sort, iterable=filenames.items())


def parse_args():