from classes.comparator import (
    DictComparator,
    ListComparator,
    NumberComparator,
    TextComparator,
)
from scheduler.block import BaseBlock


class NumberCondition(BaseBlock):
    def __init__(self, comparator: NumberComparator):
        self.comparator = comparator

    def __call__(self, input: int) -> bool:
        return self.comparator(input)


class TextCondition(BaseBlock):
    def __init__(self, comparator: TextComparator):
        self.comparator = comparator

    def __call__(self, input: str) -> bool:
        return self.comparator(input)


class ListCondition(BaseBlock):
    def __init__(self, comparator: ListComparator):
        self.comparator = comparator

    def __call__(self, input: list) -> bool:
        return self.comparator(input)


class DictCondition(BaseBlock):
    def __init__(self, comparator: DictComparator):
        self.comparator = comparator

    def __call__(self, input: dict) -> bool:
        return self.comparator(input)