osmarks
/
autobotrobot
mirror of https://github.com/osmarks/autobotrobot
1
0
Fork 0
Code Issues Releases Wiki Activity

fortnight support

This commit is contained in:
osmarks 2020-10-30 12:33:55 +00:00
parent 1f77fb9c1f
commit 6ad622e027
1 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/util.py
View File

@ -25,13 +25,14 @@ number = "(-?[0-9]+(?:\.[0-9]+)?)(" + "|".join(prefixes.keys()) + ")?"
# from here: https://github.com/Rapptz/RoboDanny/blob/18b92ae2f53927aedebc25fb5eca02c8f6d7a874/cogs/utils/time.py
short_timedelta_regex = re.compile(f"""
(?:(?P<years>{number})(?:years?|y))? # e.g. 2y
(?:(?P<months>{number})(?:months?|mo))? # e.g. 2months
(?:(?P<weeks>{number})(?:weeks?|w))? # e.g. 10w
(?:(?P<days>{number})(?:days?|d))? # e.g. 14d
(?:(?P<hours>{number})(?:hours?|h))? # e.g. 12h
(?:(?P<minutes>{number})(?:minutes?|m))? # e.g. 10m
(?:(?P<seconds>{number})(?:seconds?|s))? # e.g. 15s """, re.VERBOSE)
(?:(?P<years>{number})(?:years?|y))? # e.g. 2y
(?:(?P<months>{number})(?:months?|mo))? # e.g. 2months
(?:(?P<fortnights>{number})(?:fortnights?|fn|f))? # e.g. 10fn
(?:(?P<weeks>{number})(?:weeks?|w))? # e.g. 10w
(?:(?P<days>{number})(?:days?|d))? # e.g. 14d
(?:(?P<hours>{number})(?:hours?|h))? # e.g. 12h
(?:(?P<minutes>{number})(?:minutes?|m))? # e.g. 10m
(?:(?P<seconds>{number})(?:seconds?|s))? # e.g. 15s """, re.VERBOSE)
def parse_prefixed(s):
match = re.match(number, s)
@ -45,6 +46,8 @@ def parse_short_timedelta(text):
match = short_timedelta_regex.fullmatch(text)
if match is None or not match.group(0): raise ValueError("parse failed")
data = { k: parse_prefixed(v) if v else 0 for k, v in match.groupdict().items() }
data["weeks"] += data["fortnights"] * 2
del data["fortnights"]
return datetime.datetime.now(tz=datetime.timezone.utc) + relativedelta(**data)
cal = parsedatetime.Calendar()