import site
from os import environ
from os.path import join
import sys
if 'VIRTUAL_ENV' in environ:
virtual_env = join(environ.get('VIRTUAL_ENV'),
'lib',
'python%d.%d' % sys.version_info[:2],
'site-packages')
# Remember original sys.path.
prev_sys_path = list(sys.path)
site.addsitedir(virtual_env)
# Personal hack
# Remove /usr/lib/python2.7/dist-packages and associates from the path
# Counter the effects of this code, in /usr/bin/ipython
# import sys
# if "/usr/lib/python2.7/dist-packages" not in sys.path:
# sys.path.append("/usr/lib/python2.7/dist-packages")
for item in sys.path:
if '/usr/lib/python2.7/dist-packages' in item:
sys.path.remove(item)
# Reorder sys.path so new directories at the front.
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[1:1] = new_sys_path
print 'VIRTUAL_ENV ->', virtual_env
del virtual_env
del sys, site, environ, join