From 580a4f4ab019a09fed6975f0a26065d16f767b98 Mon Sep 17 00:00:00 2001 From: David Halter Date: Sun, 18 Nov 2012 02:44:43 +0100 Subject: [PATCH] support for virtual envs, #36, fixes jedi-vim issue 31 --- jedi/builtin.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/jedi/builtin.py b/jedi/builtin.py index 1792555d..a51525b1 100644 --- a/jedi/builtin.py +++ b/jedi/builtin.py @@ -14,7 +14,19 @@ import parsing def get_sys_path(): - return sys.path[1:] + def check_virtual_env(sys_path): + """ Add virtualenv's site-packages to the `sys.path`.""" + venv = os.getenv('VIRTUAL_ENV') + if not venv: + return + venv = os.path.abspath(venv) + p = os.path.join( + venv, 'lib', 'python%d.%d' % sys.version_info[:2], 'site-packages') + sys_path.insert(0, p) + + p = sys.path[1:] + check_virtual_env(p) + return p class CachedModule(object): """