Function sort_class

scripts/sort_class.py:177–207

(class_name: str, filename: str, dry_run: bool, locks: dict[str, multiprocessing.Lock])

def sort_class(class_name: str, filename: str, dry_run: bool, locks: dict[str, multiprocessing.Lock]):
    print(f"Sorting {class_name} ({filename})")

    file_lock = locks.get(filename) if not dry_run else None
    stdout_lock = locks.get("stdout")

    if file_lock:
        file_lock.acquire()
    try:
        with open(filename) as r:
            code = "".join(r.readlines())

        tree = cst.parse_module(code)
        transformer = SortMethodsTransformer(class_name)
        tree_updated = tree.visit(transformer)

        if dry_run:
            diff = "".join(difflib.unified_diff(code.splitlines(1), tree_updated.code.splitlines(1)))
            if diff:
                stdout_lock.acquire()
                print(f"Diff of {class_name}:")
                print(diff)
                print()
                stdout_lock.release()
        else:
            if not tree_updated.deep_equals(tree):
                with open(filename, "w") as w:
                    w.write(tree_updated.code)
    finally:
        if file_lock:
            file_lock.release()


class SortFileWorker: