Correction du bug "wl is not found"

This commit is contained in:
Mathis 2026-02-27 12:26:52 +01:00
parent dd24225120
commit 6372014c99
2 changed files with 158 additions and 17 deletions

View file

@ -7,7 +7,7 @@ from typing import List
WHITELIST_FILE = "data/whitelist.json"
class Whitelist(commands.GroupCog, name="whitelist"):
class Whitelist(commands.Cog):
def __init__(self, bot):
self.bot = bot
self.whitelist = self.load_whitelist()
@ -32,58 +32,95 @@ class Whitelist(commands.GroupCog, name="whitelist"):
self.whitelist[str(guild_id)] = users
self.save_whitelist()
@commands.hybrid_group(name="whitelist", aliases=["wl"], description="Gère la whitelist du serveur")
async def whitelist_group(self, ctx):
"""Groupe de commandes pour gérer la whitelist"""
if ctx.invoked_subcommand is None:
await ctx.send_help(ctx.command)
# ✅ Ajouter un utilisateur
@app_commands.command(name="ajouter", description="Ajoute un utilisateur à la whitelist")
@whitelist_group.command(name="ajouter", description="Ajoute un utilisateur à la whitelist")
@app_commands.describe(utilisateur="L'utilisateur à ajouter")
async def ajouter(self, interaction: discord.Interaction, utilisateur: discord.User):
guild_id = interaction.guild.id
async def ajouter(self, ctx, utilisateur: discord.User):
interaction = ctx.interaction
guild_id = ctx.guild.id
users = self._get_guild_whitelist(guild_id)
user_id = str(utilisateur.id)
if user_id in users:
await interaction.response.send_message(f"🔒 {utilisateur.mention} est déjà dans la whitelist.", ephemeral=True)
msg = f"🔒 {utilisateur.mention} est déjà dans la whitelist."
if interaction:
await interaction.response.send_message(msg, ephemeral=True)
else:
await ctx.send(msg)
return
users.append(user_id)
self._set_guild_whitelist(guild_id, users)
await interaction.response.send_message(f"{utilisateur.mention} a été ajouté à la whitelist.", ephemeral=True)
msg = f"{utilisateur.mention} a été ajouté à la whitelist."
if interaction:
await interaction.response.send_message(msg, ephemeral=True)
else:
await ctx.send(msg)
# ❌ Retirer un utilisateur
@app_commands.command(name="retirer", description="Retire un utilisateur de la whitelist")
@whitelist_group.command(name="retirer", description="Retire un utilisateur de la whitelist")
@app_commands.describe(utilisateur="L'utilisateur à retirer")
async def retirer(self, interaction: discord.Interaction, utilisateur: discord.User):
guild_id = interaction.guild.id
async def retirer(self, ctx, utilisateur: discord.User):
interaction = ctx.interaction
guild_id = ctx.guild.id
users = self._get_guild_whitelist(guild_id)
user_id = str(utilisateur.id)
if user_id not in users:
await interaction.response.send_message(f"⚠️ {utilisateur.mention} n'est pas dans la whitelist.", ephemeral=True)
msg = f"⚠️ {utilisateur.mention} n'est pas dans la whitelist."
if interaction:
await interaction.response.send_message(msg, ephemeral=True)
else:
await ctx.send(msg)
return
users.remove(user_id)
self._set_guild_whitelist(guild_id, users)
await interaction.response.send_message(f"{utilisateur.mention} a été retiré de la whitelist.", ephemeral=True)
msg = f"{utilisateur.mention} a été retiré de la whitelist."
if interaction:
await interaction.response.send_message(msg, ephemeral=True)
else:
await ctx.send(msg)
# 📋 Voir la whitelist
@app_commands.command(name="liste", description="Affiche les utilisateurs de la whitelist")
async def liste(self, interaction: discord.Interaction):
guild_id = interaction.guild.id
@whitelist_group.command(name="liste", description="Affiche les utilisateurs de la whitelist")
async def liste(self, ctx):
interaction = ctx.interaction
guild_id = ctx.guild.id
users = self._get_guild_whitelist(guild_id)
if not users:
await interaction.response.send_message("📭 La whitelist est vide pour ce serveur.", ephemeral=True)
msg = "📭 La whitelist est vide pour ce serveur."
if interaction:
await interaction.response.send_message(msg, ephemeral=True)
else:
await ctx.send(msg)
return
# Defer if interaction because fetch_user might take time
if interaction:
await interaction.response.defer(ephemeral=True)
membres = []
for user_id in users:
try:
utilisateur = await self.bot.fetch_user(int(user_id))
membres.append(f"- {utilisateur.mention} ({utilisateur.name}#{utilisateur.discriminator})")
membres.append(f"- {utilisateur.mention} ({utilisateur.name})")
except discord.NotFound:
membres.append(f"- <@{user_id}> (Utilisateur introuvable)")
texte = "\n".join(membres)
await interaction.response.send_message(f"📃 **Utilisateurs whitelistés pour ce serveur :**\n{texte}", ephemeral=True)
msg = f"📃 **Utilisateurs whitelistés pour ce serveur :**\n{texte}"
if interaction:
await interaction.followup.send(msg, ephemeral=True)
else:
await ctx.send(msg)
def is_whitelisted(self, guild_id: int, user_id: int) -> bool:
"""Check if a user is whitelisted for a specific guild."""