Skip to content

Commit

Permalink
00189: Instead of bundled wheels, use our RPM packaged wheels
Browse files Browse the repository at this point in the history
We keep them in /usr/share/python-wheels

Downstream only: upstream bundles
We might eventually pursuit upstream support, but it's low prio
  • Loading branch information
hroncok authored and hrnciar committed Nov 16, 2021
1 parent 271103a commit ed9f5da
Showing 1 changed file with 26 additions and 11 deletions.
37 changes: 26 additions & 11 deletions Lib/ensurepip/__init__.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
import distutils.version
import glob
import os
import os.path
import sys
Expand All @@ -6,13 +8,29 @@
import subprocess
from importlib import resources

from . import _bundled



__all__ = ["version", "bootstrap"]
_SETUPTOOLS_VERSION = "58.1.0"
_PIP_VERSION = "21.2.4"

_WHEEL_DIR = "/usr/share/python-wheels/"

_wheels = {}

def _get_most_recent_wheel_version(pkg):
prefix = os.path.join(_WHEEL_DIR, "{}-".format(pkg))
_wheels[pkg] = {}
for suffix in "-py2.py3-none-any.whl", "-py3-none-any.whl":
pattern = "{}*{}".format(prefix, suffix)
for path in glob.glob(pattern):
version_str = path[len(prefix):-len(suffix)]
_wheels[pkg][version_str] = os.path.basename(path)
return str(max(_wheels[pkg], key=distutils.version.LooseVersion))


_SETUPTOOLS_VERSION = _get_most_recent_wheel_version("setuptools")

_PIP_VERSION = _get_most_recent_wheel_version("pip")

_PROJECTS = [
("setuptools", _SETUPTOOLS_VERSION, "py3"),
("pip", _PIP_VERSION, "py3"),
Expand Down Expand Up @@ -101,13 +119,10 @@ def _bootstrap(*, root=None, upgrade=False, user=False,
# additional paths that need added to sys.path
additional_paths = []
for project, version, py_tag in _PROJECTS:
wheel_name = "{}-{}-{}-none-any.whl".format(project, version, py_tag)
whl = resources.read_binary(
_bundled,
wheel_name,
)
with open(os.path.join(tmpdir, wheel_name), "wb") as fp:
fp.write(whl)
wheel_name = _wheels[project][version]
with open(os.path.join(_WHEEL_DIR, wheel_name), "rb") as sfp:
with open(os.path.join(tmpdir, wheel_name), "wb") as fp:
fp.write(sfp.read())

additional_paths.append(os.path.join(tmpdir, wheel_name))

Expand Down

0 comments on commit ed9f5da

Please sign in to comment.