Viewing file: collector.py (1.3 KB) -rw-r--r-- Select action/file-type: (+) | (+) | (+) | Code (+) | Session (+) | (+) | SDB (+) | (+) | (+) | (+) | (+) | (+) |
from io import StringIO from typing import List
from pylint.exceptions import UnknownMessageError from pylint.message import Message as PylintMessage from pylint.reporters import BaseReporter
from prospector.message import Location, Message
class Collector(BaseReporter): name = "collector"
def __init__(self, message_store): BaseReporter.__init__(self, output=StringIO()) self._message_store = message_store self._messages = []
def handle_message(self, msg: PylintMessage) -> None: loc = Location(msg.abspath, msg.module, msg.obj, msg.line, msg.column)
# At this point pylint will give us the code but we want the # more user-friendly symbol try: msg_data = self._message_store.get_message_definitions(msg.msg_id) except UnknownMessageError: # this shouldn't happen, as all pylint errors should be # in the message store, but just in case we'll fall back # to using the code. msg_symbol = msg.msg_id else: msg_symbol = msg_data[0].symbol
message = Message("pylint", msg_symbol, loc, msg.msg) self._messages.append(message)
def _display(self, layout) -> None: pass
def get_messages(self) -> List[Message]: return self._messages
|