from datetime import timedelta, datetime import json from redbot.core import commands from discord import Message, ButtonStyle, Interaction, ui from ..database import connect, create_guild_table, mysql_log class ImportModerationView(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...") file = await self.ctx.message.attachments[0].read() data: [dict] = sorted(json.loads(file), key=lambda x: x['moderation_id']) user_mod_types = [ 'NOTE', 'WARN', 'MUTE', 'UNMUTE', 'KICK', 'BAN', 'UNBAN' ] channel_mod_types = [ 'SLOWMODE', 'LOCKDOWN' ] failed_cases = [] for case in data: if case['moderation_id'] == 0: continue if 'target_type' not in case or not case['target_type']: if case['moderation_type'] in user_mod_types: case['target_type'] = 'USER' elif case['moderation_type'] in channel_mod_types: case['target_type'] = 'CHANNEL' if 'role_id' not in case or not case['role_id']: case['role_id'] = 0 if 'changes' not in case or not case['changes']: case['changes'] = [] if 'metadata' not in case or not case['metadata']: case['metadata'] = {} if case['duration'] != 'NULL': duration_t = datetime.strptime(case['duration'], "%H:%M:%S") duration = timedelta( hours=duration_t.hours, minutes=duration_t.minutes, seconds=duration_t.seconds ) else: duration = 'NULL' await mysql_log( self.ctx.guild.id, case['moderator_id'], case['moderation_type'], case['target_type'], case['target_id'], case['role_id'], duration, case['reason'], timestamp=case['timestamp'], resolved=case['resolved'], resolved_by=case['resolved_by'], resolved_reason=case['resolve_reason'], changes=case['changes'], metadata=case['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)