Added birthday_set and birthday_announcements commands and added json data saving

.gitignore

@@ -158,3 +158,5 @@ cython_debug/
 #  and can be added to the global gitignore or merged into this file.  For a more nuclear
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
 #.idea/
+
+data/

main.py

@@ -1,5 +1,7 @@
-import os, discord, logging
+import os, discord, logging, re, json
 from dotenv import load_dotenv
+from discord import option
+from typing import Union
 from discord.ext import tasks
 from datetime import datetime, timedelta
 
@@ -19,20 +21,113 @@ bot_intents.reactions = True
 
 bot = discord.Bot(command_prefix='$', intents=bot_intents)
 
-@bot.slash_command(description="Test")
+DATA_DIR = 'data/'
+
+def get_guild_data(guild_id:int):
+    path = os.path.join(DATA_DIR, f'{guild_id}.json')
+    with open(path, 'r') as f:
+        return json.load(f)
+
+def save_guild_data(guild_id:int, data):
+    path = os.path.join(DATA_DIR, f'{guild_id}.json')
+    with open(path, 'w') as f:
+        json.dump(data, f, indent=2)
+
+@bot.slash_command()
 async def test(ctx:discord.ApplicationContext):
+    """Test description"""
     user = ctx.author
     LOGGER.debug(f'{user.name} used /test.')
     await ctx.send_response('test',ephemeral=True)
 
+
 @tasks.loop(time=datetime.time(datetime.strptime('00:00','%H:%M')))
 async def test_loop():
     LOGGER.debug('test_loop')
-    
+
+@bot.slash_command(description='Sets birthday date. Must be in DAY/MONTH format.')
+@option(name='date',description='Birthday date.',required=True)
+@option(name='for_user',description='The user to set the birthday date for.',required=False,type=discord.Member)
+async def birthday_set(ctx:discord.ApplicationContext, date:str, for_user:discord.Member):
+    """Sets birthday date. Must be in DAY/MONTH format."""
+    guild_id = ctx.guild.id
+    guild_data = get_guild_data(guild_id)
+    author = ctx.author
+    if re.match(r'\d{2}\/\d{2}',date):
+        if for_user:
+            if author.guild_permissions.administrator:
+                LOGGER.debug(f'{author.name} used /set_birthday {date} for {for_user.name}.')
+                if 'birthday' not in guild_data.keys():
+                    guild_data['birthday'] = dict()
+                if for_user.id not in guild_data['birthday'].keys():
+                    guild_data['birthday'][str(for_user.id)] = dict()
+                guild_data['birthday'][str(for_user.id)]['date'] = date
+                if 'announcements' not in guild_data['birthday'][str(for_user.id)].keys():
+                    guild_data['birthday'][str(for_user.id)]['announcements'] = True
+                save_guild_data(guild_id, guild_data)
+                await ctx.send_response(f'{author.name} used /set_birthday {date} for {for_user.name}.',ephemeral=True)
+            else:
+                LOGGER.debug(f'{author.name} used /set_birthday {date} for {for_user.name} but is not an administrator.')
+                await ctx.send_response('You must be an administrator to run the command with "for".',ephemeral=True)
+        else:
+            LOGGER.debug(f'{author.name} used /set_birthday {date}.')
+            if 'birthday' not in guild_data.keys():
+                guild_data['birthday'] = dict()
+            if author.id not in guild_data['birthday'].keys():
+                guild_data['birthday'][str(author.id)] = dict()
+            guild_data['birthday'][str(author.id)]['date'] = date
+            if 'announcements' not in guild_data['birthday'][str(author.id)].keys():
+                guild_data['birthday'][str(author.id)]['announcements'] = True
+            save_guild_data(guild_id, guild_data)
+            announcements_status = 'are enabled' if guild_data['birthday'][str(author.id)]['announcements'] else 'aren\'t enabled'
+            await ctx.send_response(f'Your birthday has been set to {date} and announcements {announcements_status}.',ephemeral=True)
+    else:
+        LOGGER.debug(f'{author.name} used /set_birthday {date} but the format is not correct.')
+        await ctx.send_response(f'The date must is DAY/MONTH format.',ephemeral=True)
+
+@bot.slash_command(description='Enable or disable birthday announcements.')
+@option(name='enable',description='Enable birthday announcements.',required=True)
+@option(name='for_user',description='The user to enable or disable birthday announcements for.',required=False)
+async def birthday_announcements(ctx:discord.ApplicationContext, enable:bool, for_user:discord.Member):
+    """Enable or disable birthday announcements."""
+    guild_id = ctx.guild.id
+    guild_data = get_guild_data(guild_id) or dict()
+    author = ctx.author
+    if for_user:
+        if author.guild_permissions.administrator:
+            LOGGER.debug(f'{author.name} used /enable_birthday {enable} for {for_user.name}.')
+            guild_data['birthday'][str(for_user.id)]['announcements'] = enable
+            save_guild_data(guild_id, guild_data)
+            await ctx.send_response(f'Announcements for {for_user.name} are set to {enable}',ephemeral=True)
+        else:
+            LOGGER.debug(f'{author.name} used /enable_birthday {enable} for {for_user.name} but is not an administrator.')
+            await ctx.send_response('You must be an administrator to run the command with "for".',ephemeral=True)
+    else:
+        LOGGER.debug(f'{author.name} used /enable_birthday {enable}.')
+        guild_data['birthday'][str(author.id)]['announcements'] = enable
+        save_guild_data(guild_id, guild_data)
+        await ctx.send_response(f'Your birthday announcements have been set to {enable}',ephemeral=True)
+
+@bot.event
+async def on_guild_remove(guild:discord.Guild):
+    LOGGER.info(f'Bot left guild {guild.id}.')
+    path = os.path.join(DATA_DIR, f'{guild.id}.json')
+    if os.path.exists(path):
+        LOGGER.info(f'Deleting data file for guild {guild.id}.')
+        os.remove(path)
+
 @bot.event
 async def on_ready():
     LOGGER.info(f'{bot.user.name} has connected to Discord!')
     LOGGER.debug(f'Guilds: {bot.guilds}')
+    # Creates guild data files if they don't exist
+    for guild in bot.guilds:
+        guild_id = guild.id
+        path = os.path.join(DATA_DIR, f'{guild_id}.json')
+        if not os.path.exists(path):
+            LOGGER.info(f'Creating data file for guild {guild_id}')
+            with open(path, 'x') as f:
+                json.dump({}, f, indent=2)
     test_loop.start()
 
 if __name__ == '__main__':