Backups/multi-user
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: handle multiple platform for wheels extraction

Browse Source
This commit is contained in:
Swann
2024-02-28 22:42:31 +01:00
parent 8a3adc6cfa
commit 037be421cb
2 changed files with 12 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
multi_user/environment.py
View File

@ -52,7 +52,7 @@ def preload_modules():
wheels.load_wheel_global("ordered_set", "ordered_set") wheels.load_wheel_global("ordered_set", "ordered_set")
wheels.load_wheel_global("deepdiff", "deepdiff") wheels.load_wheel_global("deepdiff", "deepdiff")
wheels.load_wheel_global("replication", "replication") wheels.load_wheel_global("replication", "replication")
wheels.load_wheel_global("zmq", "pyzmq") wheels.load_wheel_global("zmq", "pyzmq", match_platform=True)

15
multi_user/wheels/__init__.py
View File

@ -5,8 +5,11 @@ import importlib
from pathlib import Path from pathlib import Path
import sys import sys
import logging import logging
import sysconfig
from types import ModuleType from types import ModuleType
from typing import Iterator, Iterable from typing import Iterator, Iterable
import zipfile
_my_dir = Path(__file__).parent _my_dir = Path(__file__).parent
_log = logging.getLogger(__name__) _log = logging.getLogger(__name__)
@ -77,10 +80,10 @@ def load_wheel_global(module_name: str, fname_prefix: str = "", match_platform:
) )
return module return module
wheel = _wheel_filename(fname_prefix) wheel = _wheel_filename(fname_prefix, match_platform=match_platform)
wheel_filepath = str(wheel) wheel_filepath = str(wheel)
import zipfile
wheel_archive = zipfile.ZipFile(wheel_filepath) wheel_archive = zipfile.ZipFile(wheel_filepath)
@ -123,8 +126,12 @@ def _sys_path_mod_backup(wheel_file: Path) -> Iterator[None]:
sys.modules.update(old_sysmod) sys.modules.update(old_sysmod)
def _wheel_filename(fname_prefix: str) -> Path: def _wheel_filename(fname_prefix: str, match_platform: bool = False) -> Path:
path_pattern = "%s*.whl" % fname_prefix if match_platform:
platform_tag = sysconfig.get_platform().replace('-','_').replace('.','_')
path_pattern = f"{fname_prefix}*{platform_tag}.whl"
else:
path_pattern = f"{fname_prefix}*.whl"
wheels: list[Path] = list(_my_dir.glob(path_pattern)) wheels: list[Path] = list(_my_dir.glob(path_pattern))
if not wheels: if not wheels:
raise RuntimeError("Unable to find wheel at %r" % path_pattern) raise RuntimeError("Unable to find wheel at %r" % path_pattern)