bitbot-3.11-fork/modules/birthday.py

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, "example": "1995-09-15"})
class Module(ModuleManager.BaseModule):
    @utils.hook("received.command.birthday")
    def birthday(self, event):
        """
        :help: Get your, or someone else's, birthday
        :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")