[Date Prev][Date Next][Thread Prev][Thread Next][Date Index][Thread Index]
[RFC PATCH v3 08/20] mkvenv: add ensure subcommand
From: |
John Snow |
Subject: |
[RFC PATCH v3 08/20] mkvenv: add ensure subcommand |
Date: |
Mon, 24 Apr 2023 16:02:36 -0400 |
This command is to be used to add various packages (or ensure they're
already present) into the configure-provided venv in a modular fashion.
Examples:
mkvenv ensure --online --dir "${source_dir}/python/wheels/" "meson>=0.61.5"
mkvenv ensure --online "sphinx>=1.6.0"
mkvenv ensure "qemu.qmp==0.0.2"
It's designed to look for packages in three places, in order:
(1) In system packages, if the version installed is already good
enough. This way your distribution-provided meson, sphinx, etc are
always used as first preference.
(2) In a vendored packages directory. Here I am suggesting
qemu.git/python/wheels/ as that directory. This is intended to serve as
a replacement for vendoring the meson source for QEMU tarballs. It is
also highly likely to be extremely useful for packaging the "qemu.qmp"
package in source distributions for platforms that do not yet package
qemu.qmp separately.
(3) Online, via PyPI, ***only when "--online" is passed***. This is only
ever used as a fallback if the first two sources do not have an
appropriate package that meets the requirement. The ability to build
QEMU and run tests *completely offline* is not impinged.
Signed-off-by: John Snow <jsnow@redhat.com>
---
python/scripts/mkvenv.py | 116 ++++++++++++++++++++++++++++++++++++++-
1 file changed, 114 insertions(+), 2 deletions(-)
diff --git a/python/scripts/mkvenv.py b/python/scripts/mkvenv.py
index 45d1b772e5..937664ea9c 100644
--- a/python/scripts/mkvenv.py
+++ b/python/scripts/mkvenv.py
@@ -13,6 +13,7 @@
create create a venv
post_init
post-venv initialization
+ ensure Ensure that the specified package is installed.
--------------------------------------------------
@@ -34,6 +35,18 @@
--gen GEN Regenerate console_scripts for given packages, if found.
--binpath BINPATH Path where console script shims should be generated
+--------------------------------------------------
+
+usage: mkvenv ensure [-h] [--online] [--dir DIR] dep_spec
+
+positional arguments:
+ dep_spec PEP 508 Dependency specification, e.g. 'meson>=0.61.5'
+
+options:
+ -h, --help show this help message and exit
+ --online Install packages from PyPI, if necessary.
+ --dir DIR Path to vendored packages where we may install from.
+
"""
# Copyright (C) 2022-2023 Red Hat, Inc.
@@ -530,6 +543,71 @@ def checkpip() -> None:
logging.debug("Pip is now (hopefully) repaired!")
+def pip_install(
+ args: Sequence[str],
+ online: bool = False,
+ wheels_dir: Optional[Union[str, Path]] = None,
+) -> None:
+ """
+ Use pip to install a package or package(s) as specified in @args.
+ """
+ loud = bool(
+ os.environ.get("DEBUG")
+ or os.environ.get("GITLAB_CI")
+ or os.environ.get("V")
+ )
+
+ full_args = [
+ sys.executable,
+ "-m",
+ "pip",
+ "install",
+ "--disable-pip-version-check",
+ "-v" if loud else "-q",
+ ]
+ if not online:
+ full_args += ["--no-index"]
+ if wheels_dir:
+ full_args += ["--find-links", f"file://{str(wheels_dir)}"]
+ full_args += list(args)
+ subprocess.run(full_args, check=True)
+
+
+def ensure(
+ dep_spec: str,
+ online: bool = False,
+ wheels_dir: Optional[Union[str, Path]] = None,
+) -> None:
+ """
+ Use pip to ensure we have the package specified by @dep_spec.
+
+ If the package is already installed, do nothing. If online and
+ wheels_dir are both provided, prefer packages found in wheels_dir
+ first before connecting to PyPI.
+
+ :param dep_spec:
+ A PEP 508 dependency specification. e.g. 'meson>=0.61.5'.
+ :param online: If True, fall back to PyPI.
+ :param wheels_dir: If specified, search this path for packages.
+ """
+ # This first install command will:
+ # (A) Do nothing, if we already have a suitable package.
+ # (B) Install the package from vendored source, if possible.
+ # (C) Fail if neither A nor B.
+ try:
+ pip_install([dep_spec], online=False, wheels_dir=wheels_dir)
+ # (A) or (B) happened. Success.
+ return
+ except subprocess.CalledProcessError:
+ # (C) Happened.
+ # The package is missing or isn't a suitable version,
+ # and we weren't able to install a suitable vendored package.
+ if online:
+ pip_install([dep_spec], online=True)
+ else:
+ raise
+
+
def post_venv_setup(bin_path: str, packages: Sequence[str] = ()) -> None:
"""
This is intended to be run *inside the venv* after it is created.
@@ -578,6 +656,29 @@ def _add_post_init_subcommand(subparsers: Any) -> None:
)
+def _add_ensure_subcommand(subparsers: Any) -> None:
+ subparser = subparsers.add_parser(
+ "ensure", help="Ensure that the specified package is installed."
+ )
+ subparser.add_argument(
+ "--online",
+ action="store_true",
+ help="Install packages from PyPI, if necessary.",
+ )
+ subparser.add_argument(
+ "--dir",
+ type=str,
+ action="store",
+ help="Path to vendored packages where we may install from.",
+ )
+ subparser.add_argument(
+ "dep_spec",
+ type=str,
+ action="store",
+ help="PEP 508 Dependency specification, e.g. 'meson>=0.61.5'",
+ )
+
+
def main() -> int:
"""CLI interface to make_qemu_venv. See module docstring."""
if os.environ.get("DEBUG") or os.environ.get("GITLAB_CI"):
@@ -600,14 +701,18 @@ def main() -> int:
_add_create_subcommand(subparsers)
_add_post_init_subcommand(subparsers)
+ _add_ensure_subcommand(subparsers)
args = parser.parse_args()
script_packages = []
- for element in args.gen or ():
- script_packages.extend(element.split(","))
+
+ def _normalize_gen() -> None:
+ for element in args.gen or ():
+ script_packages.extend(element.split(","))
try:
if args.command == "create":
+ _normalize_gen()
make_venv(
args.target,
system_site_packages=True,
@@ -615,7 +720,14 @@ def main() -> int:
script_packages=script_packages,
)
if args.command == "post_init":
+ _normalize_gen()
post_venv_setup(args.binpath, script_packages)
+ if args.command == "ensure":
+ ensure(
+ dep_spec=args.dep_spec,
+ online=args.online,
+ wheels_dir=args.dir,
+ )
logger.debug("mkvenv.py %s: exiting", args.command)
except Ouch as exc:
print("\n*** Ouch! ***\n", file=sys.stderr)
--
2.39.2
- [RFC PATCH v3 00/20] configure: create a python venv and ensure meson, sphinx, John Snow, 2023/04/24
- [RFC PATCH v3 05/20] mkvenv: generate console entry shims from inside the venv, John Snow, 2023/04/24
- [RFC PATCH v3 03/20] mkvenv: add console script entry point generation, John Snow, 2023/04/24
- [RFC PATCH v3 06/20] mkvenv: work around broken pip installations on Debian 10, John Snow, 2023/04/24
- [RFC PATCH v3 04/20] mkvenv: Add better error message for missing pyexpat module, John Snow, 2023/04/24
- [RFC PATCH v3 02/20] python: add mkvenv.py, John Snow, 2023/04/24
- [RFC PATCH v3 08/20] mkvenv: add ensure subcommand,
John Snow <=
- [RFC PATCH v3 07/20] mkvenv: add nested venv workaround, John Snow, 2023/04/24
- [RFC PATCH v3 10/20] tests/vm: Configure netbsd to use Python 3.10, John Snow, 2023/04/24
- [RFC PATCH v3 09/20] tests/docker: add python3-venv dependency, John Snow, 2023/04/24
- [RFC PATCH v3 11/20] tests/vm: add py310-expat to NetBSD, John Snow, 2023/04/24
- [RFC PATCH v3 13/20] configure: create a python venv unconditionally, John Snow, 2023/04/24
- [RFC PATCH v3 15/20] configure: add --enable-pypi and --disable-pypi, John Snow, 2023/04/24