forked from python-mode/python-mode
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathvirtualenv.py
More file actions
70 lines (54 loc) · 2 KB
/
virtualenv.py
File metadata and controls
70 lines (54 loc) · 2 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
""" Support virtualenv in pymode. """
import os
import sys
import site
from .environment import env
@env.catch_exceptions
def enable_virtualenv():
""" Enable virtualenv for vim.
:return bool:
"""
path = env.var('g:pymode_virtualenv_path')
enabled = env.var('g:pymode_virtualenv_enabled')
if path == enabled:
env.message('Virtualenv %s already enabled.' % path)
return env.stop()
activate_this = os.path.join(os.path.join(path, 'bin'), 'activate_this.py')
# Fix for windows
if not os.path.exists(activate_this):
activate_this = os.path.join(
os.path.join(path, 'Scripts'), 'activate_this.py')
try:
with open(activate_this) as f:
source = f.read()
exec(compile( # noqa
source, activate_this, 'exec'), dict(__file__=activate_this))
except IOError:
_activate_env_from_path(path)
env.message('Activate virtualenv: ' + path)
env.let('g:pymode_virtualenv_enabled', path)
return True
def _activate_env_from_path(env_path):
""" Fix when `activate_this.py` does not exist.
For Python 3.3 and newer, a new command-line tool `pyvenv` create venv
will not provide 'activate_this.py'.
"""
prev_sys_path = list(sys.path)
if sys.platform == 'win32':
site_packages_paths = [os.path.join(env_path, 'Lib', 'site-packages')]
else:
lib_path = os.path.join(env_path, 'lib')
site_packages_paths = [os.path.join(lib_path, lib, 'site-packages')
for lib in os.listdir(lib_path)]
for site_packages_path in site_packages_paths:
site.addsitedir(site_packages_path)
sys.real_prefix = sys.prefix
sys.prefix = env_path
sys.exec_prefix = env_path
# Move the added items to the front of the path:
new_sys_path = []
for item in list(sys.path):
if item not in prev_sys_path:
new_sys_path.append(item)
sys.path.remove(item)
sys.path[:0] = new_sys_path