diff --git a/cogs/moderation.py b/cogs/moderation.py index d42f5a8..6c534bc 100644 --- a/cogs/moderation.py +++ b/cogs/moderation.py @@ -15,38 +15,72 @@ db_user = os.getenv('DB_USER') db_password = os.getenv('DB_PASSWORD') db = os.getenv('DB') -moddb = mysql.connector.connect( -host=db_host, -user=db_user, -password=db_password, -database=db -) -cursor = moddb.cursor() - class Moderation(commands.Cog): def __init__(self, bot): self.bot = bot disable_dateutil() - @commands.command(name="mute", aliases=["timeout"]) - async def mute(self, ctx: commands.Context, target: commands.MemberConverter, *, duration: str = "1 hour"): + def mysql_log(self, type, target_id, duration, reason): + moddb = mysql.connector.connect(host=db_host,user=db_user,password=db_password,database=db) + cursor = moddb.cursor() + cursor.execute("SELECT moderation_id FROM `mod` ORDER BY moderation_id DESC LIMIT 1") + moderation_id = cursor.fetchone()[0] + 1 + sql = "INSERT INTO `mod` (moderation_id, moderation_type, target_id, duration, reason, resolved) VALUES (%s, %s, %s, %s, %s, %s)" + val = (moderation_id, type, target_id, duration, reason, 0) + cursor.execute(sql, val) + moddb.commit() + moddb.close() + print(f"MySQL Row Inserted!\n{moderation_id}, {type}, {target_id}, {duration}, {reason}, 0") + + @commands.command(name="timeout", aliases=["mute"]) + async def timeout(self, ctx: commands.Context, target: commands.MemberConverter, duration: str, *, reason: str): try: - parsed_time = parse(duration, as_timedelta=True, raise_exception=True) + parsed_time = parse(sval=duration, as_timedelta=True, raise_exception=True) except ValueError: await ctx.message.reply(f"Please provide a valid duration!\nSee `{prefix}tdc`") return - await target.timeout(parsed_time) - await ctx.message.reply(f"{target.mention} has been timed out for {str(parsed_time)}!") - embeds = [revolt.SendableEmbed(title="Timed Out", description=f"You have been timed out for {str(parsed_time)}.", colour="#5d82d1")] - await target.send(embeds=embeds) - # latest_id = cursor.execute("SELECT * FROM mod ORDER BY moderation_id DESC LIMIT 1;") - # sql = "INSERT INTO mod (moderation_id, moderation_type, target_id, duration, reason) VALUES (%s, %s, %s, %s, %s)" - # val = (latest_id, "Timeout", target.id, parsed_time, "Testing") - # cursor.execute(sql, val) + # await target.timeout(parsed_time) + await ctx.message.reply(f"{target.mention} has been timed out for {str(parsed_time)}!\n**Reason** - `{reason}`") + embeds = [revolt.SendableEmbed(title="Timed Out", description=f"You have been timed out for {str(parsed_time)}.\n### Reason\n`{reason}", colour="#5d82d1")] + # await target.send(embeds=embeds) + Moderation.mysql_log(self, type='Timeout', target_id=target.id, duration=parsed_time, reason=reason) - # moddb.commit() + @commands.command() + async def warn(self, ctx: commands.Context, target: commands.MemberConverter, *, reason: str): + if not reason: + await ctx.message.reply("Please include a reason!") + return + await ctx.message.reply(f"{target.mention} has been warned!\n**Reason** - `{reason}`") + embeds = [revolt.SendableEmbed(title="Warned", description=f"You have been warned.\n### Reason\n`{reason}", colour="#5d82d1")] + # await target.send(embeds=embeds) + Moderation.mysql_log(self, type='Warning', target_id=target.id, duration='NULL', reason=reason) + + @commands.command() + async def ban(self, ctx: commands.Context, target: commands.MemberConverter, *, reason: str): + if not reason: + await ctx.message.reply("Please include a reason!") + return + try: + embeds = [revolt.SendableEmbed(title="Banned", description=f"You have been banned.\n### Reason\n`{reason}", colour="#5d82d1")] + # await target.send(embeds=embeds) + await target.ban(reason=reason) + await ctx.message.reply(f"{target.mention} has been banned!\n**Reason** - `{reason}`") + Moderation.mysql_log(self, type='Ban', target_id=target.id, duration='NULL', reason=reason) + except revolt.errors.HTTPError: + await ctx.message.reply(f"{target.mention} is already banned!") + + @commands.command() + async def unban(self, ctx: commands.Context, target: commands.MemberConverter): + try: + await target.unban() + embeds = [revolt.SendableEmbed(title="Unbanned", description=f"You have been unbanned.", colour="#5d82d1")] + # await target.send(embeds=embeds) + await ctx.message.reply(f"{target.mention} has been unbanned!") + Moderation.mysql_log(self, type='Unban', target_id=target.id, duration='NULL', reason=f'Unbanned through {prefix}unban') + except revolt.errors.HTTPError: + await ctx.message.reply(f"{target.mention} is not banned!") + return - # print(moddb.rowcount, "record inserted.") @commands.command(aliases=["tdc"]) async def timedeltaconvert(self, ctx, *, duration):