diff options
Diffstat (limited to 'modules')
| -rw-r--r-- | modules/badges.py | 13 |
1 files changed, 9 insertions, 4 deletions
diff --git a/modules/badges.py b/modules/badges.py index b289768c..d0a5c3a4 100644 --- a/modules/badges.py +++ b/modules/badges.py @@ -4,6 +4,7 @@ from src import ModuleManager, utils RE_HUMAN_FORMAT = re.compile(r"(\d\d\d\d)-(\d?\d)-(\d?\d)") HUMAN_FORMAT_HELP = "year-month-day (e.g. 2018-12-29)" DATETIME_FORMAT = "%Y-%m-%dT%H:%M:%SZ" +DATE_FORMAT = "%Y-%m-%d" class Module(ModuleManager.BaseModule): def _now(self): @@ -13,6 +14,9 @@ class Module(ModuleManager.BaseModule): def _parse_datetime(self, dt: str): return datetime.datetime.strptime(dt, DATETIME_FORMAT) + def _date_str(self, dt: datetime.datetime): + return datetime.datetime.strftime(dt, DATE_FORMAT) + def _round_up_day(self, dt: datetime.datetime): return dt.date()+datetime.timedelta(days=1) def _days_since(self, now: datetime.date, dt: datetime.datetime): @@ -38,10 +42,11 @@ class Module(ModuleManager.BaseModule): break if found_badge: - days_since = self._days_since(now, - self._parse_datetime(badges[found_badge])) - event["stdout"].write("(%s) %s on day %s" % ( - event["user"].nickname, found_badge, days_since)) + dt = self._parse_datetime(badges[found_badge]) + days_since = self._days_since(now, dt) + event["stdout"].write("(%s) %s on day %s (%s)" % ( + event["user"].nickname, found_badge, days_since, + self._date_str(dt))) else: event["stderr"].write("You have no '%s' badge" % badge) |