Mini Shell

from collections import defaultdict

from prospector.formatters.text import TextFormatter

__all__ = ("GroupedFormatter",)


class GroupedFormatter(TextFormatter):
    def render_messages(self):
        output = [
            "Messages",
            "========",
            "",
        ]

        groups = defaultdict(lambda: defaultdict(list))

        for message in self.messages:
            groups[self._make_path(message.location.path)][message.location.line].append(message)

        for filename in sorted(groups.keys()):
            output.append(str(filename))

            for line in sorted(groups[filename].keys(), key=lambda x: 0 if x is None else int(x)):
                output.append("  Line: %s" % line)

                for message in groups[filename][line]:
                    output.append(
                        "    %s: %s / %s%s"
                        % (
                            message.source,
                            message.code,
                            message.message,
                            (" (col %s)" % message.location.character) if message.location.character else "",
                        )
                    )

            output.append("")

        return "\n".join(output)