Firepup650/bitbot-3.11-fork
2
0
Fork 0
Code Issues Pull requests Activity Actions

Add !upsertbadge, to update or add a badge with a date (badges.py)

Browse source
This commit is contained in:
jesopo 2018-12-22 09:34:13 +00:00
parent 6c38f2f637
commit 13cf49ac5d
1 changed files with 38 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

53
modules/badges.py
View file

@ -14,6 +14,20 @@ class Module(ModuleManager.BaseModule):
def _parse_datetime(self, dt: str): def _parse_datetime(self, dt: str):
return datetime.datetime.strptime(dt, DATETIME_FORMAT) return datetime.datetime.strptime(dt, DATETIME_FORMAT)
def _parse_date(self, dt: str):
if value.lower() == "today":
return self._now()
else:
match = RE_HUMAN_FORMAT.match(value)
if not match:
raise utils.EventError("Invalid date format, please use %s" %
HUMAN_FORMAT_HELP)
return datetime.datetime(
year=int(match.group(1)),
month=int(match.group(2)),
day=int(match.group(3))
)
def _date_str(self, dt: datetime.datetime): def _date_str(self, dt: datetime.datetime):
return datetime.datetime.strftime(dt, DATE_FORMAT) return datetime.datetime.strftime(dt, DATE_FORMAT)
@ -146,20 +160,29 @@ class Module(ModuleManager.BaseModule):
if not found_badge: if not found_badge:
raise utils.EventError("You have no '%s' badge" % badge) raise utils.EventError("You have no '%s' badge" % badge)
value = event["args_split"][-1] dt = self._parse_date(event["args_split"][-1])
if value.lower() == "today":
value = self._now()
else:
match = RE_HUMAN_FORMAT.match(value)
if not match:
raise utils.EventError("Invalid date format, please use %s" %
HUMAN_FORMAT_HELP)
value = datetime.datetime(
year=int(match.group(1)),
month=int(match.group(2)),
day=int(match.group(3))
)
badges[found_badge] = self._format_datetime(value) badges[found_badge] = self._format_datetime(dt)
self._set_badges(event["user"], badges) self._set_badges(event["user"], badges)
event["stdout"].write("Updated '%s' badge" % found_badge) event["stdout"].write("Updated '%s' badge to %s" % (
found_badge, self._date_str(dt)))
@utils.hook("received.command.upsertbadge", min_args=2)
def upsert_badge(self, event):
"""
:help: Add or update a badge
:usage: <badge> today|<yyyy-mm-dd>
"""
badge == " ".join(event["args_split"][:-1])
badge_lower = badge.lower()
badges = self._get_badges(event["user"])
found_badge = self._find_badge(badges, badge)
dt = self._parse_date(event["args_split"][-1])
badges[found_badge or badge] = self._format_datetime(dt)
self._set_badges(event["user"], badges)
add_or_update = "Added" if not found_badge else "Updated"
event["stdout"].write("%s '%s' badge to %s" % (
add_or_update, badge, self._date_str(dt)))