#852 Capture error codes for pycodestyle, and consider every code except E999 to be style errors or warnings

1 files changed, 14 insertions, 5 deletions

diff --git a/ale_linters/python/pycodestyle.vim b/ale_linters/python/pycodestyle.vim
index ad895999..0382e9d0 100644
--- a/ale_linters/python/pycodestyle.vim
+++ b/ale_linters/python/pycodestyle.vim
@@ -17,18 +17,27 @@ function! ale_linters#python#pycodestyle#GetCommand(buffer) abort
 endfunction
 
 function! ale_linters#python#pycodestyle#Handle(buffer, lines) abort
-    let l:pattern = '\v^(\S*):(\d*):(\d*): ((([EW])\d+) .*)$'
+    let l:pattern = '\v^(\S*):(\d*):(\d*): ([EW]\d+) (.*)$'
     let l:output = []
 
     " lines are formatted as follows:
     " file.py:21:26: W291 trailing whitespace
     for l:match in ale#util#GetMatches(a:lines, l:pattern)
-        call add(l:output, {
+        let l:item = {
         \   'lnum': l:match[2] + 0,
         \   'col': l:match[3] + 0,
-        \   'type': l:match[6],
-        \   'text': l:match[4],
-        \})
+        \   'type': l:match[4][0],
+        \   'sub_type': 'style',
+        \   'text': l:match[5],
+        \   'code': l:match[4],
+        \}
+
+        " E999 is not a style error, it's a syntax error.
+        if l:match[4] is# 'E999'
+            unlet l:item.sub_type
+        endif
+
+        call add(l:output, l:item)
     endfor
 
     return l:output