First Commit

implementing/sleepers/birthday.py

@@ -0,0 +1,129 @@
+import discord
+import datetime
+
+from discord.ext import commands, tasks
+from utils.sql_commands import DatabaseManager
+from dotenv import dotenv_values
+
+
+
+
+def check(author):
+    def inner_check(message):
+        return message.author == author
+
+    return inner_check
+
+
+class Birthday(commands.Cog):
+    def __init__(self, client):
+        self.client = client        
+        self.db = DatabaseManager("")
+        self.birthday_checks = {}
+
+    async def cog_load(self):
+        self.check_birthdays.start()
+
+    async def cog_unload(self):
+        self.check_birthdays.cancel()
+
+    @tasks.loop(seconds=60*60*24) # daily
+    async def check_birthdays(self):
+        today = datetime.date.today()
+        for guild in self.client.guilds:
+            guild_id = guild.id
+            for member in guild.members:
+                user_id = member.id
+                user_data = self.db.fetch_one("SELECT * FROM users WHERE ID = %s", (user_id,))
+                if user_data is None or user_data.get("birthday") is None:
+                    continue
+
+                birthday = datetime.datetime.strptime(user_data.get("birthday"), "%Y-%m-%d").date()
+                if birthday.month == today.month and birthday.day == today.day:
+                    # send birthday message
+                    try:
+                        channel = guild.text_channels[0]
+                        await channel.send(f"Happy birthday, {member.mention}! 🎉")
+                    except discord.Forbidden:
+                        continue
+                    except discord.HTTPException:
+                        continue
+                    # give birthday reward
+                    reward = self.db.fetch_one("SELECT * FROM rewards WHERE type = 'birthday'")
+                    if reward is not None:
+                        try:
+                            await self.client.get_cog("Economy").add_points(user_id, guild_id, reward.get("amount"), "birthday reward")
+                        except Exception as e:
+                            print(f"Error giving birthday reward: {e}")
+                    # give birthday star role
+                    role = discord.utils.get(guild.roles, name="Birthday Star")
+                    if role is not None:
+                        try:
+                            await member.add_roles(role)
+                        except discord.Forbidden:
+                            continue
+                        except discord.HTTPException:
+                            continue
+                    # log birthday reward
+                    try:
+                        self.db.execute_query("INSERT INTO logs (guild_id, user_id, type, message) VALUES (%s, %s, 'birthday_reward', 'claimed birthday reward')", (guild_id, user_id))
+                    except Exception as e:
+                        print(f"Error logging birthday reward: {e}")
+
+    @check_birthdays.before_loop
+    async def before_check_birthdays(self):
+        await self.client.wait_until_ready()
+
+    @commands.command(name="setbirthday", aliases=["sb"])
+    async def setbirthday(self, ctx, month: int, day: int):
+        if month < 1 or month > 12 or day < 1 or day > 31:
+            return await ctx.send("Invalid date")
+
+        user_id = ctx.author.id
+        user_data = self.db.fetch_one("SELECT * FROM users WHERE ID = %s", (user_id,))
+
+        if user_data is not None and user_data.get("birthday") is not None:
+            return await ctx.send("You've already set your birthday")
+
+        self.db.execute_query("UPDATE users SET birthday = %s WHERE ID = %s", (f"{datetime.date.today().year}-{month}-{day}", user_id))
+        await ctx.send("Your birthday has been set")
+
+    @commands.command(name="claimbirthday", aliases=["cb"])
+    async def claimbirthday(self, ctx):
+        user_id = ctx.author.id
+        user_data = self.db.fetch_one("SELECT * FROM users WHERE ID = %s", (user_id,))
+
+        if user_data is None or user_data.get("birthday") is None:
+            return await ctx.send("You haven't set your birthday")
+
+        today = datetime.date.today()
+        birthday = datetime.datetime.strptime(user_data.get("birthday"), "%Y-%m-%d").date()
+        if birthday.month != today.month or birthday.day != today.day:
+            return await ctx.send("It's not your birthday")
+
+        reward = self.db.fetch_one("SELECT * FROM rewards WHERE type = 'birthday'")
+        if reward is None:
+            return await ctx.send("No birthday reward set")
+
+        await self.client.get_cog("Economy").add_points(user_id, ctx.guild.id, reward.get("amount"), "birthday reward")
+        await ctx.send("You've claimed your birthday reward")
+
+    @commands.command(name="listbirthdays", aliases=["lb"])
+    async def listbirthdays(self, ctx):
+        today = datetime.date.today()
+        birthdays = []
+        for guild in self.client.guilds:
+            for member in guild.members:
+                user_id = member.id
+                user_data = self.db.fetch_one("SELECT * FROM users WHERE ID = %s", (user_id,))
+                if user_data is None or user_data.get("birthday") is None:
+                    continue
+
+                birthday = datetime.datetime.strptime(user_data.get("birthday"), "%Y-%m-%d").date()
+                if birthday.month == today.month and birthday.day == today.day:
+                    birthdays.append(member.mention)
+
+        if not birthdays:
+            await ctx.send("No birthdays today")
+        else:
+            await ctx.send(f"Happy birthday to {', '.join(birthdays)}! 🎉")