warn_sort.py

import dataclasses
import operator


@dataclasses.dataclass
class WarningFilter:
    command: str
    msg: str
    cls: str
    module: str | None = None

    def format(self):
        prefix = f"{self.command}:{self.msg}:{self.cls}"
        return prefix if self.module is None else f"{prefix}:{self.module}"

    def sort_key(self):
        return (self.module is not None, self.module, self.cls, self.msg)

    @classmethod
    def from_format(cls, fmt):
        return cls(*fmt.strip().split(":"))


def warn_sort(v):
    return "\n".join(
        h.format()
        for h in sorted(
            (WarningFilter.from_format(x) for x in v.split("\n") if x.strip()),
            key=operator.methodcaller("sort_key"),
        )
    )