From 2f0a76fa54ea58c6957df4ccae5a92ae6e6faa53 Mon Sep 17 00:00:00 2001
From: Michiel Scholten <michiel@diginaut.net>
Date: Sun, 10 Feb 2019 16:15:17 +0100
Subject: [PATCH] Properly detect Django (well, availability of pylint-django)

---
 .vim/after/ftplugin/python.vim | 40 ++++++++++++++++++++++++++++++++++
 .vimrc                         |  7 ++----
 2 files changed, 42 insertions(+), 5 deletions(-)
 create mode 100644 .vim/after/ftplugin/python.vim

diff --git a/.vim/after/ftplugin/python.vim b/.vim/after/ftplugin/python.vim
new file mode 100644
index 0000000..0a3fbc0
--- /dev/null
+++ b/.vim/after/ftplugin/python.vim
@@ -0,0 +1,40 @@
+python3 <<EOF
+import vim
+from importlib.util import find_spec
+import os
+import glob
+
+def find_virtualenv(virtualenv_names):
+    cwd = vim.eval('getcwd()')
+    while cwd != '/':
+        for virtualenv_name in virtualenv_names:
+            venv_path = os.path.join(cwd, virtualenv_name)
+            if os.path.exists(venv_path):
+                return venv_path
+
+        cwd, _ignored = os.path.split(cwd)
+
+    return os.environ.get('VIRTUAL_ENV')
+
+# If we have a virtualenv, check to see whether it contains pylint and the
+# django module. If we don't, just try to import both.  We can't even use
+# ale's "ale_virtualenv_dir_names" here, because it's not set yet.
+
+virtualenv_path = find_virtualenv(['virtualenv', 'venv'])  #vim.eval('ale_virtualenv_dir_names')
+
+if virtualenv_path:
+    has_pylint_django = glob.glob(os.path.join(virtualenv_path, 'lib/*/site-packages/pylint_django'))
+else:
+    try:
+        find_spec('pylint_django')
+        has_pylint_django = True
+    except ImportError:
+        has_pylint_django = False
+
+if has_pylint_django:
+    # No silly 80-char line limit. Sorry pep-8. Also, Django support. Disable 'invalid name', 'missing docstring'
+    vim.command("let g:ale_python_pylint_options='--max-line-length=120 --load-plugins pylint_django --disable=invalid-name,missing-docstring'")
+else:
+    # No silly 80-char line limit. Sorry pep-8. Disable 'invalid name', 'missing docstring'
+    vim.command("let g:ale_python_pylint_options='--max-line-length=120 --disable=invalid-name,missing-docstring'")
+EOF
diff --git a/.vimrc b/.vimrc
index 83019ad..4ccfbe5 100644
--- a/.vimrc
+++ b/.vimrc
@@ -313,11 +313,8 @@ let g:ale_fixers = {}
 " Python specific settings
 let g:ale_fixers.python = ['isort']
 " No silly 80-char line limit. Sorry pep-8. Also, Django support. Disable 'invalid name', 'missing docstring'
-if exists('b:is_django')
-    let g:ale_python_pylint_options="--max-line-length=120 --load-plugins pylint_django --disable=invalid-name,missing-docstring"
-else
-    let g:ale_python_pylint_options="--max-line-length=120 --disable=invalid-name,missing-docstring"
-endif
+"let g:ale_python_pylint_options="--max-line-length=120 --disable=invalid-name,missing-docstring"
+" VIMHOME/after/ftplugin/python.vim takes care of this
 " Show errors or warnings in the statusline
 let g:airline#extensions#ale#enabled = 1
 " UI