[Wheel Variants] Wheel Variant Support for PIP

src/pip/__init__.py

@@ -1,6 +1,6 @@
 from typing import List, Optional
 
-__version__ = "25.1.dev0"
+__version__ = "25.1.dev0+pep-xxx-wheel-variants"
 
 
 def main(args: Optional[List[str]] = None) -> int:

src/pip/_internal/cache.py

@@ -16,6 +16,7 @@
 from pip._internal.models.wheel import Wheel
 from pip._internal.utils.temp_dir import TempDirectory, tempdir_kinds
 from pip._internal.utils.urls import path_to_url
+from pip._internal.utils.variant import variant_wheel_supported
 
 logger = logging.getLogger(__name__)
 
@@ -150,7 +151,10 @@ def get(
                     package_name,
                 )
                 continue
-            if not wheel.supported(supported_tags):
+            if (
+                not wheel.supported(supported_tags)
+                or not variant_wheel_supported(wheel, link)
+            ):
                 # Built for a different python/arch/etc
                 continue
             candidates.append(

src/pip/_internal/cli/req_command.py

@@ -337,6 +337,7 @@ def _build_package_finder(
             allow_all_prereleases=options.pre,
             prefer_binary=options.prefer_binary,
             ignore_requires_python=ignore_requires_python,
+            use_variants=options.use_variants,
         )
 
         return PackageFinder.create(

src/pip/_internal/commands/index.py

@@ -116,7 +116,7 @@ def get_available_package_versions(self, options: Values, args: List[Any]) -> No
             )
 
             versions: Iterable[Version] = (
-                candidate.version for candidate in finder.find_all_candidates(query)
+                candidate.version for candidate, _ in finder.find_all_candidates(query)
             )
 
             if not options.pre:

src/pip/_internal/commands/install.py

@@ -270,6 +270,16 @@ def add_options(self) -> None:
             ),
         )
 
+        self.cmd_opts.add_option(
+            "--no-variant",
+            dest="use_variants",
+            action="store_false",
+            default=True,
+            help=(
+                "Disable installing variant wheels."
+            ),
+        )
+
     @with_cleanup
     def run(self, options: Values, args: List[str]) -> int:
         if options.use_user_site and options.target_dir is not None:
@@ -314,8 +324,8 @@ def run(self, options: Values, args: List[str]) -> int:
             options.target_dir = os.path.abspath(options.target_dir)
             if (
                 # fmt: off
-                os.path.exists(options.target_dir) and
-                not os.path.isdir(options.target_dir)
+                os.path.exists(options.target_dir)
+                and not os.path.isdir(options.target_dir)
                 # fmt: on
             ):
                 raise CommandError(
@@ -397,13 +407,24 @@ def run(self, options: Values, args: List[str]) -> int:
 
             if options.dry_run:
                 would_install_items = sorted(
-                    (r.metadata["name"], r.metadata["version"])
+                    (
+                        r.metadata["name"],
+                        r.metadata["version"],
+                        r.metadata["variant-hash"],
+                    )
+                    if "variant-hash" in r.metadata
+                    else (r.metadata["name"], r.metadata["version"])
                     for r in requirement_set.requirements_to_install
                 )
+
                 if would_install_items:
                     write_output(
                         "Would install %s",
-                        " ".join("-".join(item) for item in would_install_items),
+                        " ".join(
+                            "-".join(item[:2])
+                            + (f"-{item[2]}" if len(item) > 2 else "")
+                            for item in would_install_items
+                        ),
                     )
                 return SUCCESS
 
@@ -482,14 +503,20 @@ def run(self, options: Values, args: List[str]) -> int:
             summary = []
             installed_versions = {}
             for distribution in env.iter_all_distributions():
-                installed_versions[distribution.canonical_name] = distribution.version
+                installed_versions[distribution.canonical_name] = (
+                    distribution.version,
+                    distribution.metadata["variant-hash"],
+                )
             for package in installed:
                 display_name = package.name
-                version = installed_versions.get(canonicalize_name(display_name), None)
+                version, variant_hash = installed_versions.get(
+                    canonicalize_name(display_name), (None, None)
+                )
+                text = f"{display_name}"
                 if version:
-                    text = f"{display_name}-{version}"
-                else:
-                    text = display_name
+                    text = f"{text}-{version}"
+                if variant_hash:
+                    text = f"{text}-{variant_hash}"
                 summary.append(text)
 
             if conflicts is not None:

src/pip/_internal/commands/list.py

@@ -243,7 +243,7 @@ def iter_packages_latest_infos(
             def latest_info(
                 dist: "_DistWithLatestInfo",
             ) -> Optional["_DistWithLatestInfo"]:
-                all_candidates = finder.find_all_candidates(dist.canonical_name)
+                all_candidates, _ = finder.find_all_candidates(dist.canonical_name)
                 if not options.pre:
                     # Remove prereleases
                     all_candidates = [