cswimr/IndiumRevolt
cswimr/IndiumRevolt
1
1
Fork 0
Code Issues 4 Pull requests Projects 1 Activity Actions

big commit

Browse source 
made mysql_log a method to reduce repeat code, added warn command, added ban command, added unban command
This commit is contained in:
SeaswimmerTheFsh 2023-06-21 09:20:16 -04:00
parent 48e0b710c2
commit f1754b3e1e
1 changed files with 55 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

76
cogs/moderation.py
View file

@ -15,38 +15,72 @@ db_user = os.getenv('DB_USER')
db_password = os.getenv('DB_PASSWORD') db_password = os.getenv('DB_PASSWORD')
db = os.getenv('DB') 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): class Moderation(commands.Cog):
def __init__(self, bot): def __init__(self, bot):
self.bot = bot self.bot = bot
disable_dateutil() disable_dateutil()
@commands.command(name="mute", aliases=["timeout"]) def mysql_log(self, type, target_id, duration, reason):
async def mute(self, ctx: commands.Context, target: commands.MemberConverter, *, duration: str = "1 hour"): 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: try:
parsed_time = parse(duration, as_timedelta=True, raise_exception=True) parsed_time = parse(sval=duration, as_timedelta=True, raise_exception=True)
except ValueError: except ValueError:
await ctx.message.reply(f"Please provide a valid duration!\nSee `{prefix}tdc`") await ctx.message.reply(f"Please provide a valid duration!\nSee `{prefix}tdc`")
return return
await target.timeout(parsed_time) # await target.timeout(parsed_time)
await ctx.message.reply(f"{target.mention} has been timed out for {str(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)}.", colour="#5d82d1")] 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) # await target.send(embeds=embeds)
# latest_id = cursor.execute("SELECT * FROM mod ORDER BY moderation_id DESC LIMIT 1;") Moderation.mysql_log(self, type='Timeout', target_id=target.id, duration=parsed_time, reason=reason)
# 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)
# 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"]) @commands.command(aliases=["tdc"])
async def timedeltaconvert(self, ctx, *, duration): async def timedeltaconvert(self, ctx, *, duration):