diff --git a/main.py b/main.py index 6e69ae8..5c62097 100644 --- a/main.py +++ b/main.py @@ -1,120 +1,37 @@ -import os, discord, logging, re, json +import os, discord, json +from datetime import datetime from dotenv import load_dotenv -from discord import option -from typing import Union from discord.ext import tasks -from datetime import datetime, timedelta - -logging.basicConfig(level=logging.INFO, format='[%(levelname)s] %(asctime)s: %(name)s: %(message)s') -LOGGER = logging.getLogger('botmafieux') -LOGGER.setLevel(logging.DEBUG) -handler = logging.FileHandler(filename='botmafieux.log', encoding='utf-8', mode='w') -handler.setFormatter(logging.Formatter('%(asctime)s:%(levelname)s:%(name)s: %(message)s')) -LOGGER.addHandler(handler) - -bot_intents = discord.Intents.default() -bot_intents.message_content = True -bot_intents.members = True -bot_intents.presences = True -bot_intents.guilds = True -bot_intents.reactions = True - -bot = discord.Bot(command_prefix='$', intents=bot_intents) - -DATA_DIR = 'data/' - -def get_guild_data(guild_id:int): - """Reads and returns the dictionary of guild data from the guild's JSON data file.""" - 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): - """Saves the dictionary of guild data to the guild's JSON data file.""" - 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) - +from modules.setup.bot import bot +from modules.setup.logger import LOGGER +from modules.setup.data import DATA_DIR, get_guild_data @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.""" - command_name = 'birthday_set' - guild_id = ctx.guild.id - guild_data = get_guild_data(guild_id) - author = ctx.author - if re.match(r'\d{2}\/\d{2}',date): # Check if date is in DAY/MONTH format - if for_user: - if author.guild_permissions.administrator: # Check if author is an administrator in case they want to set the birthday for another user - LOGGER.debug(f'{author.name} used /{command_name} {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'Birthday for {for_user.name} has been set to {date}.',ephemeral=True) - else: - LOGGER.debug(f'{author.name} tried to use /{command_name} {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_user".',ephemeral=True) +async def birthday_anouncements_task(): + LOGGER.debug('birthday_anouncements_task started.') + for guild in bot.guilds: + guild_data = get_guild_data(guild.id) + if 'birthday_announcements_channel' in guild_data.keys(): + channel = guild.get_channel(guild_data['birthday_announcements_channel']) + if 'birthday' in guild_data.keys(): + for user_id in guild_data['birthday'].keys(): + user = bot.get_user(int(user_id)) + if user: + if guild_data['birthday'][user_id]['announcements']: + date = guild_data['birthday'][user_id]['date'] + today = datetime.today().strftime('%d/%m') + if date == today: + await channel.send(f'Joyeux anniversaire {user.mention}!') else: - # Set birthday for author - LOGGER.debug(f'{author.name} used /{command_name} {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) - - # The status defines the human readable status of the announcements to be displayed in the response. - 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('The date must is DAY/MONTH format.',ephemeral=True) + LOGGER.debug(f'No birthday_announcements_channel set for guild {guild.name} ({guild.id}).') + LOGGER.debug('birthday_anouncements_task ended.') -@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.""" - command_name = '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: # Check if author is an administrator in case they want to set the birthday for another user - LOGGER.debug(f'{author.name} used /{command_name} {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} tried to use /{command_name} {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_user".',ephemeral=True) - else: - LOGGER.debug(f'{author.name} used /{command_name} {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.slash_command(name='help',description='Displays this message.') +async def help(ctx:discord.ApplicationContext): + """Displays this message.""" + command_name = 'help' + LOGGER.debug(f'{ctx.author.name} used /{command_name}.') + await ctx.send_response('test',ephemeral=True) @bot.event async def on_guild_join(guild:discord.Guild): @@ -145,7 +62,7 @@ async def on_ready(): LOGGER.info(f'Creating data file for guild {guild_id}') with open(path, 'x') as f: json.dump({}, f, indent=2) - test_loop.start() + birthday_anouncements_task.start() if __name__ == '__main__': load_dotenv()