diff --git a/src/utils/datetime/parse.py b/src/utils/datetime/parse.py index 8e372f8c..51941d78 100644 --- a/src/utils/datetime/parse.py +++ b/src/utils/datetime/parse.py @@ -6,6 +6,12 @@ from .common import * def iso8601(s: str) -> _datetime.datetime: return dateutil.parser.parse(s) +def date_human(s: str) -> typing.Optional[_datetime.datetime]: + try: + return _datetime.datetime.strptime(s, DATE_HUMAN) + except ValueError: + return None + REGEX_PRETTYTIME = re.compile( r"(?:(\d+)w)?(?:(\d+)d)?(?:(\d+)h)?(?:(\d+)m)?(?:(\d+)s)?", re.I) diff --git a/src/utils/parse/spec.py b/src/utils/parse/spec.py index 4e30c294..1aa24610 100644 --- a/src/utils/parse/spec.py +++ b/src/utils/parse/spec.py @@ -1,6 +1,7 @@ import enum, typing from .time import duration from .types import try_int +from src.utils.datetime.parse import date_human class SpecArgumentContext(enum.IntFlag): CHANNEL = 1 @@ -67,6 +68,14 @@ class SpecArgumentTypeDuration(SpecArgumentType): def error(self) -> typing.Optional[str]: return "Invalid timeframe" +class SpecArgumentTypeDate(SpecArgumentType): + def name(self): + return SpecArgumentType.name(self) or "yyyy-mm-dd" + def simple(self, args): + if args: + return date_human(args[0], 1) + return None, 1 + class SpecArgumentPrivateType(SpecArgumentType): context = SpecArgumentContext.PRIVATE @@ -77,6 +86,7 @@ SPEC_ARGUMENT_TYPES = { "string": SpecArgumentTypeString, "tstring": SpecArgumentTypeTrimString, "int": SpecArgumentTypeInt, + "date": SpecArgumentTypeDate, "duration": SpecArgumentTypeDuration }