Changed data storage to a distant database
This commit is contained in:
@@ -1,8 +1,10 @@
|
||||
import discord, re
|
||||
from discord import option
|
||||
from discord.ext import commands
|
||||
from datetime import datetime
|
||||
from ..setup.logger import LOGGER
|
||||
from ..setup.data import get_guild_data, save_guild_data, is_feature_enabled
|
||||
from ..setup import db
|
||||
|
||||
class BirthdayCog(commands.Cog):
|
||||
def __init__(self, bot):
|
||||
@@ -10,96 +12,106 @@ class BirthdayCog(commands.Cog):
|
||||
|
||||
@commands.slash_command(description='Sets birthday date. Must be in DAY/MONTH format.')
|
||||
@option(name='date',description='Birthday date.',required=True)
|
||||
@option(name='announcements',description='Enable or disable birthday announcements.',required=True,type=bool)
|
||||
@option(name='message',description='The message to send when it is the user\'s birthday.',required=False)
|
||||
@option(name='for_user',description='The user to set the birthday date for.',required=False,type=discord.Member)
|
||||
async def birthday_set(self,ctx:discord.ApplicationContext, date:str, for_user:discord.Member):
|
||||
async def birthday_set(self,ctx:discord.ApplicationContext, date:str,announcements:bool,message: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 is_feature_enabled('birthday',data=guild_data):
|
||||
if re.match(r'\d{2}\/\d{2}',date): # Check if date is in DAY/MONTH format
|
||||
feature_data = guild_data['features']['birthday']
|
||||
if 'birthdays' not in feature_data.keys():
|
||||
feature_data['birthdays'] = dict()
|
||||
if str(for_user.id) not in feature_data['birthdays'].keys():
|
||||
feature_data[str(for_user.id)] = dict()
|
||||
birthday_settings = db.select('birthday_settings',f'guild_id = {guild_id}')
|
||||
if birthday_settings:
|
||||
if birthday_settings['is_enabled']:
|
||||
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}.')
|
||||
feature_data['birthdays'][str(for_user.id)]['date'] = date
|
||||
feature_data['birthdays'][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)
|
||||
if author.guild_permissions.administrator:
|
||||
if re.match(r'^\d{1,2}\/\d{1,2}$', date):
|
||||
user_data = db.select('guild_user_birthday',f'guild_id = {guild_id} AND user_id = {for_user.id}')
|
||||
if user_data:
|
||||
# Update user data
|
||||
if message:
|
||||
LOGGER.debug(f'{author.name} used /{command_name} {date} {announcements} {message} for {for_user.name}.')
|
||||
db.update('guild_user_birthday',f'guild_id = {guild_id} AND user_id = {for_user.id}',birthday=date,is_enabled=announcements,birthday_message=message)
|
||||
else:
|
||||
LOGGER.debug(f'{author.name} used /{command_name} {date} {announcements} for {for_user.name}.')
|
||||
db.update('guild_user_birthday',f'guild_id = {guild_id} AND user_id = {for_user.id}',birthday=date,is_enabled=announcements)
|
||||
await ctx.send_response(f'{for_user.name}\'s birthday has been set to {date}.',ephemeral=True)
|
||||
else:
|
||||
# Create user data
|
||||
if message:
|
||||
LOGGER.debug(f'{author.name} used /{command_name} {date} {announcements} {message} for {for_user.name}.')
|
||||
db.insert('guild_user_birthday',guild_id=guild_id,user_id=for_user.id,birthday=date,is_enabled=announcements,birthday_message=message)
|
||||
else:
|
||||
LOGGER.debug(f'{author.name} used /{command_name} {date} {announcements} for {for_user.name}.')
|
||||
db.insert('guild_user_birthday',guild_id=guild_id,user_id=for_user.id,birthday=date,is_enabled=announcements)
|
||||
await ctx.send_response(f'{for_user.name}\'s birthday has been set to {date}.',ephemeral=True)
|
||||
else:
|
||||
LOGGER.debug(f'{author.name} tried to use /{command_name} {date} {announcements} {message} for {for_user.name} but the date format is invalid.')
|
||||
await ctx.send_response('The date format must be DAY/MONTH.',ephemeral=True)
|
||||
else:
|
||||
LOGGER.debug(f'{author.name} tried to use /{command_name} {date} for {for_user.name} but is not an administrator.')
|
||||
LOGGER.debug(f'{author.name} tried to use /{command_name} {date} {announcements} {message} 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:
|
||||
# Set birthday for author
|
||||
LOGGER.debug(f'{author.name} used /{command_name} {date}.')
|
||||
feature_data['birthdays'][str(author.id)]['date'] = date
|
||||
feature_data['birthdays'][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 = 'will be' if feature_data['birthdays'][str(author.id)]['announcements'] else 'will not be'
|
||||
await ctx.send_response(f'Your birthday has been set to {date} and {announcements_status} announced.',ephemeral=True)
|
||||
if re.match(r'^\d{1,2}\/\d{1,2}$', date):
|
||||
user_data = db.select('guild_user_birthday',f'guild_id = {guild_id} AND user_id = {author.id}')
|
||||
if user_data:
|
||||
# Update user data
|
||||
if message:
|
||||
LOGGER.debug(f'{author.name} used /{command_name} {date} {announcements} {message}.')
|
||||
db.update('guild_user_birthday',f'guild_id = {guild_id} AND user_id = {author.id}',birthday=date,is_enabled=announcements,birthday_message=message)
|
||||
else:
|
||||
LOGGER.debug(f'{author.name} used /{command_name} {date} {announcements}.')
|
||||
db.update('guild_user_birthday',f'guild_id = {guild_id} AND user_id = {author.id}',birthday=date,is_enabled=announcements)
|
||||
await ctx.send_response(f'Your birthday has been set to {date}.',ephemeral=True)
|
||||
else:
|
||||
# Create user data
|
||||
if message:
|
||||
LOGGER.debug(f'{author.name} used /{command_name} {date} {announcements} {message}.')
|
||||
db.insert('guild_user_birthday',guild_id=guild_id,user_id=author.id,birthday=date,is_enabled=announcements,birthday_message=message)
|
||||
else:
|
||||
LOGGER.debug(f'{author.name} used /{command_name} {date} {announcements}.')
|
||||
db.insert('guild_user_birthday',guild_id=guild_id,user_id=author.id,birthday=date,is_enabled=announcements)
|
||||
await ctx.send_response(f'Your birthday has been set to {date}.',ephemeral=True)
|
||||
else:
|
||||
LOGGER.debug(f'{author.name} tried to use /{command_name} {date} {announcements} {message} but the date format is invalid.')
|
||||
await ctx.send_response('The date format must be DAY/MONTH.',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'{author.name} tried to use /{command_name} {date} {announcements} {message} but the birthday feature is not enabled.')
|
||||
await ctx.send_response('The birthday feature is not enabled.',ephemeral=True)
|
||||
else:
|
||||
LOGGER.debug(f'{author.name} tried to use /set_birthday {date} but the birthday feature is not enabled.')
|
||||
LOGGER.debug(f'{author.name} tried to use /{command_name} {date} {announcements} {message} but the birthday feature is not enabled.')
|
||||
await ctx.send_response('The birthday feature is not enabled.',ephemeral=True)
|
||||
|
||||
@commands.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,type=discord.Member)
|
||||
async def birthday_announcements(self,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 is_feature_enabled('birthday',data=guild_data):
|
||||
feature_data = guild_data['features']['birthday']
|
||||
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}.')
|
||||
feature_data[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}.')
|
||||
feature_data['birthdays'][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)
|
||||
else:
|
||||
LOGGER.debug(f'{author.name} tried to use /{command_name} {enable} but the birthday feature is not enabled.')
|
||||
await ctx.send_response('The birthday feature is not enabled.',ephemeral=True)
|
||||
|
||||
@commands.slash_command(description='Sets the channel to send birthday announcements to.')
|
||||
@commands.slash_command(description='Configurate birthday announcements.')
|
||||
@option(name='channel',description='The channel to send birthday announcements to.',required=True,type=discord.TextChannel)
|
||||
async def birthday_announcements_channel(self,ctx:discord.ApplicationContext, channel:discord.TextChannel):
|
||||
"""Sets the channel to send birthday announcements to."""
|
||||
command_name = 'birthday_announcements_channel'
|
||||
@option(name='enable',description='Enable or disable birthday announcements.',required=True,type=bool)
|
||||
async def birthday_config(self,ctx:discord.ApplicationContext, channel:discord.TextChannel,enable:bool):
|
||||
"""Configurate birthday announcements."""
|
||||
command_name = 'birthday_config'
|
||||
guild_id = ctx.guild.id
|
||||
guild_data = get_guild_data(guild_id) or dict()
|
||||
author = ctx.author
|
||||
if is_feature_enabled('birthday',data=guild_data):
|
||||
feature_data = guild_data['features']['birthday']
|
||||
if author.guild_permissions.administrator:
|
||||
LOGGER.debug(f'{author.name} used /{command_name} {channel.name}.')
|
||||
feature_data['birthday_announcements_channel'] = channel.id
|
||||
save_guild_data(guild_id, guild_data)
|
||||
await ctx.send_response(f'Birthday announcements will be sent to {channel.mention}.',ephemeral=True)
|
||||
birthday_settings = db.select('birthday_settings',f'guild_id = {guild_id}')
|
||||
if birthday_settings:
|
||||
# Update guild birthday settings
|
||||
if birthday_settings['is_enabled']:
|
||||
if author.guild_permissions.administrator:
|
||||
LOGGER.debug(f'{author.name} used /{command_name} {channel.name} {enable}.')
|
||||
db.update('birthday_settings',f'guild_id = {guild_id}',channel_id=channel.id,is_enabled=enable)
|
||||
await ctx.send_response(f'Birthday announcements have been {"enabled" if enable else "disabled"} in {channel.mention}.',ephemeral=True)
|
||||
else:
|
||||
LOGGER.debug(f'{author.name} tried to use /{command_name} {channel.name} {enable} but is not an administrator.')
|
||||
await ctx.send_response('You must be an administrator to run this command.',ephemeral=True)
|
||||
else:
|
||||
LOGGER.debug(f'{author.name} tried to use /{command_name} {channel.mention} but is not an administrator.')
|
||||
await ctx.send_response('You must be an administrator to run this command.',ephemeral=True)
|
||||
LOGGER.debug(f'{author.name} tried to use /{command_name} {channel.name} {enable} but the birthday feature is not enabled.')
|
||||
await ctx.send_response('The birthday feature is not enabled.',ephemeral=True)
|
||||
else:
|
||||
LOGGER.debug(f'{author.name} tried to use /{command_name} {channel.mention} but the birthday feature is not enabled.')
|
||||
await ctx.send_response('The birthday feature is not enabled.',ephemeral=True)
|
||||
# Create guild birthday settings
|
||||
if author.guild_permissions.administrator:
|
||||
LOGGER.debug(f'{author.name} used /{command_name} {channel.name} {enable}.')
|
||||
db.insert('birthday_settings',guild_id=guild_id,channel_id=channel.id,is_enabled=enable)
|
||||
await ctx.send_response(f'Birthday announcements have been {"enabled" if enable else "disabled"} in {channel.mention}.',ephemeral=True)
|
||||
else:
|
||||
LOGGER.debug(f'{author.name} tried to use /{command_name} {channel.name} {enable} but is not an administrator.')
|
||||
await ctx.send_response('You must be an administrator to run this command.',ephemeral=True)
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user