summaryrefslogtreecommitdiffstats
path: root/src/ansiblelint/requirements.py
blob: 96381b980f3f8608378b72db3966126b345abb46 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
"""Utilities for checking python packages requirements."""

import importlib_metadata
from packaging.requirements import Requirement
from packaging.specifiers import SpecifierSet
from packaging.version import Version


class Reqs(dict[str, SpecifierSet]):
    """Utility class for working with package dependencies."""

    reqs: dict[str, SpecifierSet]

    def __init__(self, name: str = "ansible-lint") -> None:
        """Load linter metadata requirements."""
        for req_str in importlib_metadata.metadata(name).json["requires_dist"]:
            req = Requirement(req_str)
            if req.name:
                self[req.name] = req.specifier

    def matches(self, req_name: str, req_version: str | Version) -> bool:
        """Verify if given version is matching current metadata dependencies."""
        if req_name not in self:
            return False
        return all(
            specifier.contains(str(req_version), prereleases=True)
            for specifier in self[req_name]
        )