requirements.py

from __future__ import annotations

import itertools
import sys
from collections.abc import Iterable

from packaging.requirements import Requirement

from ts_utils.metadata import read_dependencies, read_stubtest_settings
from ts_utils.paths import STUBS_PATH


def get_external_stub_requirements(distributions: Iterable[str] = ()) -> set[Requirement]:
    if not distributions:
        distributions = [distribution.name for distribution in STUBS_PATH.iterdir()]

    return set(itertools.chain.from_iterable([read_dependencies(distribution).external_pkgs for distribution in distributions]))


def get_stubtest_system_requirements(distributions: Iterable[str] = (), platform: str = sys.platform) -> list[str]:
    if not distributions:
        distributions = [distribution.name for distribution in STUBS_PATH.iterdir()]

    requirements: list[str] = []
    for distribution in distributions:
        requirements.extend(read_stubtest_settings(distribution).system_requirements_for_platform(platform))
    return requirements