flake8-generate-per-file-ignores.py

"""
Generates per-file-ignores section for the .flake8 configuration file based on
flake8 run output (can be either piped using STDIN or read from the file as the
first argument to this script).

Use this to disable some linting in legacy code, but keep full linting for
newly written code.
"""

import fileinput
from collections import defaultdict

per_file_ignores = defaultdict(set)

for line in fileinput.input():
    path, line, column, error = line.split(":", 3)
    code, message = error.strip().split(" ", 1)
    per_file_ignores[path].add(code)

for path, codes in per_file_ignores.items():
    print(f"{path}:{','.join(codes)}")