Source code for pyobs.robotic.scheduler.merits.aftertime

from __future__ import annotations

from typing import TYPE_CHECKING

from astropydantic import AstroPydanticTime  # type: ignore
from pydantic import Field

from pyobs.utils.time import Time

from .merit import Merit

if TYPE_CHECKING:
    from pyobs.robotic import Task

    from ..dataprovider import DataProvider




class AfterTimeMerit(Merit):
    """Merit function that gives 1 after a given time."""

    time: AstroPydanticTime = Field(default_factory=Time.now)

    async def __call__(self, time: Time, task: Task, data: DataProvider) -> float:
        return 1.0 if time >= self.time else 0.0



__all__ = ["AfterTimeMerit"]