93 lines
2.6 KiB
Python
93 lines
2.6 KiB
Python
|
import os
|
||
|
|
import os.path
|
||
|
|
import pkgutil
|
||
|
|
import sys
|
||
|
|
import tempfile
|
||
|
|
|
||
|
|
# TODO: Remove the --pre flag when a pip 1.5 final copy is available
|
||
|
|
|
||
|
|
|
||
|
|
__all__ = ["version", "bootstrap"]
|
||
|
|
|
||
|
|
|
||
|
|
_SETUPTOOLS_VERSION = "1.3.2"
|
||
|
|
|
||
|
|
_PIP_VERSION = "1.5.dev1"
|
||
|
|
|
||
|
|
_PROJECTS = [
|
||
|
|
("setuptools", _SETUPTOOLS_VERSION),
|
||
|
|
("pip", _PIP_VERSION),
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
def _run_pip(args, additional_paths):
|
||
|
|
# Add our bundled software to the sys.path so we can import it
|
||
|
|
sys.path = additional_paths + sys.path
|
||
|
|
|
||
|
|
# Install the bundled software
|
||
|
|
import pip
|
||
|
|
pip.main(args)
|
||
|
|
|
||
|
|
|
||
|
|
def version():
|
||
|
|
"""
|
||
|
|
Returns a string specifying the bundled version of pip.
|
||
|
|
"""
|
||
|
|
return _PIP_VERSION
|
||
|
|
|
||
|
|
|
||
|
|
def bootstrap(*, root=None, upgrade=False, user=False,
|
||
|
|
altinstall=False, default_pip=False,
|
||
|
|
verbosity=0):
|
||
|
|
"""
|
||
|
|
Bootstrap pip into the current Python installation (or the given root
|
||
|
|
directory).
|
||
|
|
"""
|
||
|
|
if altinstall and default_pip:
|
||
|
|
raise ValueError("Cannot use altinstall and default_pip together")
|
||
|
|
|
||
|
|
# By default, installing pip and setuptools installs all of the
|
||
|
|
# following scripts (X.Y == running Python version):
|
||
|
|
#
|
||
|
|
# pip, pipX, pipX.Y, easy_install, easy_install-X.Y
|
||
|
|
#
|
||
|
|
# pip 1.5+ allows ensurepip to request that some of those be left out
|
||
|
|
if altinstall:
|
||
|
|
# omit pip, pipX and easy_install
|
||
|
|
os.environ["ENSUREPIP_OPTIONS"] = "altinstall"
|
||
|
|
elif not default_pip:
|
||
|
|
# omit pip and easy_install
|
||
|
|
os.environ["ENSUREPIP_OPTIONS"] = "install"
|
||
|
|
|
||
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
||
|
|
# Put our bundled wheels into a temporary directory and construct the
|
||
|
|
# additional paths that need added to sys.path
|
||
|
|
additional_paths = []
|
||
|
|
for project, version in _PROJECTS:
|
||
|
|
wheel_name = "{}-{}-py2.py3-none-any.whl".format(project, version)
|
||
|
|
whl = pkgutil.get_data(
|
||
|
|
"ensurepip",
|
||
|
|
"_bundled/{}".format(wheel_name),
|
||
|
|
)
|
||
|
|
with open(os.path.join(tmpdir, wheel_name), "wb") as fp:
|
||
|
|
fp.write(whl)
|
||
|
|
|
||
|
|
additional_paths.append(os.path.join(tmpdir, wheel_name))
|
||
|
|
|
||
|
|
# Construct the arguments to be passed to the pip command
|
||
|
|
args = [
|
||
|
|
"install", "--no-index", "--find-links", tmpdir,
|
||
|
|
# Temporary until pip 1.5 is final
|
||
|
|
"--pre",
|
||
|
|
]
|
||
|
|
if root:
|
||
|
|
args += ["--root", root]
|
||
|
|
if upgrade:
|
||
|
|
args += ["--upgrade"]
|
||
|
|
if user:
|
||
|
|
args += ["--user"]
|
||
|
|
if verbosity:
|
||
|
|
args += ["-" + "v" * verbosity]
|
||
|
|
|
||
|
|
_run_pip(args + [p[0] for p in _PROJECTS], additional_paths)
|