2023-12-14 18:55:35 -05:00
# _____ _
# / ____| (_)
# | (___ ___ __ _ _____ ___ _ __ ___ _ __ ___ ___ _ __
# \___ \ / _ \/ _` / __\ \ /\ / / | '_ ` _ \| '_ ` _ \ / _ \ '__|
# ____) | __/ (_| \__ \\ V V /| | | | | | | | | | | | __/ |
# |_____/ \___|\__,_|___/ \_/\_/ |_|_| |_| |_|_| |_| |_|\___|_|
2023-12-14 18:35:25 -05:00
import json
import time
import os
from datetime import datetime , timedelta , timezone
import discord
import humanize
import mysql . connector
from discord . ext import tasks
from pytimeparse2 import disable_dateutil , parse
from redbot . core import app_commands , checks , Config , commands , data_manager
from redbot . core . app_commands import Choice
2023-12-17 02:16:44 -05:00
from . database import connect , create_guild_table , fetch_case , mysql_log
from . embed_factory import embed_factory
2023-12-17 12:54:41 -05:00
from . utils import check_conf , check_permissions , check_moddable , fetch_channel_dict , fetch_user_dict , generate_dict , log
2023-12-17 02:16:44 -05:00
from . logger import logger
2023-12-14 18:35:25 -05:00
class Moderation ( commands . Cog ) :
""" Custom moderation cog.
Developed by SeaswimmerTheFsh . """
2023-12-14 19:04:27 -05:00
async def red_delete_data_for_user ( self , * , requester , user_id : int ) :
if requester == " discord_deleted_user " :
await self . config . user_from_id ( user_id ) . clear ( )
2023-12-17 02:16:44 -05:00
database = await connect ( )
2023-12-14 19:04:27 -05:00
cursor = database . cursor ( )
cursor . execute ( " SHOW TABLES; " )
tables = [ table [ 0 ] for table in cursor . fetchall ( ) ]
condition = " target_id = %s OR moderator_id = %s ; "
for table in tables :
delete_query = f " DELETE FROM { table [ 0 ] } WHERE { condition } "
cursor . execute ( delete_query , ( user_id , user_id ) )
database . commit ( )
cursor . close ( )
database . close ( )
if requester == " owner " :
await self . config . user_from_id ( user_id ) . clear ( )
if requester == " user " :
await self . config . user_from_id ( user_id ) . clear ( )
if requester == " user_strict " :
await self . config . user_from_id ( user_id ) . clear ( )
else :
2023-12-17 02:16:44 -05:00
logger . warning ( " Invalid requester passed to red_delete_data_for_user: %s " , requester )
2023-12-14 19:04:27 -05:00
2023-12-14 18:35:25 -05:00
def __init__ ( self , bot ) :
self . bot = bot
self . config = Config . get_conf ( self , identifier = 481923957134912 )
self . config . register_global (
mysql_address = " " ,
mysql_database = " " ,
mysql_username = " " ,
mysql_password = " "
)
self . config . register_guild (
2023-12-15 13:54:58 -05:00
use_discord_permissions = True ,
2023-12-14 18:35:25 -05:00
ignore_other_bots = True ,
dm_users = True ,
log_channel = " " ,
2023-12-14 21:19:14 -05:00
immune_roles = [ ] ,
2023-12-14 18:35:25 -05:00
history_ephemeral = False ,
history_inline = False ,
history_pagesize = 5 ,
2023-12-14 21:19:14 -05:00
history_inline_pagesize = 6 ,
blacklist_roles = [ ]
2023-12-14 18:35:25 -05:00
)
self . config . register_user (
history_ephemeral = None ,
history_inline = None ,
history_pagesize = None ,
history_inline_pagesize = None
)
disable_dateutil ( )
self . handle_expiry . start ( ) # pylint: disable=no-member
async def cog_load ( self ) :
""" This method prepares the database schema for all of the guilds the bot is currently in. """
2023-12-17 02:16:44 -05:00
conf = await check_conf ( [
2023-12-14 18:35:25 -05:00
' mysql_address ' ,
' mysql_database ' ,
' mysql_username ' ,
' mysql_password '
] )
if conf :
2023-12-17 02:16:44 -05:00
logger . error ( " Failed to create tables, due to MySQL connection configuration being unset. " )
2023-12-14 18:35:25 -05:00
return
guilds : list [ discord . Guild ] = self . bot . guilds
try :
for guild in guilds :
if not await self . bot . cog_disabled_in_guild ( self , guild ) :
2023-12-17 02:36:18 -05:00
await create_guild_table ( guild )
2023-12-14 18:35:25 -05:00
except ConnectionRefusedError :
return
async def cog_unload ( self ) :
self . handle_expiry . cancel ( ) # pylint: disable=no-member
@commands.Cog.listener ( ' on_guild_join ' )
async def db_generate_guild_join ( self , guild : discord . Guild ) :
""" This method prepares the database schema whenever the bot joins a guild. """
if not await self . bot . cog_disabled_in_guild ( self , guild ) :
2023-12-17 02:16:44 -05:00
conf = await check_conf ( [
2023-12-14 18:35:25 -05:00
' mysql_address ' ,
' mysql_database ' ,
' mysql_username ' ,
' mysql_password '
] )
if conf :
2023-12-17 02:16:44 -05:00
logger . error ( " Failed to create a table for %s , due to MySQL connection configuration being unset. " , guild . id )
2023-12-14 18:35:25 -05:00
return
try :
2023-12-17 02:16:44 -05:00
await create_guild_table ( guild )
2023-12-14 18:35:25 -05:00
except ConnectionRefusedError :
return
@commands.Cog.listener ( ' on_audit_log_entry_create ' )
async def autologger ( self , entry : discord . AuditLogEntry ) :
""" This method automatically logs moderations done by users manually ( " right clicks " ). """
if not await self . bot . cog_disabled_in_guild ( self , entry . guild ) :
2023-12-15 17:54:38 -05:00
if await self . config . guild ( entry . guild ) . ignore_other_bots ( ) is True :
2023-12-14 18:35:25 -05:00
if entry . user . bot or entry . target . bot :
return
else :
if entry . user . id == self . bot . user . id :
return
duration = " NULL "
if entry . reason :
reason = entry . reason + " (This action was performed without the bot.) "
else :
reason = " This action was performed without the bot. "
if entry . action == discord . AuditLogAction . kick :
moderation_type = ' KICK '
elif entry . action == discord . AuditLogAction . ban :
moderation_type = ' BAN '
elif entry . action == discord . AuditLogAction . unban :
moderation_type = ' UNBAN '
elif entry . action == discord . AuditLogAction . member_update :
if entry . after . timed_out_until is not None :
timed_out_until_aware = entry . after . timed_out_until . replace ( tzinfo = timezone . utc )
duration_datetime = timed_out_until_aware - datetime . now ( tz = timezone . utc )
minutes = round ( duration_datetime . total_seconds ( ) / 60 )
duration = timedelta ( minutes = minutes )
moderation_type = ' MUTE '
else :
moderation_type = ' UNMUTE '
else :
return
2023-12-17 02:16:44 -05:00
await mysql_log ( entry . guild . id , entry . user . id , moderation_type , entry . target . id , 0 , duration , reason )
2023-12-14 18:35:25 -05:00
#######################################################################################################################
### COMMANDS
#######################################################################################################################
@app_commands.command ( name = " note " )
async def note ( self , interaction : discord . Interaction , target : discord . User , reason : str , silent : bool = None ) :
""" Add a note to a user.
Parameters
- - - - - - - - - - -
target : discord . User
Who are you noting ?
reason : str
Why are you noting this user ?
silent : bool
Should the user be messaged ? """
2023-12-17 02:16:44 -05:00
if not await check_moddable ( target , interaction , [ ' moderate_members ' ] ) :
2023-12-15 13:54:58 -05:00
return
2023-12-14 18:35:25 -05:00
await interaction . response . send_message ( content = f " { target . mention } has recieved a note! \n **Reason** - ` { reason } ` " )
if silent is None :
silent = not await self . config . guild ( interaction . guild ) . dm_users ( )
if silent is False :
try :
2023-12-17 03:35:59 -05:00
embed = await embed_factory ( ' message ' , await self . bot . get_embed_color ( None ) , guild = interaction . guild , reason = reason , moderation_type = ' note ' , response = await interaction . original_response ( ) )
2023-12-14 18:35:25 -05:00
await target . send ( embed = embed )
except discord . errors . HTTPException :
pass
2023-12-17 12:56:46 -05:00
moderation_id = await mysql_log ( interaction . guild . id , interaction . user . id , ' NOTE ' , ' USER ' , target . id , 0 , ' NULL ' , reason )
2023-12-16 21:28:09 -05:00
await interaction . edit_original_response ( content = f " { target . mention } has received a note! (Case `# { moderation_id : , } `) \n **Reason** - ` { reason } ` " )
2023-12-17 03:06:09 -05:00
await log ( interaction , moderation_id )
2023-12-14 18:35:25 -05:00
@app_commands.command ( name = " warn " )
async def warn ( self , interaction : discord . Interaction , target : discord . Member , reason : str , silent : bool = None ) :
""" Warn a user.
Parameters
- - - - - - - - - - -
target : discord . Member
Who are you warning ?
reason : str
Why are you warning this user ?
silent : bool
Should the user be messaged ? """
2023-12-17 02:16:44 -05:00
if not await check_moddable ( target , interaction , [ ' moderate_members ' ] ) :
2023-12-15 13:54:58 -05:00
return
2023-12-14 18:35:25 -05:00
await interaction . response . send_message ( content = f " { target . mention } has been warned! \n **Reason** - ` { reason } ` " )
if silent is None :
silent = not await self . config . guild ( interaction . guild ) . dm_users ( )
if silent is False :
try :
2023-12-17 03:35:59 -05:00
embed = await embed_factory ( ' message ' , await self . bot . get_embed_color ( None ) , guild = interaction . guild , reason = reason , moderation_type = ' warned ' , response = await interaction . original_response ( ) )
2023-12-14 18:35:25 -05:00
await target . send ( embed = embed )
except discord . errors . HTTPException :
pass
2023-12-17 12:56:46 -05:00
moderation_id = await mysql_log ( interaction . guild . id , interaction . user . id , ' WARN ' , ' USER ' , target . id , 0 , ' NULL ' , reason )
2023-12-16 21:28:09 -05:00
await interaction . edit_original_response ( content = f " { target . mention } has been warned! (Case `# { moderation_id : , } `) \n **Reason** - ` { reason } ` " )
2023-12-17 03:06:09 -05:00
await log ( interaction , moderation_id )
2023-12-14 18:35:25 -05:00
2023-12-17 02:36:18 -05:00
async def blacklist_autocomplete ( self , interaction : discord . Interaction , current : str , ) - > list [ app_commands . Choice [ str ] ] : # pylint: disable=unused-argument
2023-12-16 21:02:54 -05:00
""" Autocompletes a blacklist role. """
blacklist_roles = await self . config . guild ( interaction . guild ) . blacklist_roles ( )
2023-12-15 17:38:53 -05:00
2023-12-16 21:02:54 -05:00
if blacklist_roles :
return [ app_commands . Choice ( name = role . name , value = role . id ) for role in interaction . guild . roles if role . id in blacklist_roles ]
2023-12-17 02:36:18 -05:00
return [ ]
2023-12-16 21:02:54 -05:00
@app_commands.command ( name = " blacklist " )
2023-12-16 21:19:40 -05:00
@app_commands.autocomplete ( role = blacklist_autocomplete )
2023-12-16 21:02:54 -05:00
async def blacklist ( self , interaction : discord . Interaction , target : discord . Member , role : str , reason : str , silent : bool = None ) :
""" Add a blacklist role to a user.
Parameters
- - - - - - - - - - -
target : discord . Member
Who are you blacklisting ?
role : str
What blacklist type are you applying to the target ?
reason : str
Why are you blacklisting this user ?
silent : bool
Should the user be messaged ? """
blacklist_roles = await self . config . guild ( interaction . guild ) . blacklist_roles ( )
if not blacklist_roles :
2023-12-17 02:36:18 -05:00
await interaction . response . send_message ( content = " There are no blacklist types set for this server! " , ephemeral = True )
2023-12-16 21:02:54 -05:00
return
matching_role = None
for role_dict in blacklist_roles :
if role_dict [ ' id ' ] == role :
matching_role = role_dict
break
if not matching_role :
2023-12-17 02:36:18 -05:00
await interaction . response . send_message ( content = " Please provide a valid blacklist type! " , ephemeral = True )
2023-12-16 21:02:54 -05:00
return
2023-12-17 02:16:44 -05:00
if not await check_moddable ( target , interaction , [ ' moderate_members ' , ' manage_roles ' ] ) :
2023-12-16 21:02:54 -05:00
return
if role in [ role . id for role in target . roles ] :
await interaction . response . send_message ( content = f " { target . mention } already has the blacklist role! " , ephemeral = True )
return
2023-12-17 02:36:18 -05:00
if silent is None :
silent = not await self . config . guild ( interaction . guild ) . dm_users ( )
if silent is False :
try :
2023-12-17 03:35:59 -05:00
embed = await embed_factory ( ' message ' , await self . bot . get_embed_color ( None ) , guild = interaction . guild , reason = reason , moderation_type = ' blacklisted ' , response = await interaction . original_response ( ) , duration = matching_role [ ' duration ' ] )
2023-12-17 02:36:18 -05:00
await target . send ( embed = embed )
except discord . errors . HTTPException :
pass
2023-12-16 21:02:54 -05:00
role_obj = interaction . guild . get_role ( role )
await target . add_roles ( role , reason = f " Blacklisted by { interaction . user . id } for { humanize . precisedelta ( matching_role [ ' duration ' ] ) } for: { reason } " )
await interaction . response . send_message ( content = f " { target . mention } has been blacklisted with the { role_obj . name } role for { humanize . precisedelta ( matching_role [ ' duration ' ] ) } ! \n **Reason** - ` { reason } ` " )
2023-12-17 12:56:46 -05:00
moderation_id = await mysql_log ( interaction . guild . id , interaction . user . id , ' BLACKLIST ' , ' USER ' , target . id , role , matching_role [ ' duration ' ] , reason )
2023-12-16 21:28:09 -05:00
await interaction . edit_original_response ( content = f " { target . mention } has been blacklisted with the { role_obj . name } role for { humanize . precisedelta ( matching_role [ ' duration ' ] ) } ! (Case `# { moderation_id : , } `) \n **Reason** - ` { reason } ` " )
2023-12-17 03:06:09 -05:00
await log ( interaction , moderation_id )
2023-12-15 17:38:53 -05:00
2023-12-14 18:35:25 -05:00
@app_commands.command ( name = " mute " )
async def mute ( self , interaction : discord . Interaction , target : discord . Member , duration : str , reason : str , silent : bool = None ) :
""" Mute a user.
Parameters
- - - - - - - - - - -
target : discord . Member
Who are you unbanning ?
duration : str
How long are you muting this user for ?
reason : str
Why are you unbanning this user ?
silent : bool
Should the user be messaged ? """
2023-12-17 02:16:44 -05:00
if not await check_moddable ( target , interaction , [ ' moderate_members ' ] ) :
2023-12-14 18:35:25 -05:00
return
2023-12-15 13:54:58 -05:00
2023-12-14 18:35:25 -05:00
if target . is_timed_out ( ) is True :
await interaction . response . send_message ( f " { target . mention } is already muted! " , allowed_mentions = discord . AllowedMentions ( users = False ) , ephemeral = True )
return
try :
parsed_time = parse ( sval = duration , as_timedelta = True , raise_exception = True )
except ValueError :
await interaction . response . send_message ( " Please provide a valid duration! " , ephemeral = True )
return
if parsed_time . total_seconds ( ) / 1000 > 2419200000 :
await interaction . response . send_message ( " Please provide a duration that is less than 28 days. " )
return
await target . timeout ( parsed_time , reason = f " Muted by { interaction . user . id } for: { reason } " )
await interaction . response . send_message ( content = f " { target . mention } has been muted for { humanize . precisedelta ( parsed_time ) } ! \n **Reason** - ` { reason } ` " )
if silent is None :
silent = not await self . config . guild ( interaction . guild ) . dm_users ( )
if silent is False :
try :
2023-12-17 03:35:59 -05:00
embed = await embed_factory ( ' message ' , await self . bot . get_embed_color ( None ) , guild = interaction . guild , reason = reason , moderation_type = ' muted ' , response = await interaction . original_response ( ) , duration = parsed_time )
2023-12-14 18:35:25 -05:00
await target . send ( embed = embed )
except discord . errors . HTTPException :
pass
2023-12-17 12:56:46 -05:00
moderation_id = await mysql_log ( interaction . guild . id , interaction . user . id , ' MUTE ' , ' USER ' , target . id , 0 , parsed_time , reason )
2023-12-16 21:28:09 -05:00
await interaction . edit_original_response ( content = f " { target . mention } has been muted for { humanize . precisedelta ( parsed_time ) } ! (Case `# { moderation_id : , } `) \n **Reason** - ` { reason } ` " )
2023-12-17 03:06:09 -05:00
await log ( interaction , moderation_id )
2023-12-14 18:35:25 -05:00
@app_commands.command ( name = " unmute " )
async def unmute ( self , interaction : discord . Interaction , target : discord . Member , reason : str = None , silent : bool = None ) :
""" Unmute a user.
Parameters
- - - - - - - - - - -
target : discord . user
Who are you unmuting ?
reason : str
Why are you unmuting this user ?
silent : bool
Should the user be messaged ? """
2023-12-17 02:16:44 -05:00
if not await check_moddable ( target , interaction , [ ' moderate_members ' ] ) :
2023-12-14 18:35:25 -05:00
return
2023-12-15 13:54:58 -05:00
2023-12-14 18:35:25 -05:00
if target . is_timed_out ( ) is False :
await interaction . response . send_message ( f " { target . mention } is not muted! " , allowed_mentions = discord . AllowedMentions ( users = False ) , ephemeral = True )
return
2023-12-15 13:54:58 -05:00
2023-12-14 18:35:25 -05:00
if reason :
await target . timeout ( None , reason = f " Unmuted by { interaction . user . id } for: { reason } " )
else :
await target . timeout ( None , reason = f " Unbanned by { interaction . user . id } " )
reason = " No reason given. "
await interaction . response . send_message ( content = f " { target . mention } has been unmuted! \n **Reason** - ` { reason } ` " )
if silent is None :
silent = not await self . config . guild ( interaction . guild ) . dm_users ( )
if silent is False :
try :
2023-12-17 03:35:59 -05:00
embed = await embed_factory ( ' message ' , await self . bot . get_embed_color ( None ) , guild = interaction . guild , reason = reason , moderation_type = ' unmuted ' , response = await interaction . original_response ( ) )
2023-12-14 18:35:25 -05:00
await target . send ( embed = embed )
except discord . errors . HTTPException :
pass
2023-12-17 12:56:46 -05:00
moderation_id = await mysql_log ( interaction . guild . id , interaction . user . id , ' UNMUTE ' , ' USER ' , target . id , 0 , ' NULL ' , reason )
2023-12-16 21:28:09 -05:00
await interaction . edit_original_response ( content = f " { target . mention } has been unmuted! (Case `# { moderation_id : , } `) \n **Reason** - ` { reason } ` " )
2023-12-17 03:06:09 -05:00
await log ( interaction , moderation_id )
2023-12-14 18:35:25 -05:00
@app_commands.command ( name = " kick " )
async def kick ( self , interaction : discord . Interaction , target : discord . Member , reason : str , silent : bool = None ) :
""" Kick a user.
Parameters
- - - - - - - - - - -
target : discord . user
Who are you kicking ?
reason : str
Why are you kicking this user ?
silent : bool
Should the user be messaged ? """
2023-12-17 02:16:44 -05:00
if not await check_moddable ( target , interaction , [ ' kick_members ' ] ) :
2023-12-14 18:35:25 -05:00
return
await interaction . response . send_message ( content = f " { target . mention } has been kicked! \n **Reason** - ` { reason } ` " )
if silent is None :
silent = not await self . config . guild ( interaction . guild ) . dm_users ( )
if silent is False :
try :
2023-12-17 03:35:59 -05:00
embed = await embed_factory ( ' message ' , await self . bot . get_embed_color ( None ) , guild = interaction . guild , reason = reason , moderation_type = ' kicked ' , response = await interaction . original_response ( ) )
2023-12-14 18:35:25 -05:00
await target . send ( embed = embed )
except discord . errors . HTTPException :
pass
2023-12-16 20:29:54 -05:00
await target . kick ( reason = f " Kicked by { interaction . user . id } for: { reason } " )
2023-12-14 18:35:25 -05:00
2023-12-17 12:56:46 -05:00
moderation_id = await mysql_log ( interaction . guild . id , interaction . user . id , ' KICK ' , ' USER ' , target . id , 0 , ' NULL ' , reason )
2023-12-16 21:28:09 -05:00
await interaction . edit_original_response ( content = f " { target . mention } has been kicked! (Case `# { moderation_id : , } `) \n **Reason** - ` { reason } ` " )
2023-12-17 03:06:09 -05:00
await log ( interaction , moderation_id )
2023-12-14 18:35:25 -05:00
@app_commands.command ( name = " ban " )
@app_commands.choices ( delete_messages = [
Choice ( name = " None " , value = 0 ) ,
Choice ( name = ' 1 Hour ' , value = 3600 ) ,
Choice ( name = ' 12 Hours ' , value = 43200 ) ,
Choice ( name = ' 1 Day ' , value = 86400 ) ,
Choice ( name = ' 3 Days ' , value = 259200 ) ,
Choice ( name = ' 7 Days ' , value = 604800 ) ,
] )
async def ban ( self , interaction : discord . Interaction , target : discord . User , reason : str , duration : str = None , delete_messages : Choice [ int ] = 0 , silent : bool = None ) :
""" Ban a user.
Parameters
- - - - - - - - - - -
target : discord . user
Who are you banning ?
duration : str
How long are you banning this user for ?
reason : str
Why are you banning this user ?
delete_messages : Choices [ int ]
How many days of messages to delete ?
silent : bool
Should the user be messaged ? """
2023-12-17 02:16:44 -05:00
if not await check_moddable ( target , interaction , [ ' ban_members ' ] ) :
2023-12-14 18:35:25 -05:00
return
2023-12-15 13:54:58 -05:00
2023-12-14 18:35:25 -05:00
try :
await interaction . guild . fetch_ban ( target )
await interaction . response . send_message ( content = f " { target . mention } is already banned! " , ephemeral = True )
return
except discord . errors . NotFound :
pass
if duration :
try :
parsed_time = parse ( sval = duration , as_timedelta = True , raise_exception = True )
except ValueError :
await interaction . response . send_message ( " Please provide a valid duration! " , ephemeral = True )
return
await interaction . response . send_message ( content = f " { target . mention } has been banned for { humanize . precisedelta ( parsed_time ) } ! \n **Reason** - ` { reason } ` " )
try :
2023-12-17 03:35:59 -05:00
embed = await embed_factory ( ' message ' , await self . bot . get_embed_color ( None ) , guild = interaction . guild , reason = reason , moderation_type = ' tempbanned ' , response = await interaction . original_response ( ) , duration = parsed_time )
2023-12-14 18:35:25 -05:00
await target . send ( embed = embed )
except discord . errors . HTTPException :
pass
await interaction . guild . ban ( target , reason = f " Tempbanned by { interaction . user . id } for: { reason } (Duration: { parsed_time } ) " , delete_message_seconds = delete_messages )
2023-12-17 02:16:44 -05:00
moderation_id = await mysql_log ( interaction . guild . id , interaction . user . id , ' TEMPBAN ' , target . id , 0 , parsed_time , reason )
2023-12-14 18:35:25 -05:00
await interaction . edit_original_response ( content = f " { target . mention } has been banned for { humanize . precisedelta ( parsed_time ) } ! (Case `# { moderation_id } `) \n **Reason** - ` { reason } ` " )
2023-12-17 03:06:09 -05:00
await log ( interaction , moderation_id )
2023-12-14 18:35:25 -05:00
else :
await interaction . response . send_message ( content = f " { target . mention } has been banned! \n **Reason** - ` { reason } ` " )
if silent is None :
silent = not await self . config . guild ( interaction . guild ) . dm_users ( )
if silent is False :
try :
2023-12-17 03:35:59 -05:00
embed = embed = await embed_factory ( ' message ' , await self . bot . get_embed_color ( None ) , guild = interaction . guild , reason = reason , moderation_type = ' banned ' , response = await interaction . original_response ( ) )
2023-12-14 18:35:25 -05:00
await target . send ( embed = embed )
except discord . errors . HTTPException :
pass
await interaction . guild . ban ( target , reason = f " Banned by { interaction . user . id } for: { reason } " , delete_message_seconds = delete_messages )
2023-12-17 12:56:46 -05:00
moderation_id = await mysql_log ( interaction . guild . id , interaction . user . id , ' BAN ' , ' USER ' , target . id , 0 , ' NULL ' , reason )
2023-12-16 21:28:09 -05:00
await interaction . edit_original_response ( content = f " { target . mention } has been banned! (Case `# { moderation_id : , } `) \n **Reason** - ` { reason } ` " )
2023-12-17 02:16:44 -05:00
await log ( interaction , moderation_id )
2023-12-14 18:35:25 -05:00
@app_commands.command ( name = " unban " )
async def unban ( self , interaction : discord . Interaction , target : discord . User , reason : str = None , silent : bool = None ) :
""" Unban a user.
Parameters
- - - - - - - - - - -
target : discord . user
Who are you unbanning ?
reason : str
Why are you unbanning this user ?
silent : bool
Should the user be messaged ? """
2023-12-17 02:16:44 -05:00
if not await check_moddable ( target , interaction , [ ' ban_members ' ] ) :
2023-12-14 18:35:25 -05:00
return
2023-12-15 13:54:58 -05:00
2023-12-14 18:35:25 -05:00
try :
await interaction . guild . fetch_ban ( target )
except discord . errors . NotFound :
await interaction . response . send_message ( content = f " { target . mention } is not banned! " , ephemeral = True )
return
if reason :
await interaction . guild . unban ( target , reason = f " Unbanned by { interaction . user . id } for: { reason } " )
else :
await interaction . guild . unban ( target , reason = f " Unbanned by { interaction . user . id } " )
reason = " No reason given. "
await interaction . response . send_message ( content = f " { target . mention } has been unbanned! \n **Reason** - ` { reason } ` " )
if silent is None :
silent = not await self . config . guild ( interaction . guild ) . dm_users ( )
if silent is False :
try :
2023-12-17 03:35:59 -05:00
embed = await embed_factory ( ' message ' , await self . bot . get_embed_color ( None ) , guild = interaction . guild , reason = reason , moderation_type = ' unbanned ' , response = await interaction . original_response ( ) )
2023-12-14 18:35:25 -05:00
await target . send ( embed = embed )
except discord . errors . HTTPException :
pass
2023-12-17 12:56:46 -05:00
moderation_id = await mysql_log ( interaction . guild . id , interaction . user . id , ' UNBAN ' , ' USER ' , target . id , 0 , ' NULL ' , reason )
2023-12-16 21:28:09 -05:00
await interaction . edit_original_response ( content = f " { target . mention } has been unbanned! (Case `# { moderation_id : , } `) \n **Reason** - ` { reason } ` " )
2023-12-17 03:06:09 -05:00
await log ( interaction , moderation_id )
2023-12-14 18:35:25 -05:00
@app_commands.command ( name = " history " )
2023-12-15 22:28:42 -05:00
async def history ( self , interaction : discord . Interaction , target : discord . User = None , moderator : discord . User = None , pagesize : app_commands . Range [ int , 1 , 20 ] = None , page : int = 1 , ephemeral : bool = None , inline : bool = None , export : bool = False ) :
2023-12-14 18:35:25 -05:00
""" List previous infractions.
Parameters
- - - - - - - - - - -
target : discord . User
User whose infractions to query , overrides moderator if both are given
moderator : discord . User
Query by moderator
pagesize : app_commands . Range [ int , 1 , 25 ]
Amount of infractions to list per page
page : int
Page to select
ephemeral : bool
Hide the command response
inline : bool
Display infractions in a grid arrangement ( does not look very good )
export : bool
Exports the server ' s entire moderation history to a JSON file " " "
2023-12-15 10:10:22 -05:00
if ephemeral is None :
ephemeral = ( await self . config . user ( interaction . user ) . history_ephemeral ( )
or await self . config . guild ( interaction . guild ) . history_ephemeral ( )
or False )
2023-12-14 18:35:25 -05:00
2023-12-15 10:13:40 -05:00
if inline is None :
inline = ( await self . config . user ( interaction . user ) . history_inline ( )
or await self . config . guild ( interaction . guild ) . history_inline ( )
or False )
2023-12-14 18:35:25 -05:00
if pagesize is None :
if inline is True :
pagesize = ( await self . config . user ( interaction . user ) . history_inline_pagesize ( )
or await self . config . guild ( interaction . guild ) . history_inline_pagesize ( )
or 6 )
else :
pagesize = ( await self . config . user ( interaction . user ) . history_pagesize ( )
or await self . config . guild ( interaction . guild ) . history_pagesize ( )
2023-12-15 10:13:40 -05:00
or 5 )
2023-12-14 18:35:25 -05:00
2023-12-15 10:10:22 -05:00
await interaction . response . defer ( ephemeral = ephemeral )
2023-12-17 02:16:44 -05:00
permissions = check_permissions ( interaction . client . user , [ ' embed_links ' ] , interaction )
2023-12-15 10:10:22 -05:00
if permissions :
await interaction . followup . send ( f " I do not have the ` { permissions } ` permission, required for this action. " , ephemeral = True )
return
2023-12-17 02:16:44 -05:00
database = await connect ( )
2023-12-15 10:10:22 -05:00
cursor = database . cursor ( )
2023-12-14 18:35:25 -05:00
if target :
query = """ SELECT *
FROM moderation_ % s
WHERE target_id = % s
ORDER BY moderation_id DESC ; """
cursor . execute ( query , ( interaction . guild . id , target . id ) )
elif moderator :
query = """ SELECT *
FROM moderation_ % s
WHERE moderator_id = % s
ORDER BY moderation_id DESC ; """
cursor . execute ( query , ( interaction . guild . id , moderator . id ) )
else :
query = """ SELECT *
FROM moderation_ % s
ORDER BY moderation_id DESC ; """
cursor . execute ( query , ( interaction . guild . id , ) )
results = cursor . fetchall ( )
result_dict_list = [ ]
for result in results :
2023-12-17 02:16:44 -05:00
case_dict = generate_dict ( result )
2023-12-14 18:35:25 -05:00
if case_dict [ ' moderation_id ' ] == 0 :
continue
result_dict_list . append ( case_dict )
if export :
try :
filename = str ( data_manager . cog_data_path ( cog_instance = self ) ) + str ( os . sep ) + f " moderation_ { interaction . guild . id } .json "
with open ( filename , " w " , encoding = " utf-8 " ) as f :
json . dump ( result_dict_list , f , indent = 2 )
await interaction . followup . send ( file = discord . File ( filename , f " moderation_ { interaction . guild . id } .json " ) , ephemeral = ephemeral )
os . remove ( filename )
return
except json . JSONDecodeError as e :
await interaction . followup . send ( content = f " An error occured while exporting the moderation history. \n Error: \n ``` { e } ``` " , ephemeral = ephemeral )
return
case_quantity = len ( result_dict_list )
page_quantity = round ( case_quantity / pagesize )
start_index = ( page - 1 ) * pagesize
end_index = page * pagesize
embed = discord . Embed ( color = await self . bot . get_embed_color ( None ) )
embed . set_author ( icon_url = interaction . guild . icon . url , name = ' Infraction History ' )
2023-12-16 21:28:09 -05:00
embed . set_footer ( text = f " Page { page : , } / { page_quantity : , } | { case_quantity : , } Results " )
2023-12-14 18:35:25 -05:00
memory_dict = { }
for case in result_dict_list [ start_index : end_index ] :
if case [ ' target_id ' ] not in memory_dict :
2023-12-17 12:54:41 -05:00
if case [ ' target_type ' ] == ' USER ' :
memory_dict [ str ( case [ ' target_id ' ] ) ] = await fetch_user_dict ( interaction , case [ ' target_id ' ] )
elif case [ ' target_type ' ] == ' CHANNEL ' :
memory_dict [ str ( case [ ' target_id ' ] ) ] = await fetch_channel_dict ( interaction , case [ ' target_id ' ] )
2023-12-14 18:35:25 -05:00
target_user = memory_dict [ str ( case [ ' target_id ' ] ) ]
2023-12-17 12:54:41 -05:00
if case [ ' target_type ' ] == ' USER ' :
target_name = f " ` { target_user [ ' name ' ] } ` " if target_user [ ' discriminator ' ] == " 0 " else f " ` { target_user [ ' name ' ] } # { target_user [ ' discriminator ' ] } ` "
elif case [ ' target_type ' ] == ' CHANNEL ' :
target_name = f " ` { target_user [ ' mention ' ] } ` "
2023-12-14 18:35:25 -05:00
if case [ ' moderator_id ' ] not in memory_dict :
2023-12-17 02:16:44 -05:00
memory_dict [ str ( case [ ' moderator_id ' ] ) ] = await fetch_user_dict ( interaction , case [ ' moderator_id ' ] )
2023-12-14 18:35:25 -05:00
moderator_user = memory_dict [ str ( case [ ' moderator_id ' ] ) ]
2023-12-17 12:54:41 -05:00
moderator_name = f " ` { moderator_user [ ' name ' ] } ` " if moderator_user [ ' discriminator ' ] == " 0 " else f " ` { moderator_user [ ' name ' ] } # { moderator_user [ ' discriminator ' ] } ` "
2023-12-14 18:35:25 -05:00
2023-12-16 21:28:09 -05:00
field_name = f " Case # { case [ ' moderation_id ' ] : , } ( { str . title ( case [ ' moderation_type ' ] ) } ) "
2023-12-17 12:54:41 -05:00
field_value = f " **Target:** { target_name } ( { target_user [ ' id ' ] } ) \n **Moderator:** { moderator_name } ( { moderator_user [ ' id ' ] } ) "
2023-12-14 18:35:25 -05:00
2023-12-15 22:24:29 -05:00
if len ( case [ ' reason ' ] ) > 125 :
field_value + = f " \n **Reason:** ` { str ( case [ ' reason ' ] ) [ : 125 ] } ...` "
2023-12-14 18:35:25 -05:00
else :
field_value + = f " \n **Reason:** ` { str ( case [ ' reason ' ] ) } ` "
if case [ ' duration ' ] != ' NULL ' :
td = timedelta ( * * { unit : int ( val ) for unit , val in zip ( [ " hours " , " minutes " , " seconds " ] , case [ " duration " ] . split ( " : " ) ) } )
duration_embed = f " { humanize . precisedelta ( td ) } | <t: { case [ ' end_timestamp ' ] } :R> " if bool ( case [ ' expired ' ] ) is False else f " { humanize . precisedelta ( td ) } | Expired "
field_value + = f " \n **Duration:** { duration_embed } "
field_value + = f " \n **Timestamp:** <t: { case [ ' timestamp ' ] } > | <t: { case [ ' timestamp ' ] } :R> "
if bool ( case [ ' resolved ' ] ) :
field_value + = " \n **Resolved:** True "
embed . add_field ( name = field_name , value = field_value , inline = inline )
await interaction . followup . send ( embed = embed , ephemeral = ephemeral )
@app_commands.command ( name = " resolve " )
2023-12-14 20:16:25 -05:00
async def resolve ( self , interaction : discord . Interaction , case : int , reason : str = None ) :
2023-12-14 18:35:25 -05:00
""" Resolve a specific case.
Parameters
- - - - - - - - - - -
2023-12-14 20:16:25 -05:00
case : int
2023-12-14 18:35:25 -05:00
Case number of the case you ' re trying to resolve
reason : str
Reason for resolving case """
2023-12-17 02:16:44 -05:00
permissions = check_permissions ( interaction . client . user , [ ' embed_links ' , ' moderate_members ' , ' ban_members ' ] , interaction )
2023-12-14 18:35:25 -05:00
if permissions :
await interaction . response . send_message ( f " I do not have the ` { permissions } ` permission, required for this action. " , ephemeral = True )
return
2023-12-17 02:16:44 -05:00
conf = await check_conf ( [ ' mysql_database ' ] )
2023-12-14 18:35:25 -05:00
if conf :
raise ( LookupError )
2023-12-17 02:16:44 -05:00
database = await connect ( )
2023-12-14 18:35:25 -05:00
cursor = database . cursor ( )
db = await self . config . mysql_database ( )
query_1 = " SELECT * FROM moderation_ %s WHERE moderation_id = %s ; "
2023-12-14 20:16:25 -05:00
cursor . execute ( query_1 , ( interaction . guild . id , case ) )
2023-12-14 18:35:25 -05:00
result_1 = cursor . fetchone ( )
2023-12-14 20:16:25 -05:00
if result_1 is None or case == 0 :
await interaction . response . send_message ( content = f " There is no moderation with a case number of { case } . " , ephemeral = True )
2023-12-14 18:35:25 -05:00
return
query_2 = " SELECT * FROM moderation_ %s WHERE moderation_id = %s AND resolved = 0; "
2023-12-14 20:16:25 -05:00
cursor . execute ( query_2 , ( interaction . guild . id , case ) )
2023-12-14 18:35:25 -05:00
result_2 = cursor . fetchone ( )
if result_2 is None :
2023-12-14 20:16:25 -05:00
await interaction . response . send_message ( content = f " This moderation has already been resolved! \n Use `/case { case } ` for more information. " , ephemeral = True )
2023-12-14 18:35:25 -05:00
return
2023-12-17 02:16:44 -05:00
case_dict = generate_dict ( result_2 )
2023-12-14 18:35:25 -05:00
if reason is None :
reason = " No reason given. "
2023-12-14 20:16:25 -05:00
changes : list = case_dict [ ' changes ' ]
2023-12-14 19:57:43 -05:00
if len ( changes ) > 25 :
await interaction . response . send_message ( content = " Due to limitations with Discord ' s embed system, you cannot edit a case more than 25 times. " , ephemeral = True )
return
if not changes :
changes . append (
{
' type ' : " ORIGINAL " ,
2023-12-14 20:16:25 -05:00
' timestamp ' : case_dict [ ' timestamp ' ] ,
' reason ' : case_dict [ ' reason ' ] ,
' user_id ' : case_dict [ ' moderator_id ' ]
2023-12-14 19:57:43 -05:00
}
)
changes . append (
{
' type ' : " RESOLVE " ,
' timestamp ' : int ( time . time ( ) ) ,
' reason ' : reason ,
' user_id ' : interaction . user . id
}
)
2023-12-14 20:16:25 -05:00
if case_dict [ ' moderation_type ' ] in [ ' UNMUTE ' , ' UNBAN ' ] :
2023-12-14 18:35:25 -05:00
await interaction . response . send_message ( content = " You cannot resolve this type of moderation! " , ephemeral = True )
2023-12-14 20:16:25 -05:00
if case_dict [ ' moderation_type ' ] in [ ' MUTE ' , ' TEMPBAN ' , ' BAN ' ] :
if case_dict [ ' moderation_type ' ] == ' MUTE ' :
2023-12-14 18:35:25 -05:00
try :
2023-12-14 20:16:25 -05:00
member = await interaction . guild . fetch_member ( case_dict [ ' target_id ' ] )
2023-12-14 18:35:25 -05:00
2023-12-16 21:28:09 -05:00
await member . timeout ( None , reason = f " Case # { case : , } resolved by { interaction . user . id } " )
2023-12-14 18:35:25 -05:00
except discord . NotFound :
pass
2023-12-14 20:16:25 -05:00
if case_dict [ ' moderation_type ' ] in [ ' TEMPBAN ' , ' BAN ' ] :
2023-12-14 18:35:25 -05:00
try :
2023-12-14 20:16:25 -05:00
user = await interaction . client . fetch_user ( case_dict [ ' target_id ' ] )
2023-12-14 18:35:25 -05:00
2023-12-14 20:16:25 -05:00
await interaction . guild . unban ( user , reason = f " Case # { case } resolved by { interaction . user . id } " )
2023-12-14 18:35:25 -05:00
except discord . NotFound :
pass
2023-12-15 21:31:00 -05:00
resolve_query = f " UPDATE ` { db } `.`moderation_ { interaction . guild . id } ` SET resolved = 1, changes = %s, resolved_by = %s, resolve_reason = %s WHERE moderation_id = %s "
2023-12-14 18:35:25 -05:00
else :
2023-12-14 19:57:43 -05:00
resolve_query = f " UPDATE ` { db } `.`moderation_ { interaction . guild . id } ` SET resolved = 1, changes = %s, resolved_by = %s, resolve_reason = %s WHERE moderation_id = %s "
2023-12-14 18:35:25 -05:00
2023-12-15 11:22:25 -05:00
cursor . execute ( resolve_query , ( json . dumps ( changes ) , interaction . user . id , reason , case_dict [ ' moderation_id ' ] ) )
2023-12-14 18:35:25 -05:00
database . commit ( )
2023-12-17 03:35:59 -05:00
embed = await embed_factory ( ' case ' , await self . bot . get_embed_color ( None ) , interaction = interaction , case_dict = await fetch_case ( case , interaction . guild . id ) )
2023-12-16 21:28:09 -05:00
await interaction . response . send_message ( content = f " ✅ Moderation # { case : , } resolved! " , embed = embed )
2023-12-17 03:06:09 -05:00
await log ( interaction , case )
2023-12-14 18:35:25 -05:00
cursor . close ( )
database . close ( )
@app_commands.command ( name = " case " )
2023-12-14 20:16:25 -05:00
@app_commands.choices ( export = [
2023-12-15 10:41:38 -05:00
Choice ( name = ' Export as File ' , value = ' file ' ) ,
Choice ( name = ' Export as Codeblock ' , value = ' codeblock ' )
2023-12-14 20:16:25 -05:00
] )
2023-12-15 10:41:38 -05:00
async def case ( self , interaction : discord . Interaction , case : int , ephemeral : bool = None , changes : bool = False , export : Choice [ str ] = None ) :
2023-12-14 18:35:25 -05:00
""" Check the details of a specific case.
Parameters
- - - - - - - - - - -
2023-12-14 20:16:25 -05:00
case : int
2023-12-14 18:35:25 -05:00
What case are you looking up ?
ephemeral : bool
2023-12-14 20:00:03 -05:00
Hide the command response
2023-12-14 20:16:25 -05:00
changes : bool
List the changes made to the case
2023-12-14 20:00:03 -05:00
export : bool
2023-12-15 10:41:38 -05:00
Export the case to a JSON file or codeblock """
2023-12-17 02:16:44 -05:00
permissions = check_permissions ( interaction . client . user , [ ' embed_links ' ] , interaction )
2023-12-14 18:35:25 -05:00
if permissions :
await interaction . response . send_message ( f " I do not have the ` { permissions } ` permission, required for this action. " , ephemeral = True )
return
2023-12-15 10:02:54 -05:00
if ephemeral is None :
ephemeral = ( await self . config . user ( interaction . user ) . history_ephemeral ( )
or await self . config . guild ( interaction . guild ) . history_ephemeral ( )
or False )
2023-12-14 20:16:25 -05:00
if case != 0 :
2023-12-17 02:16:44 -05:00
case_dict = await fetch_case ( case , interaction . guild . id )
2023-12-14 20:16:25 -05:00
if case_dict :
2023-12-14 19:59:22 -05:00
if export :
2023-12-15 10:41:38 -05:00
if export . value == ' file ' or len ( str ( case_dict ) ) > 1800 :
2023-12-14 20:16:25 -05:00
filename = str ( data_manager . cog_data_path ( cog_instance = self ) ) + str ( os . sep ) + f " moderation_ { interaction . guild . id } _case_ { case } .json "
2023-12-14 19:59:22 -05:00
2023-12-14 20:06:57 -05:00
with open ( filename , " w " , encoding = " utf-8 " ) as f :
2023-12-14 20:16:25 -05:00
json . dump ( case_dict , f , indent = 2 )
2023-12-14 19:59:22 -05:00
2023-12-15 10:41:38 -05:00
if export . value == ' codeblock ' :
2023-12-16 21:28:09 -05:00
content = f " Case # { case : , } exported. \n *Case was too large to export as codeblock, so it has been uploaded as a `.json` file.* "
2023-12-14 20:16:25 -05:00
else :
2023-12-16 21:28:09 -05:00
content = f " Case # { case : , } exported. "
2023-12-14 20:16:25 -05:00
await interaction . response . send_message ( content = content , file = discord . File ( filename , f " moderation_ { interaction . guild . id } _case_ { case } .json " ) , ephemeral = ephemeral )
2023-12-14 19:59:22 -05:00
2023-12-14 20:06:57 -05:00
os . remove ( filename )
return
2023-12-14 20:29:48 -05:00
await interaction . response . send_message ( content = f " ```json \n { json . dumps ( case_dict , indent = 2 ) } ``` " , ephemeral = ephemeral )
return
2023-12-14 19:57:43 -05:00
if changes :
2023-12-17 03:35:59 -05:00
embed = await embed_factory ( ' changes ' , await self . bot . get_embed_color ( None ) , interaction = interaction , case_dict = case_dict )
2023-12-14 19:57:43 -05:00
else :
2023-12-17 03:35:59 -05:00
embed = await embed_factory ( ' case ' , await self . bot . get_embed_color ( None ) , interaction = interaction , case_dict = case_dict )
2023-12-14 18:35:25 -05:00
await interaction . response . send_message ( embed = embed , ephemeral = ephemeral )
return
2023-12-14 20:16:25 -05:00
await interaction . response . send_message ( content = f " No case with case number ` { case } ` found. " , ephemeral = True )
2023-12-14 19:38:35 -05:00
@app_commands.command ( name = " edit " )
2023-12-16 16:59:48 -05:00
async def edit ( self , interaction : discord . Interaction , case : int , reason : str , duration : str = None ) :
2023-12-14 19:38:35 -05:00
""" Edit the reason of a specific case.
Parameters
- - - - - - - - - - -
2023-12-14 20:16:25 -05:00
case : int
2023-12-14 19:38:35 -05:00
What case are you editing ?
reason : str
2023-12-16 16:59:48 -05:00
What is the new reason ?
duration : str
What is the new duration ? Does not reapply the moderation if it has already expired . """
2023-12-17 02:16:44 -05:00
permissions = check_permissions ( interaction . client . user , [ ' embed_links ' ] , interaction )
2023-12-14 19:38:35 -05:00
if permissions :
await interaction . response . send_message ( f " I do not have the ` { permissions } ` permission, required for this action. " , ephemeral = True )
return
2023-12-14 20:16:25 -05:00
if case != 0 :
2023-12-16 16:59:48 -05:00
parsed_time = None
2023-12-17 02:16:44 -05:00
case_dict = await fetch_case ( case , interaction . guild . id )
2023-12-14 20:16:25 -05:00
if case_dict :
2023-12-17 02:16:44 -05:00
conf = await check_conf ( [ ' mysql_database ' ] )
2023-12-14 19:38:35 -05:00
if conf :
raise ( LookupError )
2023-12-16 16:59:48 -05:00
if duration :
try :
parsed_time = parse ( sval = duration , as_timedelta = True , raise_exception = True )
except ValueError :
await interaction . response . send_message ( " Please provide a valid duration! " , ephemeral = True )
return
end_timestamp = case_dict [ ' timestamp ' ] + parsed_time . total_seconds ( )
if case_dict [ ' type ' ] == ' MUTE ' :
if ( time . time ( ) - case_dict [ ' timestamp ' ] ) + parsed_time . total_seconds ( ) > 2419200 :
await interaction . response . send_message ( " Please provide a duration that is less than 28 days from the initial moderation. " )
return
try :
member = await interaction . guild . fetch_member ( case_dict [ ' target_id ' ] )
2023-12-16 21:28:09 -05:00
await member . timeout ( parsed_time , reason = f " Case # { case : , } edited by { interaction . user . id } " )
2023-12-16 16:59:48 -05:00
except discord . NotFound :
pass
2023-12-14 20:16:25 -05:00
changes : list = case_dict [ ' changes ' ]
2023-12-14 19:45:27 -05:00
if len ( changes ) > 25 :
await interaction . response . send_message ( content = " Due to limitations with Discord ' s embed system, you cannot edit a case more than 25 times. " , ephemeral = True )
return
2023-12-14 19:38:35 -05:00
if not changes :
2023-12-14 19:41:25 -05:00
changes . append (
{
2023-12-14 19:57:43 -05:00
' type ' : " ORIGINAL " ,
2023-12-14 20:16:25 -05:00
' timestamp ' : case_dict [ ' timestamp ' ] ,
' reason ' : case_dict [ ' reason ' ] ,
2023-12-16 16:59:48 -05:00
' user_id ' : case_dict [ ' moderator_id ' ] ,
' duration ' : case_dict [ ' duration ' ] ,
' end_timestamp ' : case_dict [ ' end_timestamp ' ]
}
)
if parsed_time :
changes . append (
{
' type ' : " EDIT " ,
' timestamp ' : int ( time . time ( ) ) ,
' reason ' : reason ,
' user_id ' : interaction . user . id ,
' duration ' : parsed_time ,
' end_timestamp ' : end_timestamp
}
)
else :
changes . append (
{
' type ' : " EDIT " ,
' timestamp ' : int ( time . time ( ) ) ,
' reason ' : reason ,
' user_id ' : interaction . user . id ,
' duration ' : case_dict [ ' duration ' ] ,
' end_timestamp ' : case_dict [ ' end_timestamp ' ]
2023-12-14 19:41:25 -05:00
}
)
2023-12-14 19:38:35 -05:00
2023-12-17 02:16:44 -05:00
database = await connect ( )
2023-12-14 19:38:35 -05:00
cursor = database . cursor ( )
db = await self . config . mysql_database ( )
2023-12-16 16:59:48 -05:00
if parsed_time :
update_query = f " UPDATE ` { db } `.`moderation_ { interaction . guild . id } ` SET changes = %s, reason = %s, duration = %s, end_timestamp = %s WHERE moderation_id = %s "
cursor . execute ( update_query , ( json . dumps ( changes ) , reason , parsed_time , end_timestamp , case ) )
else :
update_query = f " UPDATE ` { db } `.`moderation_ { interaction . guild . id } ` SET changes = %s, reason = %s WHERE moderation_id = %s "
cursor . execute ( update_query , ( json . dumps ( changes ) , reason , case ) )
2023-12-14 19:38:35 -05:00
database . commit ( )
2023-12-17 02:16:44 -05:00
new_case = await fetch_case ( case , interaction . guild . id )
2023-12-17 03:35:59 -05:00
embed = await embed_factory ( ' case ' , await self . bot . get_embed_color ( None ) , interaction = interaction , case_dict = new_case )
2023-12-14 19:38:35 -05:00
2023-12-16 21:28:09 -05:00
await interaction . response . send_message ( content = f " ✅ Moderation # { case : , } edited! " , embed = embed , ephemeral = True )
2023-12-17 03:06:09 -05:00
await log ( interaction , case )
2023-12-14 19:38:35 -05:00
cursor . close ( )
database . close ( )
return
2023-12-14 20:16:25 -05:00
await interaction . response . send_message ( content = f " No case with case number ` { case } ` found. " , ephemeral = True )
2023-12-14 18:35:25 -05:00
@tasks.loop ( minutes = 1 )
async def handle_expiry ( self ) :
2023-12-17 02:16:44 -05:00
conf = await check_conf ( [ ' mysql_database ' ] )
2023-12-14 18:35:25 -05:00
if conf :
raise ( LookupError )
2023-12-17 02:16:44 -05:00
database = await connect ( )
2023-12-14 18:35:25 -05:00
cursor = database . cursor ( )
db = await self . config . mysql_database ( )
guilds : list [ discord . Guild ] = self . bot . guilds
for guild in guilds :
if not await self . bot . cog_disabled_in_guild ( self , guild ) :
tempban_query = f " SELECT target_id, moderation_id FROM moderation_ { guild . id } WHERE end_timestamp != 0 AND end_timestamp <= %s AND moderation_type = ' TEMPBAN ' AND expired = 0 "
try :
cursor . execute ( tempban_query , ( time . time ( ) , ) )
result = cursor . fetchall ( )
except mysql . connector . errors . ProgrammingError :
continue
target_ids = [ row [ 0 ] for row in result ]
moderation_ids = [ row [ 1 ] for row in result ]
for target_id , moderation_id in zip ( target_ids , moderation_ids ) :
user : discord . User = await self . bot . fetch_user ( target_id )
try :
await guild . unban ( user , reason = f " Automatic unban from case # { moderation_id } " )
2023-12-17 03:35:59 -05:00
embed = await embed_factory ( ' message ' , self . bot . get_embed_color ( None ) , guild , f ' Automatic unban from case # { moderation_id } ' , ' unbanned ' )
2023-12-14 18:35:25 -05:00
try :
await user . send ( embed = embed )
except discord . errors . HTTPException :
pass
except [ discord . errors . NotFound , discord . errors . Forbidden , discord . errors . HTTPException ] as e :
print ( f " Failed to unban { user . name } # { user . discriminator } ( { user . id } ) from { guild . name } ( { guild . id } ) \n { e } " )
expiry_query = f " UPDATE ` { db } `.`moderation_ { guild . id } ` SET expired = 1 WHERE (end_timestamp != 0 AND end_timestamp <= %s AND expired = 0 AND moderation_type != ' BLACKLIST ' ) OR (expired = 0 AND resolved = 1 AND moderation_type != ' BLACKLIST ' ) "
cursor . execute ( expiry_query , ( time . time ( ) , ) )
blacklist_query = f " SELECT target_id, moderation_id, role_id FROM moderation_ { guild . id } WHERE end_timestamp != 0 AND end_timestamp <= %s AND moderation_type = ' BLACKLIST ' AND expired = 0 "
try :
cursor . execute ( blacklist_query , ( time . time ( ) , ) )
result = cursor . fetchall ( )
except mysql . connector . errors . ProgrammingError :
continue
target_ids = [ row [ 0 ] for row in result ]
moderation_ids = [ row [ 1 ] for row in result ]
role_ids = [ row [ 2 ] for row in result ]
for target_id , moderation_id , role_id in zip ( target_ids , moderation_ids , role_ids ) :
try :
member : discord . Member = await guild . fetch_member ( target_id )
role : discord . Role = guild . get_role ( role_id )
if role is None :
raise discord . errors . NotFound
except [ discord . errors . NotFound , discord . errors . Forbidden , discord . errors . HTTPException ] :
continue
database . commit ( )
cursor . close ( )
database . close ( )
2023-12-14 21:19:14 -05:00
#######################################################################################################################
### CONFIGURATION COMMANDS
#######################################################################################################################
2023-12-15 10:06:35 -05:00
@commands.group ( autohelp = True , aliases = [ ' modset ' , ' moderationsettings ' ] )
2023-12-14 18:35:25 -05:00
async def moderationset ( self , ctx : commands . Context ) :
""" Manage moderation commands. """
2023-12-15 10:36:52 -05:00
@moderationset.command ( name = ' list ' , aliases = [ ' view ' , ' show ' ] )
2023-12-14 19:11:22 -05:00
async def moderationset_list ( self , ctx : commands . Context ) :
""" List all moderation settings. """
2023-12-14 19:20:01 -05:00
if ctx . guild :
guild_settings = await self . config . guild ( ctx . guild ) . all ( )
2023-12-14 19:11:22 -05:00
2023-12-14 19:20:01 -05:00
guild_settings_string = " "
for setting in guild_settings :
2023-12-15 10:36:19 -05:00
if ' mysql ' in setting or ' roles ' in setting :
2023-12-14 19:20:01 -05:00
continue
if setting == ' log_channel ' :
channel = ctx . guild . get_channel ( guild_settings [ setting ] )
guild_settings_string + = f " ** { setting } **: { channel . mention } \n "
else :
guild_settings_string + = f " ** { setting } **: { guild_settings [ setting ] } \n "
2023-12-14 19:11:22 -05:00
2023-12-14 19:20:01 -05:00
user_settings = await self . config . user ( ctx . author ) . all ( )
2023-12-14 19:11:22 -05:00
user_settings_string = " "
for setting in user_settings :
user_settings_string + = f " ** { setting } **: { user_settings [ setting ] } \n "
2023-12-14 19:13:15 -05:00
embed = discord . Embed ( color = await self . bot . get_embed_color ( None ) )
2023-12-14 19:11:22 -05:00
embed . set_author ( icon_url = ctx . guild . icon . url , name = f " { ctx . guild . name } Moderation Settings " )
2023-12-14 19:20:01 -05:00
if ctx . guild :
embed . add_field ( name = " Guild Settings " , value = guild_settings_string )
2023-12-14 19:11:22 -05:00
embed . add_field ( name = " User Settings " , value = user_settings_string )
await ctx . send ( embed = embed )
2023-12-14 18:35:25 -05:00
@moderationset.group ( autohelp = True , name = ' history ' )
async def moderationset_history ( self , ctx : commands . Context ) :
""" Manage configuration for the /history command. """
@moderationset_history.command ( name = ' ephemeral ' , aliases = [ ' hidden ' , ' hide ' ] )
2023-12-15 10:05:07 -05:00
async def moderationset_history_user_ephemeral ( self , ctx : commands . Context , enabled : bool ) :
2023-12-14 18:35:25 -05:00
""" Toggle if the /history command should be ephemeral. """
2023-12-15 10:05:07 -05:00
await self . config . user ( ctx . author ) . history_ephemeral . set ( enabled )
await ctx . send ( f " Ephemeral setting set to { enabled } " )
2023-12-14 18:35:25 -05:00
@moderationset_history.command ( name = ' pagesize ' )
async def moderationset_history_user_pagesize ( self , ctx : commands . Context , pagesize : int ) :
""" Set the amount of cases to display per page. """
2023-12-15 22:28:42 -05:00
if pagesize > 20 :
await ctx . send ( " Pagesize cannot be greater than 20! " )
return
if pagesize < 1 :
await ctx . send ( " Pagesize cannot be less than 1! " )
return
2023-12-14 18:35:25 -05:00
await self . config . user ( ctx . author ) . history_pagesize . set ( pagesize )
await ctx . send ( f " Pagesize set to { await self . config . user ( ctx . author ) . history_pagesize ( ) } " )
@moderationset_history.group ( name = ' inline ' )
async def moderationset_history_inline ( self , ctx : commands . Context ) :
""" Manage configuration for the /history command ' s inline argument. """
@moderationset_history_inline.command ( name = ' toggle ' )
2023-12-15 10:05:07 -05:00
async def moderationset_history_user_inline_toggle ( self , ctx : commands . Context , enabled : bool ) :
2023-12-14 18:35:25 -05:00
""" Enable the /history command ' s inline argument by default. """
2023-12-15 10:05:07 -05:00
await self . config . user ( ctx . author ) . history_inline . set ( enabled )
await ctx . send ( f " Inline setting set to { enabled } " )
2023-12-14 18:35:25 -05:00
@moderationset_history_inline.command ( name = ' pagesize ' )
async def moderationset_history_user_inline_pagesize ( self , ctx : commands . Context , pagesize : int ) :
""" Set the amount of cases to display per page. """
2023-12-15 22:28:42 -05:00
if pagesize > 20 :
await ctx . send ( " Pagesize cannot be greater than 20! " )
return
if pagesize < 1 :
await ctx . send ( " Pagesize cannot be less than 1! " )
return
2023-12-14 18:35:25 -05:00
await self . config . user ( ctx . author ) . history_inline_pagesize . set ( pagesize )
await ctx . send ( f " Inline pagesize set to { await self . config . user ( ctx . author ) . history_inline_pagesize ( ) } " )
@moderationset_history.group ( autohelp = True , name = ' guild ' )
@checks.admin ( )
async def moderationset_history_guild ( self , ctx : commands . Context ) :
""" Manage configuration for the /history command, per guild. """
@moderationset_history_guild.command ( name = ' ephemeral ' , aliases = [ ' hidden ' , ' hide ' ] )
@checks.admin ( )
2023-12-15 10:05:07 -05:00
async def moderationset_history_guild_ephemeral ( self , ctx : commands . Context , enabled : bool ) :
2023-12-14 18:35:25 -05:00
""" Toggle if the /history command should be ephemeral. """
2023-12-15 10:05:07 -05:00
await self . config . guild ( ctx . guild ) . history_ephemeral . set ( enabled )
await ctx . send ( f " Ephemeral setting set to { enabled } " )
2023-12-14 18:35:25 -05:00
@moderationset_history_guild.command ( name = ' pagesize ' )
@checks.admin ( )
async def moderationset_history_guild_pagesize ( self , ctx : commands . Context , pagesize : int ) :
""" Set the amount of cases to display per page. """
2023-12-15 22:28:42 -05:00
if pagesize > 20 :
await ctx . send ( " Pagesize cannot be greater than 20! " )
return
if pagesize < 1 :
await ctx . send ( " Pagesize cannot be less than 1! " )
return
2023-12-14 18:35:25 -05:00
await self . config . guild ( ctx . guild ) . history_pagesize . set ( pagesize )
await ctx . send ( f " Pagesize set to { await self . config . guild ( ctx . guild ) . history_pagesize ( ) } " )
@moderationset_history_guild.group ( name = ' inline ' )
@checks.admin ( )
async def moderationset_history_guild_inline ( self , ctx : commands . Context ) :
""" Manage configuration for the /history command ' s inline argument. """
@moderationset_history_guild_inline.command ( name = ' toggle ' )
@checks.admin ( )
2023-12-15 10:05:07 -05:00
async def moderationset_history_guild_inline_toggle ( self , ctx : commands . Context , enabled : bool ) :
2023-12-14 18:35:25 -05:00
""" Enable the /history command ' s inline argument by default. """
2023-12-15 10:05:07 -05:00
await self . config . guild ( ctx . guild ) . history_inline . set ( enabled )
await ctx . send ( f " Inline setting set to { enabled } " )
2023-12-14 18:35:25 -05:00
@moderationset_history_guild_inline.command ( name = ' pagesize ' )
@checks.admin ( )
async def moderationset_history_guild_inline_pagesize ( self , ctx : commands . Context , pagesize : int ) :
""" Set the amount of cases to display per page. """
2023-12-15 22:28:42 -05:00
if pagesize > 20 :
await ctx . send ( " Pagesize cannot be greater than 20! " )
return
if pagesize < 1 :
await ctx . send ( " Pagesize cannot be less than 1! " )
return
2023-12-14 18:35:25 -05:00
await self . config . guild ( ctx . guild ) . history_inline_pagesize . set ( pagesize )
await ctx . send ( f " Inline pagesize set to { await self . config . guild ( ctx . guild ) . history_inline_pagesize ( ) } " )
2023-12-14 21:19:14 -05:00
@moderationset.group ( autohelp = True , name = ' immunity ' )
@checks.admin ( )
async def moderationset_immunity ( self , ctx : commands . Context ) :
""" Manage configuration for immune roles. """
@moderationset_immunity.command ( name = ' add ' )
@checks.admin ( )
async def moderationset_immunity_add ( self , ctx : commands . Context , role : discord . Role ) :
""" Add a role to the immune roles list. """
2023-12-14 21:21:40 -05:00
immune_roles : list = await self . config . guild ( ctx . guild ) . immune_roles ( )
2023-12-14 21:19:14 -05:00
if role . id in immune_roles :
await ctx . send ( " Role is already immune! " )
return
immune_roles . append ( role . id )
await self . config . guild ( ctx . guild ) . immune_roles . set ( immune_roles )
await ctx . send ( f " Role { role . name } added to immune roles. " )
@moderationset_immunity.command ( name = ' remove ' )
@checks.admin ( )
async def moderationset_immunity_remove ( self , ctx : commands . Context , role : discord . Role ) :
""" Remove a role from the immune roles list. """
2023-12-14 21:21:40 -05:00
immune_roles : list = await self . config . guild ( ctx . guild ) . immune_roles ( )
2023-12-14 21:19:14 -05:00
if role . id not in immune_roles :
await ctx . send ( " Role is not immune! " )
return
immune_roles . remove ( role . id )
await self . config . guild ( ctx . guild ) . immune_roles . set ( immune_roles )
await ctx . send ( f " Role { role . name } removed from immune roles. " )
@moderationset_immunity.command ( name = ' list ' )
@checks.admin ( )
async def moderationset_immunity_list ( self , ctx : commands . Context ) :
""" List all immune roles. """
2023-12-14 21:21:40 -05:00
immune_roles : list = await self . config . guild ( ctx . guild ) . immune_roles ( )
2023-12-14 21:19:14 -05:00
if not immune_roles :
await ctx . send ( " No immune roles set! " )
return
role_list = " "
for role_id in immune_roles :
role = ctx . guild . get_role ( role_id )
if role :
role_list + = f " { role . mention } \n "
if role_list :
embed = discord . Embed ( title = " Immune Roles " , description = role_list , color = await self . bot . get_embed_color ( None ) )
2023-12-14 21:22:37 -05:00
await ctx . send ( embed = embed )
2023-12-14 21:19:14 -05:00
@moderationset.group ( autohelp = True , name = ' blacklist ' )
@checks.admin ( )
async def moderationset_blacklist ( self , ctx : commands . Context ) :
""" Manage configuration for the /blacklist command. """
@moderationset_blacklist.command ( name = ' add ' )
@checks.admin ( )
2023-12-14 21:26:23 -05:00
async def moderationset_blacklist_add ( self , ctx : commands . Context , role : discord . Role , duration : str ) :
""" Add a role to the blacklist. """
2023-12-14 21:21:40 -05:00
blacklist_roles : list = await self . config . guild ( ctx . guild ) . blacklist_roles ( )
2023-12-14 21:19:14 -05:00
for blacklist_role in blacklist_roles :
if role . id == blacklist_role [ ' role ' ] :
await ctx . send ( " Role already has an associated blacklist type! " )
return
2023-12-14 21:26:23 -05:00
try :
parsed_time = parse ( sval = duration , as_timedelta = True , raise_exception = True )
except ValueError :
await ctx . send ( " Please provide a valid duration! " )
return
2023-12-14 21:19:14 -05:00
blacklist_roles . append (
{
' role ' : role . id ,
2023-12-14 21:27:45 -05:00
' duration ' : str ( parsed_time )
2023-12-14 21:19:14 -05:00
}
)
await self . config . guild ( ctx . guild ) . blacklist_roles . set ( blacklist_roles )
2023-12-14 21:26:23 -05:00
await ctx . send ( f " Role { role . mention } added as a blacklist type. " , allowed_mentions = discord . AllowedMentions . none ( ) )
2023-12-14 21:19:14 -05:00
@moderationset_blacklist.command ( name = ' remove ' )
@checks.admin ( )
async def moderationset_blacklist_remove ( self , ctx : commands . Context , role : discord . Role ) :
2023-12-14 21:26:23 -05:00
""" Remove a role ' s blacklist type. """
2023-12-14 21:21:40 -05:00
blacklist_roles : list = await self . config . guild ( ctx . guild ) . blacklist_roles ( )
2023-12-14 21:19:14 -05:00
for blacklist_role in blacklist_roles :
if role . id == blacklist_role [ ' role ' ] :
blacklist_roles . remove ( blacklist_role )
await self . config . guild ( ctx . guild ) . blacklist_roles . set ( blacklist_roles )
await ctx . send ( f " Role { role . mention } removed from blacklist types. " , allowed_mentions = discord . AllowedMentions . none ( ) )
return
await ctx . send ( " Role does not have an associated blacklist type! " )
@moderationset_blacklist.command ( name = ' list ' )
@checks.admin ( )
async def moderationset_blacklist_list ( self , ctx : commands . Context ) :
""" List all blacklist types. """
2023-12-14 21:21:40 -05:00
blacklist_roles : list = await self . config . guild ( ctx . guild ) . blacklist_roles ( )
2023-12-14 21:19:14 -05:00
if not blacklist_roles :
await ctx . send ( " No blacklist types set! " )
return
blacklist_list = " "
for blacklist_role in blacklist_roles :
role = ctx . guild . get_role ( blacklist_role [ ' role ' ] )
if role :
blacklist_list + = f " { role . mention } - { blacklist_role [ ' duration ' ] } \n "
if blacklist_list :
embed = discord . Embed ( title = " Blacklist Types " , description = blacklist_list , color = await self . bot . get_embed_color ( None ) )
2023-12-14 21:22:37 -05:00
await ctx . send ( embed = embed )
2023-12-14 21:19:14 -05:00
2023-12-14 18:35:25 -05:00
@moderationset.command ( name = " ignorebots " )
@checks.admin ( )
async def moderationset_ignorebots ( self , ctx : commands . Context ) :
""" Toggle if the cog should ignore other bots ' moderations. """
await self . config . guild ( ctx . guild ) . ignore_other_bots . set ( not await self . config . guild ( ctx . guild ) . ignore_other_bots ( ) )
await ctx . send ( f " Ignore bots setting set to { await self . config . guild ( ctx . guild ) . ignore_other_bots ( ) } " )
@moderationset.command ( name = " dm " )
@checks.admin ( )
async def moderationset_dm ( self , ctx : commands . Context ) :
""" Toggle automatically messaging moderated users.
This option can be overridden by specifying the ` silent ` argument in any moderation command . """
await self . config . guild ( ctx . guild ) . dm_users . set ( not await self . config . guild ( ctx . guild ) . dm_users ( ) )
await ctx . send ( f " DM users setting set to { await self . config . guild ( ctx . guild ) . dm_users ( ) } " )
2023-12-15 14:07:21 -05:00
@moderationset.command ( name = " permissions " )
@checks.admin ( )
async def moderationset_permissions ( self , ctx : commands . Context ) :
""" Toggle whether the bot will check for discord permissions. """
await self . config . guild ( ctx . guild ) . use_discord_permissions . set ( not await self . config . guild ( ctx . guild ) . use_discord_permissions ( ) )
2023-12-15 14:08:16 -05:00
await ctx . send ( f " Use Discord Permissions setting set to { await self . config . guild ( ctx . guild ) . use_discord_permissions ( ) } " )
2023-12-15 14:07:21 -05:00
2023-12-14 18:35:25 -05:00
@moderationset.command ( name = " logchannel " )
@checks.admin ( )
async def moderationset_logchannel ( self , ctx : commands . Context , channel : discord . TextChannel = None ) :
""" Set a channel to log infractions to. """
if channel :
await self . config . guild ( ctx . guild ) . log_channel . set ( channel . id )
await ctx . send ( f " Logging channel set to { channel . mention } . " )
else :
await self . config . guild ( ctx . guild ) . log_channel . set ( " " )
await ctx . send ( " Logging channel disabled. " )
@moderationset.command ( name = " mysql " )
@checks.is_owner ( )
async def moderationset_mysql ( self , ctx : commands . Context ) :
""" Configure MySQL connection details. """
await ctx . message . add_reaction ( " ✅ " )
await ctx . author . send ( content = " Click the button below to configure your MySQL connection details. " , view = self . ConfigButtons ( 60 ) )
class ConfigButtons ( discord . ui . View ) :
def __init__ ( self , timeout ) :
super ( ) . __init__ ( )
self . config = Config . get_conf ( None , cog_name = ' Moderation ' , identifier = 481923957134912 )
@discord.ui.button ( label = " Edit " , style = discord . ButtonStyle . success )
async def config_button ( self , interaction : discord . Interaction , button : discord . ui . Button ) : # pylint: disable=unused-argument
await interaction . response . send_modal ( Moderation . MySQLConfigModal ( self . config ) )
class MySQLConfigModal ( discord . ui . Modal , title = " MySQL Database Configuration " ) :
def __init__ ( self , config ) :
super ( ) . __init__ ( )
self . config = config
address = discord . ui . TextInput (
label = " Address " ,
placeholder = " Input your MySQL address here. " ,
style = discord . TextStyle . short ,
required = False ,
max_length = 300
)
database = discord . ui . TextInput (
label = " Database " ,
placeholder = " Input the name of your database here. " ,
style = discord . TextStyle . short ,
required = False ,
max_length = 300
)
username = discord . ui . TextInput (
label = " Username " ,
placeholder = " Input your MySQL username here. " ,
style = discord . TextStyle . short ,
required = False ,
max_length = 300
)
password = discord . ui . TextInput (
label = " Password " ,
placeholder = " Input your MySQL password here. " ,
style = discord . TextStyle . short ,
required = False ,
max_length = 300
)
async def on_submit ( self , interaction : discord . Interaction ) :
message = " "
if self . address . value != " " :
await self . config . mysql_address . set ( self . address . value )
message + = f " - Address set to \n - ` { self . address . value } ` \n "
if self . database . value != " " :
await self . config . mysql_database . set ( self . database . value )
message + = f " - Database set to \n - ` { self . database . value } ` \n "
if self . username . value != " " :
await self . config . mysql_username . set ( self . username . value )
message + = f " - Username set to \n - ` { self . username . value } ` \n "
if self . password . value != " " :
await self . config . mysql_password . set ( self . password . value )
trimmed_password = self . password . value [ : 8 ]
message + = f " - Password set to \n - ` { trimmed_password } ` - Trimmed for security \n "
if message == " " :
trimmed_password = str ( await self . config . mysql_password ( ) ) [ : 8 ]
send = f " No changes were made. \n Current configuration: \n - Address: \n - ` { await self . config . mysql_address ( ) } ` \n - Database: \n - ` { await self . config . mysql_database ( ) } ` \n - Username: \n - ` { await self . config . mysql_username ( ) } ` \n - Password: \n - ` { trimmed_password } ` - Trimmed for security "
else :
send = f " Configuration changed: \n { message } "
await interaction . response . send_message ( send , ephemeral = True )
2023-12-15 22:14:30 -05:00
@moderationset.group ( autohelp = True , name = ' import ' )
2023-12-14 18:35:25 -05:00
@checks.admin ( )
async def moderationset_import ( self , ctx : commands . Context ) :
""" Import moderations from other bots. """
@moderationset_import.command ( name = " galacticbot " )
@checks.admin ( )
async def moderationset_import_galacticbot ( self , ctx : commands . Context ) :
2023-12-15 22:14:30 -05:00
""" Import moderations from GalacticBot. """
2023-12-15 21:31:00 -05:00
if ctx . message . attachments and ctx . message . attachments [ 0 ] . content_type == ' application/json; charset=utf-8 ' :
2023-12-16 20:49:24 -05:00
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.* " )
2023-12-17 02:16:44 -05:00
await message . edit ( view = self . GalacticBotImportButtons ( 60 , ctx , message ) )
2023-12-15 21:31:00 -05:00
else :
await ctx . send ( " Please provide a valid GalacticBot moderation export file. " )
2023-12-14 18:35:25 -05:00
class GalacticBotImportButtons ( discord . ui . View ) :
2023-12-17 02:16:44 -05:00
def __init__ ( self , timeout , ctx , message ) :
2023-12-14 18:35:25 -05:00
super ( ) . __init__ ( )
2023-12-15 21:31:00 -05:00
self . ctx : commands . Context = ctx
2023-12-14 18:35:25 -05:00
self . message : discord . Message = message
self . config = Config . get_conf ( None , cog_name = ' Moderation ' , identifier = 481923957134912 )
@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
2023-12-15 21:31:00 -05:00
await self . message . delete ( )
await interaction . response . send_message ( " Deleting original table... " , ephemeral = True )
2023-12-17 02:16:44 -05:00
database = await connect ( )
2023-12-15 21:31:00 -05:00
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... " )
2023-12-17 02:16:44 -05:00
await create_guild_table ( self . ctx . guild )
2023-12-15 21:31:00 -05:00
await interaction . edit_original_response ( content = " Importing moderations... " )
accepted_types = [
' NOTE ' ,
' WARN ' ,
' MUTE ' ,
' UNMUTE ' ,
' KICK ' ,
' SOFTBAN ' ,
' BAN ' ,
2023-12-17 12:54:41 -05:00
' UNBAN ' ,
' SLOWMODE ' ,
' LOCKDOWN '
2023-12-15 21:31:00 -05:00
]
2023-12-16 13:35:12 -05:00
2023-12-15 21:31:00 -05:00
file = await self . ctx . message . attachments [ 0 ] . read ( )
data = sorted ( json . loads ( file ) , key = lambda x : x [ ' case ' ] )
2023-12-16 20:43:01 -05:00
failed_cases = [ ]
2023-12-16 13:55:50 -05:00
for case in data :
if case [ ' type ' ] not in accepted_types :
continue
timestamp = round ( case [ ' timestamp ' ] / 1000 )
2023-12-16 20:43:01 -05:00
try :
if case [ ' duration ' ] is not None and float ( case [ ' duration ' ] ) != 0 :
2023-12-16 13:55:50 -05:00
duration = timedelta ( seconds = round ( float ( case [ ' duration ' ] ) / 1000 ) )
2023-12-16 20:43:01 -05:00
else :
duration = ' NULL '
except OverflowError :
2023-12-17 02:36:18 -05:00
failed_cases . append ( case [ ' case ' ] )
continue
2023-12-15 21:31:00 -05:00
2023-12-17 12:54:41 -05:00
metadata = {
' imported_from ' : ' GalacticBot '
}
if case [ ' type ' ] == ' SLOWMODE ' :
metadata [ ' seconds ' ] = case [ ' data ' ] [ ' seconds ' ]
2023-12-16 13:55:50 -05:00
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 = [ ]
2023-12-15 21:31:00 -05:00
2023-12-17 13:00:43 -05:00
if case [ ' reason ' ] :
reason = case [ ' reason ' ]
else :
reason = " NULL "
2023-12-17 02:16:44 -05:00
await mysql_log (
2023-12-16 14:44:05 -05:00
self . ctx . guild . id ,
case [ ' executor ' ] ,
case [ ' type ' ] ,
2023-12-17 12:54:41 -05:00
case [ ' targetType ' ] ,
2023-12-16 14:44:05 -05:00
case [ ' target ' ] ,
0 ,
duration ,
2023-12-17 13:00:43 -05:00
reason ,
2023-12-16 14:44:05 -05:00
timestamp = timestamp ,
resolved = resolved ,
resolved_by = resolved_by ,
resolved_reason = resolved_reason ,
changes = changes ,
2023-12-17 12:54:41 -05:00
metadata = metadata ,
2023-12-16 14:44:05 -05:00
database = database
)
2023-12-16 13:55:50 -05:00
await interaction . edit_original_response ( content = " Import complete. " )
2023-12-16 20:43:01 -05:00
if failed_cases :
2023-12-16 20:49:24 -05:00
await interaction . edit_original_response ( content = f " Import complete. \n *Failed to import the following cases:* \n ``` { failed_cases } ``` " )
2023-12-14 18:35:25 -05:00
@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 )
2023-12-15 21:31:00 -05:00
await self . ctx . message . delete ( 10 )
2023-12-14 18:35:25 -05:00
@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.
* * Example usage * *
` [ p ] timedeltaconvert 1 day 15 hr 82 minutes 52 s `
* * Output * *
` 1 day , 16 : 22 : 52 ` """
try :
parsed_time = parse ( duration , as_timedelta = True , raise_exception = True )
await ctx . send ( f " ` { str ( parsed_time ) } ` " )
except ValueError :
await ctx . send ( " Please provide a convertible value! " )