import os import re from prospector.formatters.base import Formatter class VSCodeFormatter(Formatter): """ This formatter outputs messages in the same way as vscode prospector linter expects. """ def render(self, summary=True, messages=True, profile=False): # this formatter will always ignore the summary and profile cur_loc = None output = [] for message in sorted(self.messages): if cur_loc != message.location.path: cur_loc = message.location.path module_name = message.location.path.replace(os.path.sep, '.') module_name = re.sub(r'(\.__init__)?\.py$', '', module_name) header = '************* Module %s' % module_name output.append(header) template = '%(line)s,%(character)s,%(code)s,%(code)s:%(source)s %(message)s' output.append(template % { 'line': message.location.line, 'character': message.location.character, 'source': message.source, 'code': message.code, 'message': message.message.strip() }) return '\n'.join(output)