116 lines
No EOL
5.5 KiB
Python
116 lines
No EOL
5.5 KiB
Python
import discord
|
|
from discord.ext import commands
|
|
from discord import app_commands
|
|
import sqlite3
|
|
import aiohttp
|
|
import io
|
|
import os
|
|
import logging
|
|
from PIL import Image, ImageDraw, ImageFont
|
|
|
|
class Welcome(commands.Cog):
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
# Base de données à la racine
|
|
self.db_path = os.path.join(os.path.dirname(__file__), "..", "config.db")
|
|
self._init_db()
|
|
|
|
def _init_db(self):
|
|
conn = sqlite3.connect(self.db_path)
|
|
cursor = conn.cursor()
|
|
cursor.execute('''
|
|
CREATE TABLE IF NOT EXISTS server_configs (
|
|
server_id INTEGER PRIMARY KEY,
|
|
welcome_channel INTEGER,
|
|
goodbye_channel INTEGER,
|
|
welcome_dm_message TEXT
|
|
)
|
|
''')
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
async def create_welcome_image(self, username, avatar_url):
|
|
current_dir = os.path.dirname(__file__)
|
|
base_dir = os.path.abspath(os.path.join(current_dir, ".."))
|
|
|
|
try:
|
|
async with aiohttp.ClientSession() as session:
|
|
async with session.get(avatar_url) as resp:
|
|
avatar_bytes = await resp.read()
|
|
|
|
background = Image.open(os.path.join(base_dir, 'background_template.png'))
|
|
draw = ImageDraw.Draw(background)
|
|
|
|
font_large = ImageFont.truetype(os.path.join(base_dir, 'custom_font_large.otf'), 70)
|
|
font_small = ImageFont.truetype(os.path.join(base_dir, 'custom_font_small.otf'), 50)
|
|
font_large_omega = ImageFont.truetype(os.path.join(base_dir, 'custom_font_large.otf'), 55)
|
|
|
|
draw.text((325, 70), "Bienvenue", font=font_large, fill=(255,255,255))
|
|
draw.text((325, 135), f"{username} chez", font=font_small, fill=(255,255,255))
|
|
draw.text((325, 195), "OMEGA KUBE", font=font_large_omega, fill=(255,255,255))
|
|
|
|
avatar = Image.open(io.BytesIO(avatar_bytes)).convert("RGBA").resize((224, 224), Image.LANCZOS)
|
|
mask = Image.new('L', avatar.size, 0)
|
|
mask_draw = ImageDraw.Draw(mask)
|
|
mask_draw.ellipse((12, 12, 212, 212), fill=255)
|
|
background.paste(avatar, (40, 40), mask)
|
|
|
|
img_byte_arr = io.BytesIO()
|
|
background.save(img_byte_arr, format='PNG')
|
|
img_byte_arr.seek(0)
|
|
return img_byte_arr
|
|
except Exception as e:
|
|
logging.error(f"Erreur image: {e}")
|
|
return None
|
|
|
|
@commands.Cog.listener()
|
|
async def on_member_join(self, member):
|
|
conn = sqlite3.connect(self.db_path)
|
|
cursor = conn.cursor()
|
|
# On récupère 2 colonnes : index 0 et 1
|
|
cursor.execute('SELECT welcome_channel, welcome_dm_message FROM server_configs WHERE server_id = ?', (member.guild.id,))
|
|
result = cursor.fetchone()
|
|
conn.close()
|
|
|
|
# 1. Gestion du DM (Index 1)
|
|
welcome_dm_text = result[1] if result and result[1] else "Bienvenue {username} sur le serveur {servername} ! 🎉\n\nNous sommes ravis de t'accueillir parmi nous."
|
|
|
|
try:
|
|
formatted_message = welcome_dm_text.format(
|
|
username=member.name,
|
|
servername=member.guild.name,
|
|
member_count=member.guild.member_count
|
|
)
|
|
await member.send(formatted_message)
|
|
await member.send("https://i.giphy.com/media/v1.Y2lkPTc5MGI3NjExNXV1NW02NTk0bGF2ZHcyMGFna2F6amRrbGpobGpsMHowOW5xa2E5eCZlcD12MV9pbnRlcm5hbF9naWZfYnlfaWQmY3Q9Zw/ZtFuW4rsUbfrRzPauJ/giphy.gif")
|
|
except Exception as e:
|
|
logging.warning(f"Impossible d'envoyer le DM à {member.name}: {e}")
|
|
|
|
# 2. Gestion du salon de bienvenue (Index 0)
|
|
if result and result[0]:
|
|
channel = member.guild.get_channel(result[0])
|
|
if channel:
|
|
img = await self.create_welcome_image(member.name, member.avatar.url if member.avatar else member.default_avatar.url)
|
|
if img:
|
|
embed = discord.Embed(title=f"Merci d'accueillir {member} 🤝",
|
|
description="Bienvenue dans le serveur Omega Kube !\nSi vous cherchez des RolePlay de qualité,\nVous êtes au bon endroit. <a:sip:1316891821858619452>",
|
|
color=discord.Color.dark_purple())
|
|
embed.set_image(url="attachment://welcome.png")
|
|
await channel.send(content=member.mention, embed=embed, file=discord.File(img, filename="welcome.png"))
|
|
|
|
@app_commands.command(name="setwelcome", description="Définit le salon et le message DM de bienvenue")
|
|
@app_commands.checks.has_permissions(administrator=True)
|
|
async def set_welcome(self, interaction: discord.Interaction, channel: discord.TextChannel, message_dm: str = None):
|
|
conn = sqlite3.connect(self.db_path)
|
|
cursor = conn.cursor()
|
|
cursor.execute('''INSERT INTO server_configs (server_id, welcome_channel, welcome_dm_message)
|
|
VALUES (?, ?, ?) ON CONFLICT(server_id)
|
|
DO UPDATE SET welcome_channel=excluded.welcome_channel,
|
|
welcome_dm_message=COALESCE(excluded.welcome_dm_message, welcome_dm_message)''',
|
|
(interaction.guild_id, channel.id, message_dm))
|
|
conn.commit()
|
|
conn.close()
|
|
await interaction.response.send_message("Configuration de bienvenue enregistrée !", ephemeral=True)
|
|
|
|
async def setup(bot):
|
|
await bot.add_cog(Welcome(bot)) |