Add virtualenv support

klen · klen · commit 8acb90c241ef · 2011-10-18T23:51:23.000+04:00
diff --git a/Changelog.rst b/Changelog.rst
@@ -1,9 +1,10 @@
 Changelog
 =========
 
-## 2011-10-18 0.2.5
+## 2011-10-18 0.2.6
 -------------------
 * Add <C-space> shortcut for macvim users.
+* Add VIRTUALENV support
 
 ## 2011-10-17 0.2.4
 -------------------
diff --git a/README.rst b/README.rst
@@ -15,6 +15,7 @@ There is no need to install the pylint_ or rope_ library on your system.
 - Run python code
 - Go to definition
 - Powerful customization
+- Virtualenv support
 - And more...
 
 See screencast here: http://t.co/3b0bzeXA (sorry for quality, this my first screencast)
@@ -151,6 +152,9 @@ Default values: ::
     " Autoremove unused whitespaces
     let g:pymode_utils_whitespaces = 1
 
+    " Auto fix vim python paths if virtualenv enabled
+    let g:pymode_virtualenv = 1
+
 .. note:: See also :help ropevim.txt
 
 
diff --git a/after/plugin/pymode_virtualenv.vim b/after/plugin/pymode_virtualenv.vim
@@ -0,0 +1,17 @@
+" OPTION: g:pymode_lint -- bool. Load pylint plugin
+call helpers#SafeVar("g:pymode_virtualenv", 1)
+
+" DESC: Disable script loading
+if !g:pymode_virtualenv || !g:pymode
+    let g:pymode_virtualenv = 0
+    finish
+endif
+
+fun! pymode_virtualenv#Activate() "{{{
+python << EOF
+ve_dir = os.environ['VIRTUAL_ENV']
+ve_dir in sys.path or sys.path.insert(0, ve_dir)
+activate_this = os.path.join(os.path.join(ve_dir, 'bin'), 'activate_this.py')
+execfile(activate_this, dict(__file__=activate_this))
+EOF
+endfunction "}}}
diff --git a/doc/pymode.txt b/doc/pymode.txt
@@ -62,6 +62,8 @@ PythonMode. These options should be set in your vimrc.
 
 |'pymode_utils'|                    Turns off utils
 
+|'pymode_virtualenv'|               Turns off virtualenv
+
 |'pymode_utils_whitespaces'|        Remove unused whitespaces
 
 
@@ -171,6 +173,13 @@ Default: 1.
 
 If this option is set to 0 then utils script is disabled.
 
+------------------------------------------------------------------------------
+                                                           *'pymode_virtualenv'*
+Values: 0 or 1.
+Default: 1.
+
+If this option is set to 0 then virtualenv support is disabled.
+
 ------------------------------------------------------------------------------
                                                     *'pymode_utils_whitespaces'*
 Values: 0 or 1.
diff --git a/ftplugin/python/pymode.vim b/ftplugin/python/pymode.vim
@@ -20,6 +20,12 @@ if g:pymode
 
 endif
 
+" Add virtualenv paths
+if g:pymode_virtualenv && exists("$VIRTUAL_ENV")
+    echomsg "Enabled virtualenv: " . $VIRTUAL_ENV
+    call pymode_virtualenv#Activate()
+endif
+
 " Python documentation
 if g:pymode_doc
 
