From: Vincent Driessen <vincent@3rdcloud.com>
Date: Thu, 6 Sep 2012 06:47:30 +0000 (-0700)
Subject: Merge pull request #14 from jakebasile/max-complexity
X-Git-Url: https://git.madduck.net/etc/vim.git/commitdiff_plain/fc04a1123eb6b2aa855eddfcee6da7a04ca0408c?hp=ca1dc6d3ef003f7e44eec46231779cb18009c88d

Merge pull request #14 from jakebasile/max-complexity

Added ability to set max complexity argument for flake8.
---

diff --git a/README.mdown b/README.mdown
index 0694399..e32260d 100644
--- a/README.mdown
+++ b/README.mdown
@@ -40,6 +40,10 @@ If you want to change the max line length for PEP8:
 
     let g:flake8_max_line_length=99
 
+To set the maximum [McCabe complexity](https://en.wikipedia.org/wiki/Cyclomatic_complexity) before a warning is issued:
+
+    let g:flake8_max_complexity=10
+
 To cutomize the location of your flake8 binary, set `g:flake8_cmd`:
 
     let g:flake8_cmd="/opt/strangebin/flake8000"
diff --git a/ftplugin/python_flake8.vim b/ftplugin/python_flake8.vim
index e3b3c61..0b386c7 100644
--- a/ftplugin/python_flake8.vim
+++ b/ftplugin/python_flake8.vim
@@ -27,6 +27,11 @@ if exists("g:flake8_max_line_length")
     let s:flake8_max_line_length=" --max-line-length=".g:flake8_max_line_length
 endif
 
+let s:flake8_max_complexity=""
+if exists("g:flake8_max_complexity")
+    let s:flake8_max_complexity=" --max-complexity=".g:flake8_max_complexity
+endif
+
 if !exists("*Flake8()")
     function Flake8()
         if !executable(s:flake8_cmd)
@@ -48,7 +53,7 @@ if !exists("*Flake8()")
 
         " perform the grep itself
         let &grepformat="%f:%l:%c: %m\,%f:%l: %m"
-        let &grepprg=s:flake8_cmd.s:flake8_ignores.s:flake8_max_line_length
+        let &grepprg=s:flake8_cmd.s:flake8_ignores.s:flake8_max_line_length.s:flake8_max_complexity
         silent! grep! %
 
         " restore grep settings