1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
|
from __future__ import annotations
import os.path
from pycodestyle import BaseReport
from pycodestyle import StyleGuide
ROOT = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
class InMemoryReport(BaseReport):
"""
Collect the results in memory, without printing anything.
"""
def __init__(self, options):
super().__init__(options)
self.in_memory_errors = []
def error(self, line_number, offset, text, check):
"""
Report an error, according to options.
"""
code = text[:4]
self.in_memory_errors.append(f'{code}:{line_number}:{offset + 1}')
return super().error(line_number, offset, text, check)
def errors_from_src(src: str) -> list[str]:
guide = StyleGuide(select=('E', 'W'), max_doc_length=72)
reporter = guide.init_report(InMemoryReport)
guide.input_file(
filename='in-memory-test-file.py',
lines=src.splitlines(True),
)
return reporter.in_memory_errors
|
