FilesExpand file tree

 develop

/

virtualenv.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

70 lines (54 loc) · 2 KB

 develop

/

virtualenv.py

Copy path

Top

File metadata and controls

• Code
• Blame

70 lines (54 loc) · 2 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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