mirror of
https://github.com/brandons209/Red-bot-Cogs.git
synced 2024-09-30 17:17:35 +13:00
53 lines
1.4 KiB
Python
53 lines
1.4 KiB
Python
|
# thanks to @Sinbad for time parsing!
|
||
|
from __future__ import annotations
|
||
|
|
||
|
import re
|
||
|
from datetime import datetime as dt, timedelta
|
||
|
from typing import Optional
|
||
|
|
||
|
import pytz
|
||
|
from dateutil import parser
|
||
|
from dateutil.tz import gettz
|
||
|
|
||
|
TIME_RE_STRING = r"\s?".join(
|
||
|
[
|
||
|
r"((?P<weeks>\d+?)\s?(weeks?|w))?",
|
||
|
r"((?P<days>\d+?)\s?(days?|d))?",
|
||
|
r"((?P<hours>\d+?)\s?(hours?|hrs|hr?))?",
|
||
|
r"((?P<minutes>\d+?)\s?(minutes?|mins?|m(?!o)))?", # prevent matching "months"
|
||
|
r"((?P<seconds>\d+?)\s?(seconds?|secs?|s))?",
|
||
|
]
|
||
|
)
|
||
|
|
||
|
TIME_RE = re.compile(TIME_RE_STRING, re.I)
|
||
|
|
||
|
|
||
|
def gen_tzinfos():
|
||
|
for zone in pytz.common_timezones:
|
||
|
try:
|
||
|
tzdate = pytz.timezone(zone).localize(dt.utcnow(), is_dst=None)
|
||
|
except pytz.NonExistentTimeError:
|
||
|
pass
|
||
|
else:
|
||
|
tzinfo = gettz(zone)
|
||
|
|
||
|
if tzinfo:
|
||
|
yield tzdate.tzname(), tzinfo
|
||
|
|
||
|
|
||
|
def parse_time(datetimestring: str):
|
||
|
tzinfo = dict(gen_tzinfos())
|
||
|
ret = parser.parse(datetimestring, tzinfos=tzinfo)
|
||
|
if ret.tzinfo is not None:
|
||
|
ret = ret.astimezone(pytz.utc)
|
||
|
return ret
|
||
|
|
||
|
|
||
|
def parse_timedelta(argument: str) -> Optional[timedelta]:
|
||
|
matches = TIME_RE.match(argument)
|
||
|
if matches:
|
||
|
params = {k: int(v) for k, v in matches.groupdict().items() if v}
|
||
|
if params:
|
||
|
return timedelta(**params)
|
||
|
return None
|