Source code for impscan.lookup.req_spec

from __future__ import annotations

from enum import Enum

__all__ = ["ReqSpec", "CondaReqSpec", "PyPIReqSpec"]


class Repository(Enum):
    Conda = "conda"
    PyPI = "pypi"


class ReqSpec:
    def __init__(
        self,
        package: str,
        repository: Repository,
        channel: list[str],
        constraints: list[str],
    ):
        self.package = package
        self.repository = repository
        self.channel = channel
        self.constraints = constraints


class CondaReqSpec(ReqSpec):
    def __init__(self, package: str, channel: list[str], constraints: list[str]):
        super().__init__(
            package=package,
            repository=Repository.Conda,
            channel=channel,
            constraints=constraints,
        )


class PyPIReqSpec(ReqSpec):
    def __init__(self, package: str, channel: list[str], constraints: list[str]):
        super().__init__(
            package=package,
            repository=Repository.PyPI,
            channel=None,
            constraints=constraints,
        )