ConfrerieDuKassoulait/KassouBot
ConfrerieDuKassoulait/KassouBot
Archived
2
0
Fork 0
Code Issues 3 Pull requests Projects Releases Wiki Activity

ajout guild pour éviter de supprimer le mauvais reminder

Browse source
This commit is contained in:
Mylloon 2021-06-16 01:20:42 +02:00
parent 80abb0cce5
commit 5d9be53290
2 changed files with 9 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/cogs/utils.py
View file

@ -654,7 +654,7 @@ class Utils(commands.Cog):
else:
return await ctx.send("Veuillez renseigner un ID.")
verification = Reminder().appartenanceReminder(ctx.author.id, id)
verification = Reminder().appartenanceReminder(ctx.author.id, id, ctx.guild.id)
if verification:
Reminder().suppressionReminder(id)
if fromSlash != True:
@ -663,7 +663,7 @@ class Utils(commands.Cog):
else:
if fromSlash != True:
await ctx.message.add_reaction(emoji = '❌')
return await ctx.send("Rappel non trouvé ou qui ne vous appartiens pas.")
return await ctx.send("Rappel non trouvé, pas sur le bon serveur ou qui ne vous appartiens pas.")
@cog_ext.cog_slash(name="reminderdelete", description = "Suppprime un rappel.")
async def __reminderdelete(self, ctx, id):
return await self._reminderdelete(ctx, id, True)

16
src/utils/reminder.py
View file

@ -56,14 +56,12 @@ class Reminder(Database):
"""
return self.affichageResultat(self.requete(requete, time))
def appartenanceReminder(self, user = int, id = int):
"""Vérifie qu'un rappel appartiens à un utilisateur. Renvois False si le rappel n'existe pas."""
def appartenanceReminder(self, user = int, id = int, guildID = int):
"""Vérifie qu'un rappel appartiens à un utilisateur et que la guilde soit la bonne. Renvois False si le rappel n'existe pas."""
requete = """
SELECT user_id, id, message_id FROM reminder
WHERE id = ? AND user_id = ?
SELECT EXISTS (
SELECT 1 FROM reminder
WHERE id = ? AND user_id = ? AND (guild_id = ? OR guild_id = 0)
)
"""
resultat = self.affichageResultat(self.requete(requete, [id, user]))
if len(resultat) > 0:
return True
else:
return False
return True if self.affichageResultat(self.requete(requete, [id, user, guildID]))[0][0] == 1 else False