Function ensure_python_distribution

testing/__init__.py:728–792

(
    version,  # type: str
    python_version=None,  # type: Optional[str]
    allow_adhoc_version=False,  # type: bool
)

def ensure_python_distribution(
    version,  # type: str
    python_version=None,  # type: Optional[str]
    allow_adhoc_version=False,  # type: bool
):
    # type: (...) -> PyenvPythonDistribution
    if not allow_adhoc_version and version not in ALL_PY_VERSIONS:
        raise ValueError("Please constrain version to one of {}".format(ALL_PY_VERSIONS))

    if WINDOWS and _ALL_PY_VERSIONS_TO_VERSION_INFO[version][:2] < (3, 8):
        pytest.skip(
            "Test uses pyenv {version} interpreter which is not supported on Windows.".format(
                version=version
            )
        )

    clone_dir = os.path.abspath(
        os.path.join(PEX_TEST_DEV_ROOT, "pyenv-win" if WINDOWS else "pyenv")
    )
    with atomic_directory(target_dir=clone_dir) as pyenv_root_atomic_dir:
        if not pyenv_root_atomic_dir.is_finalized():
            bootstrap_python_installer(pyenv_root_atomic_dir.work_dir)

    pyenv_root = os.path.join(clone_dir, "pyenv-win") if WINDOWS else clone_dir
    pyenv = os.path.join(pyenv_root, "bin", "pyenv.bat" if WINDOWS else "pyenv")

    interpreter_location = os.path.join(pyenv_root, "versions", version)
    pip = os.path.join(interpreter_location, SCRIPT_DIR, script_name("pip"))

    if WINDOWS:
        python = os.path.join(interpreter_location, "python.exe")
    else:
        major, minor = (python_version or version).split(".")[:2]
        python = os.path.join(
            interpreter_location, "bin", "python{major}.{minor}".format(major=major, minor=minor)
        )

    with atomic_directory(target_dir=interpreter_location) as interpreter_target_dir:
        if not interpreter_target_dir.is_finalized():
            with pyenv_root_atomic_dir.locked():
                subprocess.check_call(args=["git", "pull", "--ff-only"], cwd=pyenv_root)

            env = os.environ.copy()
            env["PYENV_ROOT"] = pyenv_root
            env["CFLAGS"] = "-std=c11"
            if sys.platform.lower().startswith("linux"):
                env["CONFIGURE_OPTS"] = "--enable-shared"
                # The pyenv builder detects `--enable-shared` and sets up `RPATH` via
                # `LDFLAGS=-Wl,-rpath=... $LDFLAGS` to ensure the built python binary links the
                # correct libpython shared lib. Some versions of compiler set the `RUNPATH` instead
                # though which is searched _after_ the `LD_LIBRARY_PATH` environment variable. To
                # ensure an inopportune `LD_LIBRARY_PATH` doesn't fool the pyenv python binary into
                # linking the wrong libpython, force `RPATH`, which is searched 1st by the linker,
                # with `--disable-new-dtags`.
                env["LDFLAGS"] = "-Wl,--disable-new-dtags"
            subprocess.check_call([pyenv, "install", version], env=env)
            subprocess.check_call([python, "-m", "pip", "install", "-U", "pip<22.1"])