import datetime from src import ModuleManager, utils DATE_YEAR_FORMAT = "%Y-%m-%d" DATE_FORMAT = "%d-%b" def _parse(s): if s.count("-") == 1: try: return False, datetime.datetime.strptime(s, DATE_FORMAT) except ValueError: return None else: try: return True, datetime.datetime.strptime(s, DATE_YEAR_FORMAT) except ValueError: return None def _format(years, dt): return datetime.datetime.strftime(dt, DATE_YEAR_FORMAT if years else DATE_FORMAT) def _check(s): parsed = _parse(s) if parsed: years, parsed = parsed return _format(years, parsed) return None def _apostrophe(nickname): if nickname[-1].lower() == "s": return "%s'" % nickname return "%s's" % nickname @utils.export("set", {"setting": "birthday", "help": "Set your birthday", "validate": _check}) class Module(ModuleManager.BaseModule): @utils.hook("received.command.birthday") def birthday(self, event): """ :help: Get your, or someone else's, pronouns :usage: [nickname] """ target_user = event["user"] if event["args"]: target_user = event["server"].get_user(event["args_split"][0]) birthday = target_user.get_setting("birthday", None) if not birthday == None: years, birthday_parsed = _parse(birthday) birthday_parsed = birthday_parsed.date() now = datetime.datetime.utcnow().date() next_birthday = datetime.date(year=now.year, month=birthday_parsed.month, day=birthday_parsed.day) if next_birthday < now: next_birthday = next_birthday.replace(year=next_birthday.year+1) days = (next_birthday-now).days age = next_birthday.year-birthday_parsed.year if days > 0: if years: event["stdout"].write("%s is %d in %d days" % ( target_user.nickname, age, days)) else: event["stdout"].write("%s birthday is in %d days" % ( _apostrophe(target_user.nickname), days)) else: if years: event["stdout"].write("%s is %d today! 🎉" % ( target_user.nickname, age)) else: event["stdout"].write("%s birthday is today! 🎉" % _apostrophe(target_user.nickname)) else: event["stderr"].write("No birthday set for %s" % target_user.nickname) @utils.hook("received.command.birthdays") def birthdays(self, event): birthday_settings = event["server"].get_all_user_settings("birthday") birthdays = {} today = datetime.datetime.utcnow().date() for nickname, birthday in birthday_settings: years, birthday_parsed = _parse(birthday) birthday_parsed = birthday_parsed.date() if birthday_parsed.replace(year=today.year) == today: birthdays[nickname] = [years, today.year-birthday_parsed.year] if birthdays: birthdays_str = [] for nickname, (years, age) in birthdays.items(): nickname = event["server"].get_user(nickname).nickname if years: birthdays_str.append("%s (%d)" % (nickname, age)) else: birthdays_str.append(nickname) event["stdout"].write("Birthdays today: %s" % ", ".join(birthdays_str)) else: event["stdout"].write("There are no birthdays today")