do some whl_platform_tag processing

Author: aignas

python/private/pypi/select_whl.bzl

@@ -105,6 +105,46 @@ def _python_tag_priority(*, tag, implementation, py_version):
         version.key(ver),
     )
 
+def _filter_platform_tags(tags):
+    ret = []
+    replacements = {}
+    for tag in tags:
+        if not (
+            tag.startswith(_ANDROID) or
+            tag.startswith(_IOS) or
+            tag.startswith(_MACOSX) or
+            tag.startswith(_MANYLINUX) or
+            tag.startswith(_MUSLLINUX)
+        ):
+            ret.append(tag)
+            continue
+
+        want_os, sep, tail = tag.partition("_")
+        if not sep:
+            fail("could not parse the tag")
+
+        want_major, _, tail = tail.partition("_")
+        if want_major == "*":
+            # the expected match is any version
+            want_arch = tail
+        elif want_os.startswith(_ANDROID):
+            want_arch = tail
+        else:
+            # drop the minor version segment
+            _, _, want_arch = tail.partition("_")
+
+        placeholder = "{}_*_{}".format(want_os, want_arch)
+        replacements[placeholder] = tag
+        if placeholder in ret:
+            ret.remove(placeholder)
+
+        ret.append(placeholder)
+
+    return [
+        replacements.get(p, p)
+        for p in ret
+    ]
+
 def _candidates_by_priority(
         *,
         whls,
@@ -121,6 +161,9 @@ def _candidates_by_priority(
     """
     py_version = version.parse(python_version, strict = True)
     implementation = python_tag(implementation_name)
+    logger.debug(lambda: "input:  {}".format(whl_platform_tags))
+    whl_platform_tags = _filter_platform_tags(whl_platform_tags)
+    logger.debug(lambda: "output: {}".format(whl_platform_tags))
 
     ret = {}
     for whl in whls:

tests/pypi/select_whl/select_whl_tests.bzl

@@ -423,12 +423,12 @@ def _test_multiple_musllinux_exact_params(env):
         whl_abi_tags = ["none"],
         python_version = "3.12",
         limit = 2,
+        debug = True,
     )
     _match(
         env,
         got,
-        # select the one with the lowest version, because of the input to the function
-        "pkg-0.0.1-py3-none-musllinux_1_2_x86_64.whl",
+        # 1.2 is not within the candidates because it is not compatible
         "pkg-0.0.1-py3-none-musllinux_1_1_x86_64.whl",
     )