mirror of
https://github.com/thegeeklab/docker-tidy.git
synced 2024-11-22 12:10:40 +00:00
52 lines
1.1 KiB
Python
52 lines
1.1 KiB
Python
#!/usr/bin/env python3
|
|
"""Custom input type parser."""
|
|
|
|
import datetime
|
|
|
|
import environs
|
|
from dateutil import tz
|
|
from pytimeparse import timeparse
|
|
|
|
env = environs.Env()
|
|
|
|
|
|
def timedelta_validator(value):
|
|
"""Return the :class:`datetime.datetime.DateTime` for a time in the past.
|
|
|
|
:param value: a string containing a time format supported by
|
|
mod:`pytimeparse`
|
|
"""
|
|
if value is None:
|
|
return None
|
|
|
|
try:
|
|
_datetime_seconds_ago(timeparse.timeparse(value))
|
|
return value
|
|
except TypeError:
|
|
raise
|
|
|
|
|
|
def timedelta(value):
|
|
"""Return the :class:`datetime.datetime.DateTime` for a time in the past.
|
|
|
|
:param value: a string containing a time format supported by
|
|
mod:`pytimeparse`
|
|
"""
|
|
if value is None:
|
|
return None
|
|
return _datetime_seconds_ago(timeparse.timeparse(value))
|
|
|
|
|
|
def _datetime_seconds_ago(seconds):
|
|
now = datetime.datetime.now(tz.tzutc())
|
|
return now - datetime.timedelta(seconds=seconds)
|
|
|
|
|
|
@env.parser_for("timedelta_validator")
|
|
def timedelta_parser(value):
|
|
try:
|
|
timedelta_validator(value)
|
|
return value
|
|
except TypeError as e:
|
|
raise environs.EnvError(e)
|