Merge pull request #20 from drice/drice/support-show-column-numbers

feat: Add support for --show-column-numbers output

Author: orsinium

mypy_baseline/_error.py

@@ -20,6 +20,14 @@
     (?:\s\s\[(?P<category>[a-z-]+)\])?
     \s*
 """, re.VERBOSE | re.MULTILINE)
+REX_LINE = re.compile(r"""
+    (?P<path>.+\.pyi?):
+    (?P<lineno>[0-9]+):(?:[0-9]+:)?\s
+    (?P<severity>[a-z]+):\s
+    (?P<message>.+?)
+    (?:\s\s\[(?P<category>[a-z-]+)\])?
+    \s*
+""", re.VERBOSE | re.MULTILINE)
 REX_LINE_NBQA = re.compile(r"""
     (?P<path>.+\.ipynb:cell_[0-9]+):
     (?P<lineno>[0-9]+):\s

tests/test_error.py

@@ -46,3 +46,18 @@ def test_line3_parse():
     assert e.message == 'This violates the Liskov substitution principle'
     assert e.category == 'note'
     assert e.get_clean_line(Config()) == LINE3EXP
+
+# --show-column-numbers files
+LINE4 = 'my_project/api/views.py:10:42: note: This violates the Liskov substitution principle\r\n'  # noqa
+LINE4EXP = 'my_project/api/views.py:0: note: This violates the Liskov substitution principle'  # noqa
+
+
+def test_line4_parse():
+    e = Error.new(LINE4)
+    assert e is not None
+    assert e.path.parts == ('my_project', 'api', 'views.py')
+    assert e.line_number == 10
+    assert e.severity == 'note'
+    assert e.message == 'This violates the Liskov substitution principle'
+    assert e.category == 'note'
+    assert e.get_clean_line(Config()) == LINE4EXP