Add support for verifying tags on install

Signed-off-by: Michał Górny <mgorny@gentoo.org>

Author: mgorny

gpep517/__main__.py

@@ -161,13 +161,16 @@ def add_install_args(parser):
                        action="store_true",
                        help="Symlink .pyc files between optimization levels "
                        "if their contents match")
-
     group.add_argument("--symlink-to",
                        type=PurePath,
                        help="Install symlinks to another directory rather "
                        "than files if they match respective paths "
                        "in the other directory (useful for deduplicating "
                        "packages across Python implementations)")
+    group.add_argument("--verify-tags",
+                       action="store_true",
+                       help="Verify wheel compatibility via platform "
+                       "compatibility tags (requires packaging)")
 
 
 def main(argv=sys.argv):

gpep517/install.py

@@ -12,6 +12,17 @@
 from gpep517.utils import DEFAULT_PREFIX, install_scheme_dict, logger
 
 
+def verify_tags(wheel: Path) -> None:
+    from packaging.tags import sys_tags
+    from packaging.utils import parse_wheel_filename
+
+    wheel_tags = parse_wheel_filename(str(wheel.name))[-1]
+    if not any(compatible_tag in wheel_tags
+               for compatible_tag in sys_tags()):
+        raise RuntimeError(
+            f"Wheel {wheel} is not compatible with the system")
+
+
 def install_wheel_impl(args, wheel: Path):
     from installer import install
     from installer.destinations import SchemeDictionaryDestination
@@ -128,6 +139,9 @@ def _compile_bytecode(self,
                     pyc2.unlink()
                     pyc2.symlink_to(pyc1.name)
 
+    if args.verify_tags:
+        verify_tags(wheel)
+
     with WheelFile.open(wheel) as source:
         dest = DeduplicatingDestination(
             install_scheme_dict(args.prefix or DEFAULT_PREFIX,

pyproject.toml

@@ -25,7 +25,12 @@ dependencies = [
 ]
 
 [project.optional-dependencies]
+tags = [
+    "packaging >= 20.9",
+]
 test = [
+    # for abi3t tags
+    "packaging >= 26.1",
     "pytest",
 ]
 test-full = [

test/test_install.py

@@ -1,14 +1,17 @@
-# (c) 2022-2025 Michał Górny
+# (c) 2022-2026 Michał Górny
 # SPDX-License-Identifier: GPL-2.0-or-later
 
 import contextlib
 import importlib.machinery
 import importlib.util
 import os
 import pathlib
+import shutil
+import sys
 import sysconfig
 import typing
 
+import packaging.tags
 import pytest
 
 from gpep517 import __version__
@@ -379,3 +382,49 @@ def test_install_symlink_chain(tmp_path,
     assert expected_links == {path: tmp_path.joinpath(path).readlink()
                               for path, path_data in expected.items()
                               if path_data is not None and path_data[1]}
+
+
+BEST_TAG = next(packaging.tags.sys_tags())
+FREETHREADING = bool(sysconfig.get_config_var("Py_GIL_DISABLED"))
+IS_CPYTHON = sys.implementation.name == "cpython"
+
+@pytest.mark.parametrize(
+    ("wheel_name", "expected"),
+    [
+        ("test-1-py3-none-any.whl", True),
+        ("test-1-py2.py3-none-any.whl", True),
+        ("test-1-py38-none-any.whl", sys.version_info >= (3, 8)),
+        (f"test-1-{BEST_TAG}.whl", True),
+        (f"test-1-{BEST_TAG.interpreter}-none-any.whl", IS_CPYTHON),
+        (f"test-1-{BEST_TAG.interpreter}-{BEST_TAG.abi}-any.whl", False),
+        (f"test-1-{BEST_TAG.interpreter}-abi3-any.whl", False),
+        (f"test-1-py3-none-{BEST_TAG.platform}.whl", True),
+        (f"test-1-py38-none-{BEST_TAG.platform}.whl", sys.version_info >= (3, 8)),
+        (f"test-1-{BEST_TAG.interpreter}-none-{BEST_TAG.platform}.whl", True),
+        (f"test-1-{BEST_TAG.interpreter}-abi3-{BEST_TAG.platform}.whl", IS_CPYTHON and not FREETHREADING),
+        (f"test-1-{BEST_TAG.interpreter}-abi3t-{BEST_TAG.platform}.whl", IS_CPYTHON and FREETHREADING and sys.version_info >= (3, 15)),
+        (f"test-1-{BEST_TAG.interpreter}-abi3.abi3t-{BEST_TAG.platform}.whl", IS_CPYTHON),
+        (f"test-1-cp38-abi3-{BEST_TAG.platform}.whl", IS_CPYTHON and sys.version_info >= (3, 8) and not FREETHREADING),
+        (f"test-1-cp38-abi3.abi3t-{BEST_TAG.platform}.whl", IS_CPYTHON and sys.version_info >= (3, 8)),
+        (f"test-1-cp315-abi3.abi3t-{BEST_TAG.platform}.whl", IS_CPYTHON and sys.version_info >= (3, 15)),
+        (f"test-1-py3-none-win32.whl", sysconfig.get_platform() == "win32"),
+        (f"test-1-py3-none-linux_x86_64.whl", sysconfig.get_platform() == "linux-x86_64"),
+    ]
+)
+def test_verify_tags(tmp_path,
+                     wheel_name: str,
+                     expected: bool,
+                     ) -> None:
+    shutil.copy("test/test-pkg/dist/test-1-py3-none-any.whl",
+                tmp_path / wheel_name)
+
+    ctx_mgr = ( contextlib.nullcontext() if expected else
+               pytest.raises(RuntimeError,
+                             match="not compatible with the system"))
+    args = (["", "install-wheel",
+             "--destdir", str(tmp_path),
+             "--verify-tags",
+             str(tmp_path / wheel_name)]
+            )
+    with ctx_mgr:
+        assert 0 == main(args)

test/test_integration.py

@@ -105,7 +105,8 @@ def test_integration_install(tmp_path, buildsys, verify_zipfile_cleanup,
     with pushd(tmp_path):
         assert 0 == main(["", "install-from-source",
                           "--destdir", str(destdir),
-                          "--prefix", "/usr"])
+                          "--prefix", "/usr",
+                          "--verify-tags"])
 
     sitedir = destdir / (sysconfig.get_path("purelib", vars={"base": "/usr"})
                          .lstrip(os.path.sep))