From a11978848d6f2e9d728a0bbe339d8e2ce18c6b3b Mon Sep 17 00:00:00 2001 From: SeaswimmerTheFsh Date: Mon, 18 Dec 2023 16:17:08 -0500 Subject: [PATCH] misc(moderation): moved importing logic to a seperate file --- moderation/importers/galacticbot.py | 142 ++++++++++++++++++++++++++++ moderation/moderation.py | 140 +-------------------------- 2 files changed, 144 insertions(+), 138 deletions(-) create mode 100644 moderation/importers/galacticbot.py diff --git a/moderation/importers/galacticbot.py b/moderation/importers/galacticbot.py new file mode 100644 index 0000000..172d5d5 --- /dev/null +++ b/moderation/importers/galacticbot.py @@ -0,0 +1,142 @@ +from datetime import timedelta +import json +from redbot.core import commands +from discord import Message, ButtonStyle, Interaction, ui +from ..database import connect, create_guild_table, mysql_log + +class ImportGalacticBotView(ui.View): + def __init__(self, timeout, ctx, message): + super().__init__() + self.ctx: commands.Context = ctx + self.message: Message = message + + @ui.button(label="Yes", style=ButtonStyle.success) + async def import_button_y(self, interaction: Interaction, button: ui.Button): # pylint: disable=unused-argument + await self.message.delete() + await interaction.response.send_message("Deleting original table...", ephemeral=True) + + database = await connect() + cursor = database.cursor() + + query = f"DROP TABLE IF EXISTS moderation_{self.ctx.guild.id};" + cursor.execute(query) + + cursor.close() + database.commit() + + await interaction.edit_original_response(content="Creating new table...") + + await create_guild_table(self.ctx.guild) + + await interaction.edit_original_response(content="Importing moderations...") + + accepted_types = [ + 'NOTE', + 'WARN', + 'MUTE', + 'UNMUTE', + 'KICK', + 'SOFTBAN', + 'BAN', + 'UNBAN', + 'SLOWMODE', + 'LOCKDOWN' + ] + + file = await self.ctx.message.attachments[0].read() + data = sorted(json.loads(file), key=lambda x: x['case']) + + failed_cases = [] + + for case in data: + if case['type'] not in accepted_types: + continue + + timestamp = round(case['timestamp'] / 1000) + + try: + if case['duration'] is not None and float(case['duration']) != 0: + duration = timedelta(seconds=round(float(case['duration']) / 1000)) + else: + duration = 'NULL' + except OverflowError: + failed_cases.append(case['case']) + continue + + metadata = { + 'imported_from': 'GalacticBot' + } + + if case['type'] == 'SLOWMODE': + metadata['seconds'] = case['data']['seconds'] + + if case['resolved']: + resolved = 1 + resolved_by = None + resolved_reason = None + resolved_timestamp = None + if case['changes']: + for change in case['changes']: + if change['type'] == 'RESOLVE': + resolved_by = change['staff'] + resolved_reason = change['reason'] + resolved_timestamp = round(change['timestamp'] / 1000) + break + if resolved_by is None: + resolved_by = '?' + if resolved_reason is None: + resolved_reason = 'Could not get resolve reason during moderation import.' + if resolved_timestamp is None: + resolved_timestamp = timestamp + changes = [ + { + 'type': "ORIGINAL", + 'reason': case['reason'], + 'user_id': case['executor'], + 'timestamp': timestamp + }, + { + 'type': "RESOLVE", + 'reason': resolved_reason, + 'user_id': resolved_by, + 'timestamp': resolved_timestamp + } + ] + else: + resolved = 0 + resolved_by = 'NULL' + resolved_reason = 'NULL' + changes = [] + + if case['reason'] and case['reason'] != "N/A": + reason = case['reason'] + else: + reason = "NULL" + + await mysql_log( + self.ctx.guild.id, + case['executor'], + case['type'], + case['targetType'], + case['target'], + 0, + duration, + reason, + timestamp=timestamp, + resolved=resolved, + resolved_by=resolved_by, + resolved_reason=resolved_reason, + changes=changes, + metadata=metadata, + database=database + ) + + await interaction.edit_original_response(content="Import complete.") + if failed_cases: + await interaction.edit_original_response(content=f"Import complete.\n*Failed to import the following cases:*\n```{failed_cases}```") + + @ui.button(label="No", style=ButtonStyle.danger) + async def import_button_n(self, interaction: Interaction, button: ui.Button): # pylint: disable=unused-argument + await self.message.edit("Import cancelled.", view=None) + await self.message.delete(10) + await self.ctx.message.delete(10) diff --git a/moderation/moderation.py b/moderation/moderation.py index 239d9b3..a3620d8 100644 --- a/moderation/moderation.py +++ b/moderation/moderation.py @@ -24,6 +24,7 @@ from .database import connect, create_guild_table, fetch_case, mysql_log from .embed_factory import embed_factory from .logger import logger from .utils import check_conf, check_permissions, check_moddable, fetch_channel_dict, fetch_user_dict, generate_dict, log +from .importers.galacticbot import ImportGalacticBotView class Moderation(commands.Cog): """Custom moderation cog. @@ -1344,147 +1345,10 @@ class Moderation(commands.Cog): """Import moderations from GalacticBot.""" if ctx.message.attachments and ctx.message.attachments[0].content_type == 'application/json; charset=utf-8': message = await ctx.send("Are you sure you want to import GalacticBot moderations?\n**This will overwrite any moderations that already exist in this guild's moderation table.**\n*The import process will block the rest of your bot until it is complete.*") - await message.edit(view=self.GalacticBotImportButtons(60, ctx, message)) + await message.edit(view=ImportGalacticBotView(60, ctx, message)) else: await ctx.send("Please provide a valid GalacticBot moderation export file.") - class GalacticBotImportButtons(discord.ui.View): - def __init__(self, timeout, ctx, message): - super().__init__() - self.ctx: commands.Context = ctx - self.message: discord.Message = message - - @discord.ui.button(label="Yes", style=discord.ButtonStyle.success) - async def import_button_y(self, interaction: discord.Interaction, button: discord.ui.Button): # pylint: disable=unused-argument - await self.message.delete() - await interaction.response.send_message("Deleting original table...", ephemeral=True) - - database = await connect() - cursor = database.cursor() - - query = f"DROP TABLE IF EXISTS moderation_{self.ctx.guild.id};" - cursor.execute(query) - - cursor.close() - database.commit() - - await interaction.edit_original_response(content="Creating new table...") - - await create_guild_table(self.ctx.guild) - - await interaction.edit_original_response(content="Importing moderations...") - - accepted_types = [ - 'NOTE', - 'WARN', - 'MUTE', - 'UNMUTE', - 'KICK', - 'SOFTBAN', - 'BAN', - 'UNBAN', - 'SLOWMODE', - 'LOCKDOWN' - ] - - file = await self.ctx.message.attachments[0].read() - data = sorted(json.loads(file), key=lambda x: x['case']) - - failed_cases = [] - - for case in data: - if case['type'] not in accepted_types: - continue - - timestamp = round(case['timestamp'] / 1000) - - try: - if case['duration'] is not None and float(case['duration']) != 0: - duration = timedelta(seconds=round(float(case['duration']) / 1000)) - else: - duration = 'NULL' - except OverflowError: - failed_cases.append(case['case']) - continue - - metadata = { - 'imported_from': 'GalacticBot' - } - - if case['type'] == 'SLOWMODE': - metadata['seconds'] = case['data']['seconds'] - - if case['resolved']: - resolved = 1 - resolved_by = None - resolved_reason = None - resolved_timestamp = None - if case['changes']: - for change in case['changes']: - if change['type'] == 'RESOLVE': - resolved_by = change['staff'] - resolved_reason = change['reason'] - resolved_timestamp = round(change['timestamp'] / 1000) - break - if resolved_by is None: - resolved_by = '?' - if resolved_reason is None: - resolved_reason = 'Could not get resolve reason during moderation import.' - if resolved_timestamp is None: - resolved_timestamp = timestamp - changes = [ - { - 'type': "ORIGINAL", - 'reason': case['reason'], - 'user_id': case['executor'], - 'timestamp': timestamp - }, - { - 'type': "RESOLVE", - 'reason': resolved_reason, - 'user_id': resolved_by, - 'timestamp': resolved_timestamp - } - ] - else: - resolved = 0 - resolved_by = 'NULL' - resolved_reason = 'NULL' - changes = [] - - if case['reason'] and case['reason'] != "N/A": - reason = case['reason'] - else: - reason = "NULL" - - await mysql_log( - self.ctx.guild.id, - case['executor'], - case['type'], - case['targetType'], - case['target'], - 0, - duration, - reason, - timestamp=timestamp, - resolved=resolved, - resolved_by=resolved_by, - resolved_reason=resolved_reason, - changes=changes, - metadata=metadata, - database=database - ) - - await interaction.edit_original_response(content="Import complete.") - if failed_cases: - await interaction.edit_original_response(content=f"Import complete.\n*Failed to import the following cases:*\n```{failed_cases}```") - - @discord.ui.button(label="No", style=discord.ButtonStyle.danger) - async def import_button_n(self, interaction: discord.Interaction, button: discord.ui.Button): # pylint: disable=unused-argument - await self.message.edit("Import cancelled.", view=None) - await self.message.delete(10) - await self.ctx.message.delete(10) - @commands.command(aliases=["tdc"]) async def timedeltaconvert(self, ctx: commands.Context, *, duration: str): """This command converts a duration to a [`timedelta`](https://docs.python.org/3/library/datetime.html#datetime.timedelta) Python object.