Lumi/services/birthday_service.py

import datetime

import pytz

from db import database


class Birthday:
    def __init__(self, user_id, guild_id):
        self.user_id = user_id
        self.guild_id = guild_id

    def set(self, birthday):
        query = """
                INSERT INTO birthdays (user_id, guild_id, birthday)
                VALUES (%s, %s, %s)
                ON DUPLICATE KEY UPDATE birthday = VALUES(birthday);
                """

        database.execute_query(query, (self.user_id, self.guild_id, birthday))

    def delete(self):
        query = """
                DELETE FROM birthdays
                WHERE user_id = %s AND guild_id = %s;
                """

        database.execute_query(query, (self.user_id, self.guild_id))

    @staticmethod
    def get_birthdays_today():
        query = """
                SELECT user_id, guild_id
                FROM birthdays
                WHERE DATE_FORMAT(birthday, '%m-%d') = %s
                """

        tz = pytz.timezone("US/Eastern")
        today = datetime.datetime.now(tz).strftime("%m-%d")

        return database.select_query(query, (today,))

    @staticmethod
    def get_upcoming_birthdays(guild_id):
        query = """
                SELECT user_id, DATE_FORMAT(birthday, '%m-%d') AS upcoming_birthday 
                FROM birthdays 
                WHERE guild_id = %s 
                ORDER BY (DAYOFYEAR(birthday) - DAYOFYEAR(now()) + 366) % 366;
                """

        data = database.select_query(query, (guild_id,))

        return [(row[0], row[1]) for row in data]