From 2f3dc23e1bc24b74c2673ecaf167c4bfa776a71a Mon Sep 17 00:00:00 2001
From: Daniel Rice <daniel.rice@virtahealth.com>
Date: Sun, 19 Nov 2023 15:35:45 -0700
Subject: [PATCH] feat: Add support for --show-column-numbers output

---
 mypy_baseline/_error.py | 11 ++++++++++-
 tests/test_error.py     | 15 +++++++++++++++
 2 files changed, 25 insertions(+), 1 deletion(-)

diff --git a/mypy_baseline/_error.py b/mypy_baseline/_error.py
index 5c50843..cb0c552 100644
--- a/mypy_baseline/_error.py
+++ b/mypy_baseline/_error.py
@@ -20,6 +20,15 @@
     (?:\s\s\[(?P<category>[a-z-]+)\])?
     \s*
 """, re.VERBOSE | re.MULTILINE)
+REX_LINE_COLUMN = re.compile(r"""
+    (?P<path>.+\.pyi?):
+    (?P<lineno>[0-9]+):
+    (?P<columnno>[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
@@ -44,7 +53,7 @@ class Error:
     @classmethod
     def new(self, line: str) -> Error | None:
         line = _remove_color_codes(line)
-        match = REX_LINE.fullmatch(line) or REX_LINE_NBQA.fullmatch(line)
+        match = REX_LINE.fullmatch(line) or REX_LINE_NBQA.fullmatch(line) or REX_LINE_COLUMN.fullmatch(line)
         if match is None:
             return None
         return Error(line, match)
diff --git a/tests/test_error.py b/tests/test_error.py
index ae9efd1..45b213c 100644
--- a/tests/test_error.py
+++ b/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