51 lines
1.8 KiB
Python
51 lines
1.8 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from virtualenv.seed.wheels.embed import get_embed_wheel
|
||
|
|
|
||
|
|
from .periodic_update import periodic_update
|
||
|
|
from .util import Version, Wheel, discover_wheels
|
||
|
|
|
||
|
|
|
||
|
|
def from_bundle(distribution, version, for_py_version, search_dirs, app_data, do_periodic_update, env): # noqa: PLR0913
|
||
|
|
"""Load the bundled wheel to a cache directory."""
|
||
|
|
of_version = Version.of_version(version)
|
||
|
|
wheel = load_embed_wheel(app_data, distribution, for_py_version, of_version)
|
||
|
|
|
||
|
|
if version != Version.embed:
|
||
|
|
# 2. check if we have upgraded embed
|
||
|
|
if app_data.can_update:
|
||
|
|
per = do_periodic_update
|
||
|
|
wheel = periodic_update(distribution, of_version, for_py_version, wheel, search_dirs, app_data, per, env)
|
||
|
|
|
||
|
|
# 3. acquire from extra search dir
|
||
|
|
found_wheel = from_dir(distribution, of_version, for_py_version, search_dirs)
|
||
|
|
if found_wheel is not None and (wheel is None or found_wheel.version_tuple > wheel.version_tuple):
|
||
|
|
wheel = found_wheel
|
||
|
|
return wheel
|
||
|
|
|
||
|
|
|
||
|
|
def load_embed_wheel(app_data, distribution, for_py_version, version):
|
||
|
|
wheel = get_embed_wheel(distribution, for_py_version)
|
||
|
|
if wheel is not None:
|
||
|
|
version_match = version == wheel.version
|
||
|
|
if version is None or version_match:
|
||
|
|
with app_data.ensure_extracted(wheel.path, lambda: app_data.house) as wheel_path:
|
||
|
|
wheel = Wheel(wheel_path)
|
||
|
|
else: # if version does not match ignore
|
||
|
|
wheel = None
|
||
|
|
return wheel
|
||
|
|
|
||
|
|
|
||
|
|
def from_dir(distribution, version, for_py_version, directories):
|
||
|
|
"""Load a compatible wheel from a given folder."""
|
||
|
|
for folder in directories:
|
||
|
|
for wheel in discover_wheels(folder, distribution, version, for_py_version):
|
||
|
|
return wheel
|
||
|
|
return None
|
||
|
|
|
||
|
|
|
||
|
|
__all__ = [
|
||
|
|
"load_embed_wheel",
|
||
|
|
"from_bundle",
|
||
|
|
]
|