Fix des derniers détails
This commit is contained in:
parent
07198c20b7
commit
00446c5fb8
3 changed files with 130 additions and 21 deletions
|
|
@ -28,7 +28,8 @@ class Goodbye(commands.Cog):
|
|||
welcome_banner_background TEXT,
|
||||
welcome_channel_message TEXT,
|
||||
goodbye_server_name TEXT,
|
||||
goodbye_message TEXT
|
||||
goodbye_message TEXT,
|
||||
goodbye_banner_background TEXT
|
||||
)
|
||||
''')
|
||||
|
||||
|
|
@ -42,6 +43,9 @@ class Goodbye(commands.Cog):
|
|||
if 'goodbye_message' not in columns:
|
||||
cursor.execute('ALTER TABLE server_configs ADD COLUMN goodbye_message TEXT')
|
||||
|
||||
if 'goodbye_banner_background' not in columns:
|
||||
cursor.execute('ALTER TABLE server_configs ADD COLUMN goodbye_banner_background TEXT')
|
||||
|
||||
conn.commit()
|
||||
conn.close()
|
||||
|
||||
|
|
@ -73,7 +77,7 @@ class Goodbye(commands.Cog):
|
|||
seconds = delta.seconds % 60
|
||||
return f"il y a {seconds} {'seconde' if seconds <= 1 else 'secondes'}"
|
||||
|
||||
async def create_goodbye_image(self, username, avatar_url, server_name=None):
|
||||
async def create_goodbye_image(self, username, avatar_url, server_name=None, banner_background=None):
|
||||
current_dir = os.path.dirname(__file__)
|
||||
base_dir = os.path.abspath(os.path.join(current_dir, ".."))
|
||||
|
||||
|
|
@ -81,7 +85,18 @@ class Goodbye(commands.Cog):
|
|||
async with session.get(avatar_url) as resp:
|
||||
avatar_bytes = await resp.read()
|
||||
|
||||
background = Image.open(os.path.join(base_dir, 'background_template_goodbye.png'))
|
||||
# Utiliser le fond personnalisé ou celui par défaut
|
||||
bg_path = None
|
||||
if banner_background:
|
||||
# On cherche dans le dossier banners/
|
||||
potential_path = os.path.join(base_dir, "banners", banner_background)
|
||||
if os.path.exists(potential_path):
|
||||
bg_path = potential_path
|
||||
|
||||
if bg_path:
|
||||
background = Image.open(bg_path)
|
||||
else:
|
||||
background = Image.open(os.path.join(base_dir, 'background_template_goodbye.png'))
|
||||
draw = ImageDraw.Draw(background)
|
||||
|
||||
font_large = ImageFont.truetype(os.path.join(base_dir, 'custom_font_large.otf'), 70)
|
||||
|
|
@ -110,7 +125,7 @@ class Goodbye(commands.Cog):
|
|||
async def on_member_remove(self, member):
|
||||
conn = sqlite3.connect(self.db_path)
|
||||
cursor = conn.cursor()
|
||||
cursor.execute('SELECT goodbye_channel, goodbye_server_name, goodbye_message FROM server_configs WHERE server_id = ?', (member.guild.id,))
|
||||
cursor.execute('SELECT goodbye_channel, goodbye_server_name, goodbye_message, goodbye_banner_background FROM server_configs WHERE server_id = ?', (member.guild.id,))
|
||||
result = cursor.fetchone()
|
||||
conn.close()
|
||||
|
||||
|
|
@ -118,13 +133,15 @@ class Goodbye(commands.Cog):
|
|||
goodbye_channel = result[0]
|
||||
server_name = result[1]
|
||||
custom_message = result[2]
|
||||
banner_background = result[3]
|
||||
|
||||
channel = member.guild.get_channel(goodbye_channel)
|
||||
if channel:
|
||||
img = await self.create_goodbye_image(
|
||||
member.name,
|
||||
member.avatar.url if member.avatar else member.default_avatar.url,
|
||||
server_name=server_name
|
||||
server_name=server_name,
|
||||
banner_background=banner_background
|
||||
)
|
||||
joined_message = self.humanize_time_delta(member.joined_at)
|
||||
|
||||
|
|
@ -138,23 +155,67 @@ class Goodbye(commands.Cog):
|
|||
embed.set_image(url="attachment://goodbye.png")
|
||||
await channel.send(embed=embed, file=discord.File(img, filename="goodbye.png"))
|
||||
|
||||
async def save_banner_attachment(self, attachment: discord.Attachment, guild_id: int) -> str:
|
||||
"""Sauvegarde l'attachment et retourne le chemin du fichier"""
|
||||
base_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "banners"))
|
||||
os.makedirs(base_dir, exist_ok=True)
|
||||
|
||||
# Vérifier que c'est une image
|
||||
if not attachment.content_type or not attachment.content_type.startswith('image/'):
|
||||
raise ValueError("Le fichier doit être une image.")
|
||||
|
||||
# Vérifier le format PNG
|
||||
if not attachment.filename.lower().endswith('.png'):
|
||||
raise ValueError("Le fichier doit être au format PNG.")
|
||||
|
||||
# Vérifier les dimensions
|
||||
banner_path = os.path.join(base_dir, f"banner_goodbye_{guild_id}.png")
|
||||
|
||||
async with aiohttp.ClientSession() as session:
|
||||
async with session.get(attachment.url) as resp:
|
||||
if resp.status == 200:
|
||||
# Vérifier les dimensions avant de sauvegarder
|
||||
img_data = await resp.read()
|
||||
from io import BytesIO
|
||||
with Image.open(BytesIO(img_data)) as img:
|
||||
width, height = img.size
|
||||
if width != 1000 or height != 300:
|
||||
raise ValueError(f"Les dimensions doivent être 1000x300 pixels (votre image: {width}x{height})")
|
||||
|
||||
with open(banner_path, 'wb') as f:
|
||||
f.write(img_data)
|
||||
|
||||
return os.path.basename(banner_path)
|
||||
|
||||
@app_commands.command(name="setgoodbye", description="Définit le salon d'aurevoir et les paramètres personnalisés")
|
||||
@app_commands.checks.has_permissions(administrator=True)
|
||||
async def set_goodbye(self, interaction: discord.Interaction, channel: discord.TextChannel, server_name: str = None, message: str = None):
|
||||
async def set_goodbye(self, interaction: discord.Interaction, channel: discord.TextChannel, server_name: str = None, message: str = None, banner_file: discord.Attachment = None):
|
||||
banner_path = None
|
||||
|
||||
# Si un fichier est uploadé, le sauvegarder
|
||||
if banner_file:
|
||||
try:
|
||||
banner_path = await self.save_banner_attachment(banner_file, interaction.guild_id)
|
||||
except Exception as e:
|
||||
await interaction.response.send_message(f"❌ Erreur avec le fichier: {e}", ephemeral=True)
|
||||
return
|
||||
|
||||
conn = sqlite3.connect(self.db_path)
|
||||
cursor = conn.cursor()
|
||||
cursor.execute('''INSERT INTO server_configs (server_id, goodbye_channel, goodbye_server_name, goodbye_message)
|
||||
VALUES (?, ?, ?, ?) ON CONFLICT(server_id)
|
||||
cursor.execute('''INSERT INTO server_configs (server_id, goodbye_channel, goodbye_server_name, goodbye_message, goodbye_banner_background)
|
||||
VALUES (?, ?, ?, ?, ?) ON CONFLICT(server_id)
|
||||
DO UPDATE SET goodbye_channel=excluded.goodbye_channel,
|
||||
goodbye_server_name=COALESCE(excluded.goodbye_server_name, goodbye_server_name),
|
||||
goodbye_message=COALESCE(excluded.goodbye_message, goodbye_message)''',
|
||||
(interaction.guild_id, channel.id, server_name, message))
|
||||
goodbye_message=COALESCE(excluded.goodbye_message, goodbye_message),
|
||||
goodbye_banner_background=COALESCE(excluded.goodbye_banner_background, goodbye_banner_background)''',
|
||||
(interaction.guild_id, channel.id, server_name, message, banner_path))
|
||||
conn.commit()
|
||||
conn.close()
|
||||
|
||||
msg = f"✅ Salon d'aurevoir défini sur {channel.mention}"
|
||||
if server_name: msg += f"\nNom du serveur: {server_name}"
|
||||
if message: msg += f"\nMessage: {message}"
|
||||
if banner_path: msg += f"\nBannière personnalisée activée."
|
||||
|
||||
await interaction.response.send_message(msg, ephemeral=True)
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue