mirror of
https://github.com/skylanix/MamieHenriette.git
synced 2026-02-06 06:40:35 +01:00
Ajout de la gestion des messages de bienvenue et de départ pour les membres. Mise à jour des configurations pour activer ces fonctionnalités dans le panneau d'administration.
This commit is contained in:
@@ -6,9 +6,10 @@ import random
|
||||
from database import db
|
||||
from database.helpers import ConfigurationHelper
|
||||
from database.models import Configuration, Humeur, Commande
|
||||
from discord import Message, TextChannel
|
||||
from discord import Message, TextChannel, Member
|
||||
from discordbot.humblebundle import checkHumbleBundleAndNotify
|
||||
from discordbot.command import handle_warning_command, handle_remove_warning_command, handle_list_warnings_command, handle_ban_command, handle_kick_command, handle_unban_command
|
||||
from discordbot.welcome import sendWelcomeMessage, sendLeaveMessage, updateInviteCache
|
||||
from protondb import searhProtonDb
|
||||
|
||||
class DiscordBot(discord.Client):
|
||||
@@ -17,6 +18,9 @@ class DiscordBot(discord.Client):
|
||||
for c in self.get_all_channels() :
|
||||
logging.info(f'{c.id} {c.name}')
|
||||
|
||||
for guild in self.guilds:
|
||||
await updateInviteCache(guild)
|
||||
|
||||
self.loop.create_task(self.updateStatus())
|
||||
self.loop.create_task(self.updateHumbleBundle())
|
||||
|
||||
@@ -54,6 +58,8 @@ class DiscordBot(discord.Client):
|
||||
|
||||
intents = discord.Intents.default()
|
||||
intents.message_content = True
|
||||
intents.members = True
|
||||
intents.invites = True
|
||||
bot = DiscordBot(intents=intents)
|
||||
|
||||
# https://discordpy.readthedocs.io/en/stable/quickstart.html
|
||||
@@ -121,4 +127,20 @@ async def on_message(message: Message):
|
||||
await message.channel.send(msg, suppress_embeds=True)
|
||||
except Exception as e:
|
||||
logging.error(f'Échec de l\'envoi du message ProtonDB : {e}')
|
||||
|
||||
|
||||
@bot.event
|
||||
async def on_member_join(member: Member):
|
||||
await sendWelcomeMessage(bot, member)
|
||||
|
||||
@bot.event
|
||||
async def on_member_remove(member: Member):
|
||||
await sendLeaveMessage(bot, member)
|
||||
|
||||
@bot.event
|
||||
async def on_invite_create(invite):
|
||||
await updateInviteCache(invite.guild)
|
||||
|
||||
@bot.event
|
||||
async def on_invite_delete(invite):
|
||||
await updateInviteCache(invite.guild)
|
||||
|
||||
|
||||
147
discordbot/welcome.py
Normal file
147
discordbot/welcome.py
Normal file
@@ -0,0 +1,147 @@
|
||||
import discord
|
||||
import logging
|
||||
from database.helpers import ConfigurationHelper
|
||||
from discord import Member, TextChannel
|
||||
from datetime import datetime, timezone
|
||||
|
||||
invite_cache = {}
|
||||
|
||||
async def updateInviteCache(guild):
|
||||
try:
|
||||
invites = await guild.invites()
|
||||
invite_cache[guild.id] = {invite.code: invite.uses for invite in invites}
|
||||
except:
|
||||
pass
|
||||
|
||||
async def getUsedInvite(guild) -> str:
|
||||
try:
|
||||
new_invites = await guild.invites()
|
||||
for invite in new_invites:
|
||||
old_uses = invite_cache.get(guild.id, {}).get(invite.code, 0)
|
||||
if invite.uses > old_uses:
|
||||
await updateInviteCache(guild)
|
||||
inviter = f' (créée par {invite.inviter.name})' if invite.inviter else ''
|
||||
return f'`{invite.code}`{inviter}'
|
||||
await updateInviteCache(guild)
|
||||
except:
|
||||
pass
|
||||
return 'Inconnue'
|
||||
|
||||
async def sendWelcomeMessage(bot: discord.Client, member: Member):
|
||||
config = ConfigurationHelper()
|
||||
|
||||
if not config.getValue('welcome_enable'):
|
||||
return
|
||||
|
||||
channel_id = config.getIntValue('welcome_channel_id')
|
||||
if not channel_id:
|
||||
logging.warning('Canal de bienvenue non configuré')
|
||||
return
|
||||
|
||||
channel = bot.get_channel(channel_id)
|
||||
if not channel or not isinstance(channel, TextChannel):
|
||||
logging.error(f'Canal de bienvenue {channel_id} introuvable')
|
||||
return
|
||||
|
||||
welcome_message = config.getValue('welcome_message')
|
||||
if not welcome_message:
|
||||
welcome_message = 'Bienvenue sur le serveur !'
|
||||
|
||||
invite_used = await getUsedInvite(member.guild)
|
||||
|
||||
embed = discord.Embed(
|
||||
title='🎉 Nouveau membre !',
|
||||
description=welcome_message,
|
||||
color=discord.Color.green()
|
||||
)
|
||||
|
||||
embed.set_thumbnail(url=member.display_avatar.url)
|
||||
embed.add_field(name='Membre', value=member.mention, inline=True)
|
||||
embed.add_field(name='Nombre de membres', value=str(member.guild.member_count), inline=True)
|
||||
embed.add_field(name='Invitation utilisée', value=invite_used, inline=False)
|
||||
embed.set_footer(text=f'ID: {member.id}')
|
||||
|
||||
try:
|
||||
await channel.send(embed=embed)
|
||||
logging.info(f'Message de bienvenue envoyé pour {member.name}')
|
||||
except Exception as e:
|
||||
logging.error(f'Échec de l\'envoi du message de bienvenue : {e}')
|
||||
|
||||
def formatDuration(seconds: int) -> str:
|
||||
days = seconds // 86400
|
||||
hours = (seconds % 86400) // 3600
|
||||
minutes = (seconds % 3600) // 60
|
||||
|
||||
parts = []
|
||||
if days > 0:
|
||||
parts.append(f'{days} jour{"s" if days > 1 else ""}')
|
||||
if hours > 0:
|
||||
parts.append(f'{hours} heure{"s" if hours > 1 else ""}')
|
||||
if minutes > 0:
|
||||
parts.append(f'{minutes} minute{"s" if minutes > 1 else ""}')
|
||||
|
||||
if not parts:
|
||||
return 'moins d\'une minute'
|
||||
|
||||
return ' et '.join(parts)
|
||||
|
||||
async def sendLeaveMessage(bot: discord.Client, member: Member):
|
||||
config = ConfigurationHelper()
|
||||
|
||||
if not config.getValue('leave_enable'):
|
||||
return
|
||||
|
||||
channel_id = config.getIntValue('leave_channel_id')
|
||||
if not channel_id:
|
||||
logging.warning('Canal de départ non configuré')
|
||||
return
|
||||
|
||||
channel = bot.get_channel(channel_id)
|
||||
if not channel or not isinstance(channel, TextChannel):
|
||||
logging.error(f'Canal de départ {channel_id} introuvable')
|
||||
return
|
||||
|
||||
leave_message = config.getValue('leave_message')
|
||||
if not leave_message:
|
||||
leave_message = 'Un membre a quitté le serveur.'
|
||||
|
||||
now = datetime.now(timezone.utc)
|
||||
duration_seconds = int((now - member.joined_at).total_seconds()) if member.joined_at else 0
|
||||
duration_text = formatDuration(duration_seconds)
|
||||
|
||||
reason = 'Départ volontaire'
|
||||
try:
|
||||
async for entry in member.guild.audit_logs(limit=5):
|
||||
if entry.target and entry.target.id == member.id:
|
||||
if entry.action == discord.AuditLogAction.kick:
|
||||
reason = f'Expulsé par {entry.user.mention}'
|
||||
if entry.reason:
|
||||
reason += f' - Raison: {entry.reason}'
|
||||
break
|
||||
elif entry.action == discord.AuditLogAction.ban:
|
||||
reason = f'Banni par {entry.user.mention}'
|
||||
if entry.reason:
|
||||
reason += f' - Raison: {entry.reason}'
|
||||
break
|
||||
except:
|
||||
pass
|
||||
|
||||
embed = discord.Embed(
|
||||
title='👋 Membre parti',
|
||||
description=leave_message,
|
||||
color=discord.Color.red()
|
||||
)
|
||||
|
||||
embed.set_thumbnail(url=member.display_avatar.url)
|
||||
embed.add_field(name='Membre', value=f'{member.mention} ({member.name})', inline=True)
|
||||
embed.add_field(name='Nombre de membres', value=str(member.guild.member_count), inline=True)
|
||||
embed.add_field(name='Temps sur le serveur', value=duration_text, inline=False)
|
||||
embed.add_field(name='Raison', value=reason, inline=False)
|
||||
embed.set_footer(text=f'ID: {member.id}')
|
||||
|
||||
try:
|
||||
await channel.send(embed=embed)
|
||||
logging.info(f'Message de départ envoyé pour {member.name}')
|
||||
except Exception as e:
|
||||
logging.error(f'Échec de l\'envoi du message de départ : {e}')
|
||||
|
||||
Reference in New Issue
Block a user