]> git.armaanb.net Git - tall.bot.git/blobdiff - tallbot.py
Reduce amount of information returned
[tall.bot.git] / tallbot.py
index f17e04e5348def307940d2bf257618a16daf7ad5..fba1c1639eb445f2ec52de26a9f36e087525e174 100755 (executable)
@@ -1,6 +1,6 @@
 #!/usr/bin/env python3
 # tall.bot - a simple Discord Wikipedia bot
-# Armaan Bhojwani 2020
+# Armaan Bhojwani 2021
 
 import logging
 import sqlite3
@@ -48,7 +48,9 @@ def tell_more(inp, pid):
         old = int(time.time()) - 150
         c.execute("SELECT query FROM queries WHERE pid = (?) AND time > (?)",
                   (str(pid), old))
-        return c.fetchone()
+
+        (query, *_) = c.fetchone()
+        return str(query)
 
     if find_query(inp):
         return query_lookup(pid)
@@ -80,9 +82,9 @@ class TallBot(discord.Client):
         query = create_query(message.content)
         more_info = tell_more(message.content, message.author)
         if query:
-            await message.channel.send(wiki_sum(query, 2, message.author))
+            await message.channel.send(wiki_sum(query, 1, message.author))
         elif more_info:
-            await message.channel.send(wiki_sum(more_info, 8, message.author))
+            await message.channel.send(wiki_sum(more_info, 6, message.author))
 
 def main():
     prep_database()