| | """Represents a wheel file and provides access to the various parts of the |
| | name that have meaning. |
| | """ |
| | import re |
| | from typing import Dict, Iterable, List |
| |
|
| | from pip._vendor.packaging.tags import Tag |
| |
|
| | from pip._internal.exceptions import InvalidWheelFilename |
| |
|
| |
|
| | class Wheel: |
| | """A wheel file""" |
| |
|
| | wheel_file_re = re.compile( |
| | r"""^(?P<namever>(?P<name>[^\s-]+?)-(?P<ver>[^\s-]*?)) |
| | ((-(?P<build>\d[^-]*?))?-(?P<pyver>[^\s-]+?)-(?P<abi>[^\s-]+?)-(?P<plat>[^\s-]+?) |
| | \.whl|\.dist-info)$""", |
| | re.VERBOSE, |
| | ) |
| |
|
| | def __init__(self, filename: str) -> None: |
| | """ |
| | :raises InvalidWheelFilename: when the filename is invalid for a wheel |
| | """ |
| | wheel_info = self.wheel_file_re.match(filename) |
| | if not wheel_info: |
| | raise InvalidWheelFilename(f"{filename} is not a valid wheel filename.") |
| | self.filename = filename |
| | self.name = wheel_info.group("name").replace("_", "-") |
| | |
| | |
| | self.version = wheel_info.group("ver").replace("_", "-") |
| | self.build_tag = wheel_info.group("build") |
| | self.pyversions = wheel_info.group("pyver").split(".") |
| | self.abis = wheel_info.group("abi").split(".") |
| | self.plats = wheel_info.group("plat").split(".") |
| |
|
| | |
| | self.file_tags = { |
| | Tag(x, y, z) for x in self.pyversions for y in self.abis for z in self.plats |
| | } |
| |
|
| | def get_formatted_file_tags(self) -> List[str]: |
| | """Return the wheel's tags as a sorted list of strings.""" |
| | return sorted(str(tag) for tag in self.file_tags) |
| |
|
| | def support_index_min(self, tags: List[Tag]) -> int: |
| | """Return the lowest index that one of the wheel's file_tag combinations |
| | achieves in the given list of supported tags. |
| | |
| | For example, if there are 8 supported tags and one of the file tags |
| | is first in the list, then return 0. |
| | |
| | :param tags: the PEP 425 tags to check the wheel against, in order |
| | with most preferred first. |
| | |
| | :raises ValueError: If none of the wheel's file tags match one of |
| | the supported tags. |
| | """ |
| | return min(tags.index(tag) for tag in self.file_tags if tag in tags) |
| |
|
| | def find_most_preferred_tag( |
| | self, tags: List[Tag], tag_to_priority: Dict[Tag, int] |
| | ) -> int: |
| | """Return the priority of the most preferred tag that one of the wheel's file |
| | tag combinations achieves in the given list of supported tags using the given |
| | tag_to_priority mapping, where lower priorities are more-preferred. |
| | |
| | This is used in place of support_index_min in some cases in order to avoid |
| | an expensive linear scan of a large list of tags. |
| | |
| | :param tags: the PEP 425 tags to check the wheel against. |
| | :param tag_to_priority: a mapping from tag to priority of that tag, where |
| | lower is more preferred. |
| | |
| | :raises ValueError: If none of the wheel's file tags match one of |
| | the supported tags. |
| | """ |
| | return min( |
| | tag_to_priority[tag] for tag in self.file_tags if tag in tag_to_priority |
| | ) |
| |
|
| | def supported(self, tags: Iterable[Tag]) -> bool: |
| | """Return whether the wheel is compatible with one of the given tags. |
| | |
| | :param tags: the PEP 425 tags to check the wheel against. |
| | """ |
| | return not self.file_tags.isdisjoint(tags) |
| |
|
