From c638a6b5df18a8db4b0eafed3cadec5af03ced28 Mon Sep 17 00:00:00 2001
From: Mylloon <kennel.anri@tutanota.com>
Date: Wed, 4 Aug 2021 18:30:49 +0200
Subject: [PATCH] using dict instead of variables

---
 main.py | 56 +++++++++++++++++++++++++++++++++-----------------------
 1 file changed, 33 insertions(+), 23 deletions(-)

diff --git a/main.py b/main.py
index 3c6f83a..321f610 100644
--- a/main.py
+++ b/main.py
@@ -18,7 +18,7 @@ def load(variables) -> dict:
                 res = list(set(res.split(',')) - {""}) # create a list for the channels and remove blank channels and doubles
             keys[var] = res
         except KeyError:
-            print(f"Please set the environment variable {var} (.env file supported)")
+            print(f"Veuillez définir la variable d'environnement {var} (fichier .env supporté)")
             exit(1)
     return keys
 
@@ -38,18 +38,21 @@ class Listener(StreamListener):
                 lastWord = tweetText.split()[-1:][0]
                 print(f"Tweet trouvé (dernier mot: \"{lastWord}\")...", end = " ")
                 if lastWord in universalBase: # check if the last word found is a supported word
-                    if lastWord in quoiBase:
-                        answer = feur
-                    elif lastWord in ouiBase:
-                        answer = stiti
-                    elif lastWord in nonBase:
-                        answer = bril
-                    print(f"Envoie d'un {answer[0]}...", end = " ")
-                    try: # send answer
-                        self.api.update_status(status = choice(answer), in_reply_to_status_id = status._json["id"], auto_populate_reply_metadata = True)
-                        print(f"{status._json['user']['screen_name']} s'est fait {answer[0]} !")
-                    except Exception as error:
-                        print(f"\n{errorMessage} {error}")
+                    answer = None
+                    for mot in base.items():
+                        if lastWord in mot[1]:
+                            answer = answers[mot[0]]
+                    if answer == None:
+                        print(f"{errorMessage} Aucune réponse trouvée.")
+                    else:
+                        print(f"Envoie d'un {answer[0]}...", end = " ")
+                        try: # send answer
+                            self.api.update_status(status = choice(answer), in_reply_to_status_id = status._json["id"], auto_populate_reply_metadata = True)
+                            print(f"{status._json['user']['screen_name']} s'est fait {answer[0]} !")
+                        except Exception as error:
+                            print(f"\n{errorMessage} {error}")
+                else:
+                    print("Annulation parce que le dernier mot n'est pas intéressant.")
 
     def do_stuff(self):
         while True:
@@ -91,7 +94,7 @@ def permute(array: list) -> list:
             quoiListe.append(temp)
     return quoiListe
 
-def createBaseTrigger(*lists) -> list:
+def createBaseTrigger(lists) -> list:
     """Merges all given lists into one."""
     listing = []
     for liste in lists:
@@ -124,22 +127,29 @@ if __name__ == '__main__':
     --
     PSEUDO is the PSEUDO of the account you want to listen to snipe.
     """
-    errorMessage = "Error happens!" # error message
+    errorMessage = "Une erreur survient !" # error message
 
     # words to detect
-    quoiBase = ["quoi", "koi", "quoient", "q u o i"]
-    ouiBase = ["oui", "ui"]
-    nonBase = ["non", "nn"]
-    universalBase = createBaseTrigger(quoiBase, ouiBase, nonBase)
+    base = {
+        "quoi": ["quoi", "koi", "quoient", "q u o i"],
+        "oui": ["oui", "ui"],
+        "non": ["non", "nn"]
+    }
+    universalBase = createBaseTrigger(list(base.values()))
 
     # creation of the list with all alternatives (upper/lower case)
     triggerWords = permute(universalBase)
 
     # creation of answers
-    feur = createBaseAnswers("feur")
-    feur.extend(["https://twitter.com/shukuzi62/status/1422611919538724868/video/1", "feur (-isson)", "https://twitter.com/antoinelae/status/1422943594403581957/video/1"]) # add a message in addition to the default answers
-    stiti = createBaseAnswers("stiti")
-    bril = createBaseAnswers("bril")
+    answers = {
+        "quoi": createBaseAnswers("feur") + [
+            "https://twitter.com/shukuzi62/status/1422611919538724868/video/1",
+            "feur (-isson)",
+            "https://twitter.com/antoinelae/status/1422943594403581957/video/1"
+        ],
+        "oui": createBaseAnswers("stiti"),
+        "non": createBaseAnswers("bril")
+    }
     
     # loading environment variables and launching the bot
     keys = load(["TOKEN", "TOKEN_SECRET", "CONSUMER_KEY", "CONSUMER_SECRET", "PSEUDOS"])