mirror of
https://github.com/pypa/pip
synced 2023-12-13 21:30:23 +01:00
e9a7fb4e49
For me, this speeds up `py.test tests/functional/test_freeze.py` from 62 seconds to 54 seconds (about a 13% speedup).
90 lines
2.5 KiB
Python
90 lines
2.5 KiB
Python
from __future__ import absolute_import
|
|
|
|
import os
|
|
import sys
|
|
import subprocess
|
|
|
|
import virtualenv as _virtualenv
|
|
|
|
from .path import Path
|
|
|
|
|
|
# On Python < 3.3 we don't have subprocess.DEVNULL
|
|
try:
|
|
DEVNULL = subprocess.DEVNULL
|
|
except AttributeError:
|
|
DEVNULL = open(os.devnull, "wb")
|
|
|
|
|
|
class VirtualEnvironment(object):
|
|
"""
|
|
An abstraction around virtual environments, currently it only uses
|
|
virtualenv but in the future it could use pyvenv.
|
|
"""
|
|
|
|
def __init__(self, location, *args, **kwargs):
|
|
self.location = Path(location)
|
|
self.pip_source_dir = kwargs.pop("pip_source_dir")
|
|
self._system_site_packages = kwargs.pop("system_site_packages", False)
|
|
|
|
home, lib, inc, bin = _virtualenv.path_locations(self.location)
|
|
# workaround for https://github.com/pypa/virtualenv/issues/306
|
|
if hasattr(sys, "pypy_version_info"):
|
|
lib = os.path.join(home, 'lib-python', sys.version[:3])
|
|
self.lib = Path(lib)
|
|
self.bin = Path(bin)
|
|
|
|
super(VirtualEnvironment, self).__init__(*args, **kwargs)
|
|
|
|
def __repr__(self):
|
|
return "<VirtualEnvironment {}>".format(self.location)
|
|
|
|
@classmethod
|
|
def create(cls, location, clear=False, pip_source_dir=None):
|
|
obj = cls(location, pip_source_dir=pip_source_dir)
|
|
obj._create(clear=clear)
|
|
return obj
|
|
|
|
def _create(self, clear=False):
|
|
# Create the actual virtual environment
|
|
_virtualenv.create_environment(
|
|
self.location,
|
|
clear=clear,
|
|
never_download=True,
|
|
no_pip=True,
|
|
)
|
|
|
|
# Install our development version of pip install the virtual
|
|
# environment
|
|
cmd = [self.bin.join("python"), "setup.py", "install", "--no-compile"]
|
|
p = subprocess.Popen(
|
|
cmd,
|
|
cwd=self.pip_source_dir,
|
|
stderr=subprocess.STDOUT,
|
|
stdout=DEVNULL,
|
|
)
|
|
p.communicate()
|
|
if p.returncode != 0:
|
|
raise Exception(p.stderr)
|
|
raise subprocess.CalledProcessError(
|
|
p.returncode,
|
|
cmd[0],
|
|
output=p.stdout,
|
|
)
|
|
|
|
def clear(self):
|
|
self._create(clear=True)
|
|
|
|
@property
|
|
def system_site_packages(self):
|
|
return self._system_site_packages
|
|
|
|
@system_site_packages.setter
|
|
def system_site_packages(self, value):
|
|
marker = self.lib.join("no-global-site-packages.txt")
|
|
if value:
|
|
marker.rm()
|
|
else:
|
|
marker.touch()
|
|
self._system_site_packages = value
|