Fox-V3/fifo/datetime_cron_converters.py

from datetime import datetime
from typing import TYPE_CHECKING

from apscheduler.triggers.cron import CronTrigger
from discord.ext.commands import BadArgument, Converter
from dateutil import parser

from fifo.timezones import assemble_timezones

if TYPE_CHECKING:
    DatetimeConverter = datetime
    CronConverter = str
else:
    class DatetimeConverter(Converter):
        async def convert(self, ctx, argument) -> datetime:
            dt = parser.parse(argument, fuzzy=True, tzinfos=assemble_timezones())
            if dt is not None:
                return dt
            raise BadArgument()

    class CronConverter(Converter):
        async def convert(self, ctx, argument) -> str:
            try:
                CronTrigger.from_crontab(argument)
            except ValueError:
                raise BadArgument()

            return argument