misc(moderation): moved importing logic to a seperate file
This commit is contained in:
parent
25cbc33b6d
commit
a11978848d
GPG key ID:
1EBC234EEDA901AE
2 changed files with 144 additions and 138 deletions
142
moderation/importers/galacticbot.py
Normal file
142
moderation/importers/galacticbot.py
Normal file
|
|
@ -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)
|
||||||
|
|
@ -24,6 +24,7 @@ from .database import connect, create_guild_table, fetch_case, mysql_log
|
||||||
from .embed_factory import embed_factory
|
from .embed_factory import embed_factory
|
||||||
from .logger import logger
|
from .logger import logger
|
||||||
from .utils import check_conf, check_permissions, check_moddable, fetch_channel_dict, fetch_user_dict, generate_dict, log
|
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):
|
class Moderation(commands.Cog):
|
||||||
"""Custom moderation cog.
|
"""Custom moderation cog.
|
||||||
|
|
@ -1344,147 +1345,10 @@ class Moderation(commands.Cog):
|
||||||
"""Import moderations from GalacticBot."""
|
"""Import moderations from GalacticBot."""
|
||||||
if ctx.message.attachments and ctx.message.attachments[0].content_type == 'application/json; charset=utf-8':
|
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.*")
|
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:
|
else:
|
||||||
await ctx.send("Please provide a valid GalacticBot moderation export file.")
|
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"])
|
@commands.command(aliases=["tdc"])
|
||||||
async def timedeltaconvert(self, ctx: commands.Context, *, duration: str):
|
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.
|
"""This command converts a duration to a [`timedelta`](https://docs.python.org/3/library/datetime.html#datetime.timedelta) Python object.
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue