From 7d06794d93bb7011bb6e1192b53ec9edc102be24 Mon Sep 17 00:00:00 2001
From: Mylloon <kennel.anri@outlook.com>
Date: Mon, 14 Jun 2021 01:00:10 +0200
Subject: [PATCH] =?UTF-8?q?ajout=20m=C3=A9thode=20pour=20v=C3=A9rifier=20l?=
 =?UTF-8?q?'appartenance=20d'un=20rappel=20=C3=A0=20un=20user?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/utils/reminder.py | 22 +++++++++++++++++-----
 1 file changed, 17 insertions(+), 5 deletions(-)

diff --git a/src/utils/reminder.py b/src/utils/reminder.py
index 082ebb8..487e406 100644
--- a/src/utils/reminder.py
+++ b/src/utils/reminder.py
@@ -5,7 +5,7 @@ class Reminder(Database):
         super().__init__(r"db/bot.sqlite3")
 
     def creationTable(self):
-        """Créer la table qui stocker les reminders"""
+        """Créer la table qui stocker les reminders."""
         requete = """
                   CREATE TABLE IF NOT EXISTS reminder (
                       id INTEGER PRIMARY KEY,
@@ -22,7 +22,7 @@ class Reminder(Database):
         self.requete(requete)
     
     def ajoutReminder(self, messageID = int, channelID = int, mention = int, reminder = str, creation = int, expiration = int, userID = int, guildID = int):
-        """Ajoute un reminder"""
+        """Ajoute un reminder."""
         requete = """
                   INSERT INTO reminder (
                       message_id, channel_id, extrarg_id, reminder_str, creation_int, expiration_int, user_id, guild_id
@@ -33,7 +33,7 @@ class Reminder(Database):
         self.requete(requete, [messageID, channelID, mention, reminder, creation, expiration, userID, guildID])
     
     def suppressionReminder(self, id = int):
-        """Supprime un reminder"""
+        """Supprime un reminder."""
         requete = """
                   DELETE FROM reminder
                   WHERE id = ?
@@ -41,7 +41,7 @@ class Reminder(Database):
         self.requete(requete, id)
 
     def listeReminder(self, userID = int, guildID = int):
-        """Retourne la liste des reminders d'un utilisateur"""
+        """Retourne la liste des reminders d'un utilisateur."""
         requete = """
                   SELECT reminder_str, creation_int, expiration_int, id FROM reminder
                   WHERE user_id = ? AND guild_id = ?
@@ -49,9 +49,21 @@ class Reminder(Database):
         return self.affichageResultat(self.requete(requete, [userID, guildID]))
 
     def recuperationExpiration(self, time = int):
-        """Récupère les reminders qui sont arrivés à expiration et ses infos"""
+        """Récupère les reminders qui sont arrivés à expiration et ses infos."""
         requete = """
                   SELECT channel_id, extrarg_id, reminder_str, creation_int, user_id, id, message_id FROM reminder
                   WHERE expiration_int < ?
                   """
         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."""
+        requete = """
+                  SELECT user_id, id, message_id FROM reminder
+                  WHERE id = ? AND user_id = ?
+                  """
+        resultat = self.affichageResultat(self.requete(requete, [id, user]))
+        if len(resultat) > 0:
+            return True
+        else:
+            return False