first commit
This commit is contained in:
RayExo
168
bot/cogs/moderation/warn.py
Normal file
168
bot/cogs/moderation/warn.py
Normal file
@@ -0,0 +1,168 @@
|
||||
import discord
|
||||
from utils.emoji import DELETE, TICK
|
||||
from discord.ext import commands
|
||||
from discord import ui
|
||||
import aiosqlite
|
||||
import asyncio
|
||||
from utils.Tools import *
|
||||
|
||||
|
||||
#class WarnView(ui.View):
|
||||
# def __init__(self, user, author):
|
||||
# super().__init__(timeout=60)
|
||||
# self.user = user
|
||||
# self.author = author
|
||||
# self.message = None
|
||||
|
||||
# async def interaction_check(self, interaction: discord.Interaction) -> bool:
|
||||
# if interaction.user != self.author:
|
||||
# await interaction.response.send_message("You are not allowed to interact with this!", ephemeral=True)
|
||||
# return False
|
||||
# return True
|
||||
|
||||
# async def on_timeout(self):
|
||||
# for item in self.children:
|
||||
# item.disabled = True
|
||||
# if self.message:
|
||||
# try:
|
||||
# await self.message.edit(view=self)
|
||||
# except Exception:
|
||||
# pass
|
||||
|
||||
# @ui.button(style=discord.ButtonStyle.gray, emoji=DELETE)
|
||||
# async def delete(self, interaction: discord.Interaction, button: discord.ui.Button):
|
||||
# await interaction.message.delete()
|
||||
|
||||
|
||||
class Warn(commands.Cog):
|
||||
def __init__(self, bot):
|
||||
self.bot = bot
|
||||
self.color = discord.Color.from_rgb(255, 0, 0)
|
||||
self.db_path = "db/warn.db"
|
||||
|
||||
|
||||
asyncio.create_task(self.setup())
|
||||
|
||||
def get_user_avatar(self, user):
|
||||
return user.avatar.url if user.avatar else user.default_avatar.url
|
||||
|
||||
async def add_warn(self, guild_id: int, user_id: int):
|
||||
async with aiosqlite.connect(self.db_path) as db:
|
||||
await db.execute("INSERT OR IGNORE INTO warns (guild_id, user_id, warns) VALUES (?, ?, 0)", (guild_id, user_id))
|
||||
await db.execute("UPDATE warns SET warns = warns + 1 WHERE guild_id = ? AND user_id = ?", (guild_id, user_id))
|
||||
await db.commit()
|
||||
|
||||
async def get_total_warns(self, guild_id: int, user_id: int):
|
||||
async with aiosqlite.connect(self.db_path) as db:
|
||||
async with db.execute("SELECT warns FROM warns WHERE guild_id = ? AND user_id = ?", (guild_id, user_id)) as cursor:
|
||||
row = await cursor.fetchone()
|
||||
if row:
|
||||
return row[0]
|
||||
return 0
|
||||
|
||||
async def reset_warns(self, guild_id: int, user_id: int):
|
||||
async with aiosqlite.connect(self.db_path) as db:
|
||||
await db.execute("UPDATE warns SET warns = 0 WHERE guild_id = ? AND user_id = ?", (guild_id, user_id))
|
||||
await db.commit()
|
||||
|
||||
async def setup(self):
|
||||
try:
|
||||
async with aiosqlite.connect(self.db_path) as db:
|
||||
await db.execute("""
|
||||
CREATE TABLE IF NOT EXISTS warns (
|
||||
guild_id INTEGER,
|
||||
user_id INTEGER,
|
||||
warns INTEGER,
|
||||
PRIMARY KEY (guild_id, user_id)
|
||||
)
|
||||
""")
|
||||
await db.commit()
|
||||
except Exception as e:
|
||||
print(f"Error during database setup: {e}")
|
||||
|
||||
@commands.hybrid_command(
|
||||
name="warn",
|
||||
help="Warn a user in the server",
|
||||
usage="warn <user> [reason]",
|
||||
aliases=["warnuser"])
|
||||
@blacklist_check()
|
||||
@ignore_check()
|
||||
@commands.cooldown(1, 10, commands.BucketType.member)
|
||||
@commands.max_concurrency(1, per=commands.BucketType.default, wait=False)
|
||||
@commands.guild_only()
|
||||
@commands.has_permissions(moderate_members=True)
|
||||
#@commands.bot_has_permissions(manage_messages=True)
|
||||
async def warn(self, ctx, user: discord.Member, *, reason=None):
|
||||
if user == ctx.author:
|
||||
return await ctx.reply("You cannot warn yourself.")
|
||||
|
||||
if user == ctx.bot.user:
|
||||
return await ctx.reply("You cannot warn me.")
|
||||
|
||||
if not ctx.author == ctx.guild.owner:
|
||||
if user == ctx.guild.owner:
|
||||
return await ctx.reply("I cannot warn the server owner.")
|
||||
|
||||
if ctx.author.top_role <= user.top_role:
|
||||
return await ctx.reply("You cannot Warn a member with a higher or equal role.")
|
||||
|
||||
if ctx.guild.me.top_role <= user.top_role:
|
||||
return await ctx.reply("I cannot Warn a member with a higher or equal role.")
|
||||
|
||||
if user not in ctx.guild.members:
|
||||
return await ctx.reply("The user is not a member of this server.")
|
||||
try:
|
||||
|
||||
await self.add_warn(ctx.guild.id, user.id)
|
||||
total_warns = await self.get_total_warns(ctx.guild.id, user.id)
|
||||
|
||||
|
||||
reason_to_send = reason or "No reason provided"
|
||||
try:
|
||||
await user.send(f"You have been warned in **{ctx.guild.name}** by **{ctx.author}**. Reason: {reason_to_send}")
|
||||
dm_status = "Yes"
|
||||
except discord.Forbidden:
|
||||
dm_status = "No"
|
||||
except discord.HTTPException:
|
||||
dm_status = "No"
|
||||
|
||||
|
||||
embed = discord.Embed(description=f"**{TICK} | Successfully Warned [{user}](https://discord.com/users/{user.id})\nReason {reason_to_send}\nNow He has {total_warns} Warns**",
|
||||
color=self.color)
|
||||
embed.set_thumbnail(url=ctx.author.display_avatar.url)
|
||||
embed.set_author(name=f"Successfully Warned {user.name}", icon_url=self.get_user_avatar(user))
|
||||
#embed.add_field(name="Moderator:", value=ctx.author.mention, inline=False)
|
||||
embed.set_footer(text=f"Requested by {ctx.author}", icon_url=self.get_user_avatar(ctx.author))
|
||||
embed.timestamp = discord.utils.utcnow()
|
||||
|
||||
# view = WarnView(user=user, author=ctx.author)
|
||||
message = await ctx.send(embed=embed)
|
||||
# view.message = message
|
||||
except Exception as e:
|
||||
await ctx.send(f"An error occurred: {str(e)}")
|
||||
print(f"Error during warn command: {e}")
|
||||
|
||||
@commands.hybrid_command(
|
||||
name="clearwarns",
|
||||
help="Clear all warnings for a user",
|
||||
aliases=["clearwarn" , "clearwarnings"],
|
||||
usage="clearwarns <user>")
|
||||
@blacklist_check()
|
||||
@ignore_check()
|
||||
@commands.cooldown(1, 10, commands.BucketType.member)
|
||||
@commands.max_concurrency(1, per=commands.BucketType.default, wait=False)
|
||||
@commands.guild_only()
|
||||
@commands.has_permissions(moderate_members=True)
|
||||
async def clearwarns(self, ctx, user: discord.Member):
|
||||
try:
|
||||
await self.reset_warns(ctx.guild.id, user.id)
|
||||
embed = discord.Embed(description=f"{TICK} | All warnings have been cleared for **{user}** in this guild.", color=self.color)
|
||||
embed.set_author(name=f"Warnings Cleared", icon_url=self.get_user_avatar(user))
|
||||
embed.set_footer(text=f"Requested by {ctx.author}", icon_url=self.get_user_avatar(ctx.author))
|
||||
embed.timestamp = discord.utils.utcnow()
|
||||
|
||||
await ctx.send(embed=embed)
|
||||
except Exception as e:
|
||||
await ctx.send(f"An error occurred: {str(e)}")
|
||||
print(f"Error during clearwarns command: {e}")
|
||||
|
||||
Reference in New Issue
Block a user