Add pyright ignore statements for every diagnosis encountered

add_pyright_ignore.py

import subprocess
import re
import json


def run_pyright():
    # Run Pyright and capture the output
    result = subprocess.run(["pyright", "--outputjson"], capture_output=True, text=True)
    return result.stdout


def add_pyright_ignore_comments(output):
    # Parse Pyright output
    # Example output: {"severity": "warning", "message": "Unused import", "range": {"start": {"line": 5, "character": 8}, "end": {"line": 5, "character": 17}}, "file": "example.py"}
    warnings = [
        match.groupdict()
        for match in re.finditer(
            r'"severity": "(warning|error)".*?"file": "(.*?)".*?"start": {"line": (\d+),',
            output,
        )
    ]
    data = json.loads(output)

    # Add pyright: ignore comments to identified lines
    for warning in data["generalDiagnostics"]:
        filename = warning["file"]
        line_number = warning["range"]["start"]["line"]
        with open(filename, "r+") as file:
            lines = file.readlines()
            lines[line_number] = (
                lines[line_number].rstrip("\n") + "  # pyright: ignore\n"
            )
            file.seek(0)
            file.writelines(lines)


if __name__ == "__main__":
    pyright_output = run_pyright()
    add_pyright_ignore_comments(pyright_output)