From f49f34a98f09ce3c752ceedfe2268fb406e926ce Mon Sep 17 00:00:00 2001
From: jesopo
Date: Mon, 30 Sep 2019 16:28:57 +0100
Subject: first_word -> first_words, better if statement

---
 modules/markov.py | 29 ++++++++++++++---------------
 1 file changed, 14 insertions(+), 15 deletions(-)

(limited to 'modules')

diff --git a/modules/markov.py b/modules/markov.py
index 6b7cc1b7..bff01c7e 100644
--- a/modules/markov.py
+++ b/modules/markov.py
@@ -116,7 +116,7 @@ class Module(ModuleManager.BaseModule):
                 stderr.write("Failed to generate markov chain")
 
     def _generate(self, channel_id, first_words):
-        if first_word == None:
+        if not first_words:
             first_words = self.bot.database.execute_fetchall("""SELECT
                 third_word, frequency FROM markov WHERE channel_id=? AND
                 first_word IS NULL AND second_word IS NULL AND third_word
@@ -134,21 +134,20 @@ class Module(ModuleManager.BaseModule):
 
             second_word = self._choose(second_words)
             words = [first_word, second_word]
+        elif len(first_words) == 1:
+            first_word = first_word.lower()
+            second_two_words = self.bot.database.execute_fetchall("""SELECT
+                second_word, third_word, frequency FROM markov WHERE
+                channel_id=? AND first_word=? AND second_word NOT NULL AND
+                third_word NOT NULL""", [channel_id, first_word])
+            if not second_two_words:
+                return None
+
+            second_word, third_word = self._choose(
+                [[[s, t], f] for s, t, f in second_two_words])
+            words = [first_word, second_word, third_word]
         else:
-            if len(first_words) == 1:
-                first_word = first_word.lower()
-                second_two_words = self.bot.database.execute_fetchall("""SELECT
-                    second_word, third_word, frequency FROM markov WHERE
-                    channel_id=? AND first_word=? AND second_word NOT NULL AND
-                    third_word NOT NULL""", [channel_id, first_word])
-                if not second_two_words:
-                    return None
-
-                second_word, third_word = self._choose(
-                    [[[s, t], f] for s, t, f in second_two_words])
-                words = [first_word, second_word, third_word]
-            else:
-                words = [word.lower() for word in first_words]
+            words = [word.lower() for word in first_words]
 
         for i in range(30):
             two_words = words[-2:]
-- 
cgit v1.3.1-10-gc9f91