supported_wheels.py

#!/usr/bin/env python
"""Filter out wheel filenames not supported on this platform."""

from __future__ import annotations

import sys
from os.path import basename
from typing import TYPE_CHECKING

from packaging.tags import sys_tags

if TYPE_CHECKING:
    from collections.abc import Iterator

try:
    from wheel.install import WHEEL_INFO_RE as wheel_matcher  # noqa: N811
except ImportError:  # As of Wheel 0.32.0
    from wheel.wheelfile import WHEEL_INFO_RE

    wheel_matcher = WHEEL_INFO_RE.match


def tags_for(fname: str) -> Iterator[tuple[str, str, str]]:
    """Copied from WheelFile code."""  # noqa: D401
    parsed_filename = wheel_matcher(basename(fname))  # noqa: PTH119
    tags = parsed_filename.groupdict()
    for pyver in tags['pyver'].split('.'):
        for abi in tags['abi'].split('.'):
            for plat in tags['plat'].split('.'):
                yield (pyver, abi, plat)


def main() -> None:
    """Print filenames of all supported wheels."""
    supported = {(tag.interpreter, tag.abi, tag.platform) for tag in sys_tags()}
    for fname in sys.argv[1:]:
        tags = set(tags_for(fname))
        if supported.intersection(tags):
            print(fname)  # noqa: T201


if __name__ == '__main__':
    main()