mirror of
https://github.com/neoarz/Syntrel.git
synced 2025-12-25 11:40:12 +01:00
feat: add message deletions when banning
Added support for specifying a time period to delete a banned user's messages, including options for hours, days, or all messages. Improved permission checks, error handling, and user feedback for the ban command.
This commit is contained in:
neoarz
@@ -1,11 +1,3 @@
|
||||
"""
|
||||
Copyright © Krypton 2019-Present - https://github.com/kkrypt0nn (https://krypton.ninja)
|
||||
Description:
|
||||
🐍 A simple template to start to code your own and personalized Discord bot in Python
|
||||
|
||||
Version: 6.4.0
|
||||
"""
|
||||
|
||||
import discord
|
||||
from discord import app_commands
|
||||
from discord.ext import commands
|
||||
@@ -20,52 +12,205 @@ class Ban(commands.Cog, name="ban"):
|
||||
name="ban",
|
||||
description="Bans a user from the server.",
|
||||
)
|
||||
@commands.has_permissions(ban_members=True)
|
||||
@commands.bot_has_permissions(ban_members=True)
|
||||
@app_commands.describe(
|
||||
user="The user that should be banned.",
|
||||
reason="The reason why the user should be banned.",
|
||||
delete_messages="Delete messages from the user (choose time period).",
|
||||
)
|
||||
@app_commands.choices(delete_messages=[
|
||||
app_commands.Choice(name="Don't delete any messages", value="none"),
|
||||
app_commands.Choice(name="Last 1 hour", value="1h"),
|
||||
app_commands.Choice(name="Last 6 hours", value="6h"),
|
||||
app_commands.Choice(name="Last 12 hours", value="12h"),
|
||||
app_commands.Choice(name="Last 24 hours", value="1d"),
|
||||
app_commands.Choice(name="Last 3 days", value="3d"),
|
||||
app_commands.Choice(name="Last 7 days", value="7d"),
|
||||
])
|
||||
async def ban(
|
||||
self, context: Context, user: discord.User, *, reason: str = "Not specified"
|
||||
self, context: Context, user: discord.User, *, reason: str = "Not specified", delete_messages: str = "none"
|
||||
) -> None:
|
||||
"""
|
||||
Bans a user from the server.
|
||||
|
||||
:param context: The hybrid command context.
|
||||
:param user: The user that should be banned from the server.
|
||||
:param reason: The reason for the ban. Default is "Not specified".
|
||||
"""
|
||||
member = context.guild.get_member(user.id) or await context.guild.fetch_member(
|
||||
user.id
|
||||
)
|
||||
try:
|
||||
if member.guild_permissions.administrator:
|
||||
embed = discord.Embed(
|
||||
description="User has administrator permissions.", color=0xE02B2B
|
||||
)
|
||||
await context.send(embed=embed)
|
||||
else:
|
||||
embed = discord.Embed(
|
||||
description=f"**{member}** was banned by **{context.author}**!",
|
||||
color=0xBEBEFE,
|
||||
)
|
||||
embed.add_field(name="Reason:", value=reason)
|
||||
await context.send(embed=embed)
|
||||
member = context.guild.get_member(user.id)
|
||||
if not member:
|
||||
try:
|
||||
await member.send(
|
||||
f"You were banned by **{context.author}** from **{context.guild.name}**!\nReason: {reason}"
|
||||
)
|
||||
except:
|
||||
pass
|
||||
await member.ban(reason=reason)
|
||||
except:
|
||||
member = await context.guild.fetch_member(user.id)
|
||||
except discord.NotFound:
|
||||
try:
|
||||
await context.guild.ban(user, reason=reason)
|
||||
embed = discord.Embed(
|
||||
title="Ban",
|
||||
description=f"**{user}** was banned by **{context.author}**!",
|
||||
color=0x7289DA,
|
||||
).set_author(name="Moderation", icon_url="https://yes.nighty.works/raw/8VLDcg.webp")
|
||||
embed.add_field(name="Reason:", value=reason)
|
||||
await context.send(embed=embed)
|
||||
return
|
||||
except discord.Forbidden:
|
||||
embed = discord.Embed(
|
||||
title="Error!",
|
||||
description="I don't have permission to ban this user.",
|
||||
color=0xE02B2B,
|
||||
).set_author(name="Moderation", icon_url="https://yes.nighty.works/raw/8VLDcg.webp")
|
||||
await context.send(embed=embed, ephemeral=True)
|
||||
return
|
||||
except Exception:
|
||||
embed = discord.Embed(
|
||||
title="Error!",
|
||||
description="An error occurred while trying to ban the user.",
|
||||
color=0xE02B2B,
|
||||
).set_author(name="Moderation", icon_url="https://yes.nighty.works/raw/8VLDcg.webp")
|
||||
await context.send(embed=embed, ephemeral=True)
|
||||
return
|
||||
|
||||
if not context.author.guild_permissions.ban_members and context.author != context.guild.owner:
|
||||
embed = discord.Embed(
|
||||
title="Missing Permissions!",
|
||||
description="You don't have the `Ban Members` permission to use this command.",
|
||||
color=0xE02B2B,
|
||||
).set_author(name="Moderation", icon_url="https://yes.nighty.works/raw/8VLDcg.webp")
|
||||
await context.send(embed=embed, ephemeral=True)
|
||||
return
|
||||
|
||||
if member and member.top_role >= context.guild.me.top_role:
|
||||
embed = discord.Embed(
|
||||
title="Cannot Ban User",
|
||||
description="This user has a higher or equal role to me. Make sure my role is above theirs.",
|
||||
color=0xE02B2B,
|
||||
).set_author(name="Moderation", icon_url="https://yes.nighty.works/raw/8VLDcg.webp")
|
||||
await context.send(embed=embed, ephemeral=True)
|
||||
return
|
||||
|
||||
if member and context.author != context.guild.owner:
|
||||
if member.top_role >= context.author.top_role:
|
||||
embed = discord.Embed(
|
||||
title="Cannot Ban User",
|
||||
description="You cannot ban this user as they have a higher or equal role to you.",
|
||||
color=0xE02B2B,
|
||||
).set_author(name="Moderation", icon_url="https://yes.nighty.works/raw/8VLDcg.webp")
|
||||
await context.send(embed=embed, ephemeral=True)
|
||||
return
|
||||
|
||||
delete_message_days = 0
|
||||
delete_all_messages = False
|
||||
|
||||
if delete_messages != "none":
|
||||
if delete_messages == "all":
|
||||
delete_all_messages = True
|
||||
delete_message_days = 7
|
||||
elif delete_messages.endswith("h"):
|
||||
hours = int(delete_messages[:-1])
|
||||
delete_message_days = min(hours / 24, 7)
|
||||
elif delete_messages.endswith("d"):
|
||||
days = int(delete_messages[:-1])
|
||||
delete_message_days = min(days, 7)
|
||||
|
||||
try:
|
||||
if member:
|
||||
try:
|
||||
dm_embed = discord.Embed(
|
||||
title="Ban",
|
||||
description=f"You were banned by **{context.author}** from **{context.guild.name}**!\nReason: {reason}",
|
||||
color=0xE02B2B,
|
||||
).set_author(name="Moderation", icon_url="https://yes.nighty.works/raw/8VLDcg.webp")
|
||||
await member.send(embed=dm_embed)
|
||||
except (discord.Forbidden, discord.HTTPException):
|
||||
pass
|
||||
|
||||
if member:
|
||||
await member.ban(reason=reason, delete_message_days=delete_message_days)
|
||||
else:
|
||||
await context.guild.ban(user, reason=reason, delete_message_days=delete_message_days)
|
||||
|
||||
if delete_all_messages:
|
||||
await self.delete_all_user_messages(context.guild, user.id)
|
||||
|
||||
embed = discord.Embed(
|
||||
title="Ban",
|
||||
description=f"**{user}** was banned by **{context.author}**!",
|
||||
color=0x7289DA,
|
||||
).set_author(name="Moderation", icon_url="https://yes.nighty.works/raw/8VLDcg.webp")
|
||||
embed.add_field(name="Reason:", value=reason)
|
||||
|
||||
if delete_messages != "none":
|
||||
if delete_all_messages:
|
||||
embed.add_field(name="Messages Deleted:", value="All messages", inline=False)
|
||||
else:
|
||||
delete_time_text = self.format_delete_time(delete_messages)
|
||||
embed.add_field(name="Messages Deleted:", value=delete_time_text, inline=False)
|
||||
|
||||
await context.send(embed=embed)
|
||||
|
||||
except discord.Forbidden:
|
||||
embed = discord.Embed(
|
||||
title="Error!",
|
||||
description="I don't have permission to ban this user. Make sure my role is above theirs.",
|
||||
color=0xE02B2B,
|
||||
).set_author(name="Moderation", icon_url="https://yes.nighty.works/raw/8VLDcg.webp")
|
||||
await context.send(embed=embed, ephemeral=True)
|
||||
except discord.HTTPException as e:
|
||||
if "Cannot ban the owner of a guild" in str(e):
|
||||
embed = discord.Embed(
|
||||
title="Cannot Ban User",
|
||||
description="You cannot ban the server owner.",
|
||||
color=0xE02B2B,
|
||||
).set_author(name="Moderation", icon_url="https://yes.nighty.works/raw/8VLDcg.webp")
|
||||
else:
|
||||
embed = discord.Embed(
|
||||
title="Error!",
|
||||
description=f"Discord API error: {str(e)}",
|
||||
color=0xE02B2B,
|
||||
).set_author(name="Moderation", icon_url="https://yes.nighty.works/raw/8VLDcg.webp")
|
||||
await context.send(embed=embed, ephemeral=True)
|
||||
except Exception as e:
|
||||
embed = discord.Embed(
|
||||
title="Debug Error!",
|
||||
description=f"Error type: {type(e).__name__}\nError message: {str(e)}",
|
||||
color=0xE02B2B,
|
||||
).set_author(name="Moderation", icon_url="https://yes.nighty.works/raw/8VLDcg.webp")
|
||||
await context.send(embed=embed, ephemeral=True)
|
||||
|
||||
except Exception as e:
|
||||
embed = discord.Embed(
|
||||
title="Error!",
|
||||
description="An error occurred while trying to ban the user. Make sure my role is above the role of the user you want to ban.",
|
||||
description="An unexpected error occurred.",
|
||||
color=0xE02B2B,
|
||||
)
|
||||
await context.send(embed=embed)
|
||||
).set_author(name="Moderation", icon_url="https://yes.nighty.works/raw/8VLDcg.webp")
|
||||
await context.send(embed=embed, ephemeral=True)
|
||||
|
||||
async def delete_all_user_messages(self, guild: discord.Guild, user_id: int) -> None:
|
||||
for channel in guild.text_channels:
|
||||
try:
|
||||
permissions = channel.permissions_for(guild.me)
|
||||
if not (permissions.read_message_history and permissions.manage_messages):
|
||||
continue
|
||||
|
||||
deleted = True
|
||||
while deleted:
|
||||
deleted = False
|
||||
try:
|
||||
async for message in channel.history(limit=100):
|
||||
if message.author.id == user_id:
|
||||
try:
|
||||
await message.delete()
|
||||
deleted = True
|
||||
except (discord.NotFound, discord.Forbidden, discord.HTTPException):
|
||||
continue
|
||||
except (discord.Forbidden, discord.HTTPException):
|
||||
break
|
||||
|
||||
except (discord.Forbidden, discord.HTTPException):
|
||||
continue
|
||||
|
||||
def format_delete_time(self, delete_option: str) -> str:
|
||||
time_formats = {
|
||||
"1h": "Last 1 hour",
|
||||
"6h": "Last 6 hours",
|
||||
"12h": "Last 12 hours",
|
||||
"1d": "Last 24 hours",
|
||||
"3d": "Last 3 days",
|
||||
"7d": "Last 7 days"
|
||||
}
|
||||
return time_formats.get(delete_option, "Unknown time period")
|
||||
|
||||
|
||||
async def setup(bot) -> None:
|
||||
|
||||
Reference in New Issue
Block a user