diff --git a/Lib/ensurepip/__init__.py b/Lib/ensurepip/__init__.py new file mode 100644 index 0000000..de5f6e5 --- /dev/null +++ b/Lib/ensurepip/__init__.py @@ -0,0 +1,72 @@ +import os.path +import pkgutil +import sys +import tempfile + +# TODO: setuptools dependency_links reach out to the Internet +# TODO: Remove the --pre flag when a pip 1.5 final copy is available +# TODO: --root doesn't work (pip #925) + + +__all__ = ["version", "bootstrap"] + + +_SETUPTOOLS_VERSION = "1.1.6" + +_PIP_VERSION = "1.5.dev1" + +_PROJECTS = [ + ("setuptools", _SETUPTOOLS_VERSION), + ("pip", _PIP_VERSION), +] + + +def version(): + """ + Returns a string specifying the bundled version of pip. + """ + return _PIP_VERSION + + +def bootstrap(root=None, upgrade=False, user=False, verbosity=0): + """ + Bootstrap pip into the current Python installation (or the given root + directory). + """ + with tempfile.TemporaryDirectory() as tmpdir: + # Put our bundled wheels into a temporary directory + 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) + + # Add our bundled software to the sys.path so we can import it + for project, version in _PROJECTS: + sys.path.insert(0, os.path.join( + tmpdir, + "{}-{}-py2.py3-none-any.whl".format(project, version), + )) + + # Construct the arguments to be passed to the pip command + args = [ + "install", "--use-wheel", "--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] + + # Install the bundled software + import pip + pip.main(args + [p[0] for p in _PROJECTS]) diff --git a/Lib/ensurepip/__main__.py b/Lib/ensurepip/__main__.py new file mode 100644 index 0000000..0d45788 --- /dev/null +++ b/Lib/ensurepip/__main__.py @@ -0,0 +1,51 @@ +import argparse +import ensurepip +import sys + + +def main(): + parser = argparse.ArgumentParser(prog="python -m ensurepip") + parser.add_argument( + "--version", + action="version", + version="pip {}".format(ensurepip.version()), + help="Show the version of pip that is bundled with this Python.", + ) + parser.add_argument( + "-v", "--verbose", + action="count", + default=0, + dest="verbosity", + help=("Give more output. Option is additive, and can be used up to 3 " + "times."), + ) + parser.add_argument( + "-U", "--upgrade", + action="store_true", + default=False, + help="Upgrade pip and dependencies, even if already installed.", + ) + parser.add_argument( + "--user", + action="store_true", + default=False, + help="Install using the user scheme.", + ) + parser.add_argument( + "--root", + default=None, + help="Install everything relative to this alternate root directory.", + ) + + args = parser.parse_args() + + ensurepip.bootstrap( + root=args.root, + upgrade=args.upgrade, + user=args.user, + verbosity=args.verbosity, + ) + + +if __name__ == "__main__": + sys.exit(main()) diff --git a/Lib/ensurepip/_bundled/pip-1.5.dev1-py2.py3-none-any.whl b/Lib/ensurepip/_bundled/pip-1.5.dev1-py2.py3-none-any.whl new file mode 100644 index 0000000..27e892c Binary files /dev/null and b/Lib/ensurepip/_bundled/pip-1.5.dev1-py2.py3-none-any.whl differ diff --git a/Lib/ensurepip/_bundled/setuptools-1.1.6-py2.py3-none-any.whl b/Lib/ensurepip/_bundled/setuptools-1.1.6-py2.py3-none-any.whl new file mode 100644 index 0000000..a8ba92a Binary files /dev/null and b/Lib/ensurepip/_bundled/setuptools-1.1.6-py2.py3-none-any.whl differ diff --git a/Tools/scripts/checkpip.py b/Tools/scripts/checkpip.py new file mode 100644 index 0000000..7526796 --- /dev/null +++ b/Tools/scripts/checkpip.py @@ -0,0 +1,20 @@ +#/usr/bin/env python3 +""" +Checks that the version of the projects bundled in ensurepip are the latest +versions available. +""" +import importlib.machinery # Work around for #19409 +import ensurepip +import json +import urllib.request + +for project, version in ensurepip._PROJECTS: + data = json.loads(urllib.request.urlopen( + "https://pypi.python.org/pypi/{}/json".format(project), + cadefault=True, + ).read().decode("utf8")) + upstream_version = data["info"]["version"] + + if version != upstream_version: + print("The latest version of {} on PyPI is {}, but ensurepip " + "has {}".format(project, upstream_version, version))