77 lines
3 KiB
Python
77 lines
3 KiB
Python
import discord
|
||
from discord import app_commands
|
||
from discord.ext import commands, tasks
|
||
import json
|
||
from datetime import datetime, timedelta
|
||
import os
|
||
|
||
EMS_ROLE_ID = 1278317212675084328 # Remplace par l’ID du rôle EMS # Remplace par l’ID du rôle EMS
|
||
ROLE_A_ATTRIBUER_ID = 1367198876548993045 # Remplace par l’ID du rôle visite médicale # Remplace par l’ID du rôle à attribuer
|
||
FICHIER_ROLES = "visite_médicale.json"
|
||
|
||
class AttribuerRole(commands.Cog):
|
||
def __init__(self, bot):
|
||
self.bot = bot
|
||
self.roles_temporaire = self.charger_roles()
|
||
self.verification_roles.start()
|
||
|
||
def charger_roles(self):
|
||
if os.path.exists(FICHIER_ROLES):
|
||
with open(FICHIER_ROLES, "r", encoding="utf-8") as f:
|
||
return json.load(f)
|
||
return {}
|
||
|
||
def sauvegarder_roles(self):
|
||
with open(FICHIER_ROLES, "w", encoding="utf-8") as f:
|
||
json.dump(self.roles_temporaire, f, indent=4, ensure_ascii=False)
|
||
|
||
@app_commands.command(name="valider_visite_médicale", description="Attribue un rôle temporaire suite à une visite médicale (EMS uniquement).")
|
||
async def valider_visite(self, interaction: discord.Interaction, membre: discord.Member):
|
||
if not any(role.id == EMS_ROLE_ID for role in interaction.user.roles):
|
||
await interaction.response.send_message("❌ Seuls les EMS peuvent utiliser cette commande.", ephemeral=True)
|
||
return
|
||
|
||
role = interaction.guild.get_role(ROLE_A_ATTRIBUER_ID)
|
||
if not role:
|
||
await interaction.response.send_message("❌ Rôle à attribuer introuvable.", ephemeral=True)
|
||
return
|
||
|
||
await membre.add_roles(role)
|
||
expiration = (datetime.now() + timedelta(days=30)).strftime("%Y-%m-%d %H:%M:%S")
|
||
self.roles_temporaire[str(membre.id)] = {
|
||
"role_id": role.id,
|
||
"expiration": expiration
|
||
}
|
||
self.sauvegarder_roles()
|
||
|
||
await interaction.response.send_message(f"✅ Rôle temporaire attribué à {membre.mention} jusqu'au {expiration}.", ephemeral=True)
|
||
|
||
@tasks.loop(hours=1)
|
||
async def verification_roles(self):
|
||
now = datetime.now()
|
||
to_remove = []
|
||
|
||
for user_id, data in self.roles_temporaire.items():
|
||
expiration = datetime.strptime(data["expiration"], "%Y-%m-%d %H:%M:%S")
|
||
if now >= expiration:
|
||
to_remove.append(user_id)
|
||
|
||
for user_id in to_remove:
|
||
member = self.bot.get_user(int(user_id)) or await self.bot.fetch_user(int(user_id))
|
||
guild = discord.utils.get(self.bot.guilds)
|
||
member = guild.get_member(int(user_id))
|
||
role = guild.get_role(ROLE_A_ATTRIBUER_ID)
|
||
|
||
if member and role:
|
||
await member.remove_roles(role)
|
||
del self.roles_temporaire[user_id]
|
||
|
||
if to_remove:
|
||
self.sauvegarder_roles()
|
||
|
||
@verification_roles.before_loop
|
||
async def before_verification(self):
|
||
await self.bot.wait_until_ready()
|
||
|
||
async def setup(bot):
|
||
await bot.add_cog(AttribuerRole(bot))
|