2 # tall.bot - a simple Discord Wikipedia bot
12 def wiki_sum(search, sen, pid):
14 logging.info(str(pid) + " | " + str(int(time.time())) + " | " + str(search))
15 log_query(search, pid)
16 return "<Wikipedia> " + wikipedia.summary(search,
19 except Exception as e:
20 logging.debug(str(pid) + " | " + str(search) + " | " + str(e))
23 def create_query(inp):
24 key1 = ["What", "what", "who", "Who", "what's", "What's", "whats", "Whats",
26 key2 = ["is", "are", "were", "was"]
27 key3 = ["a", "an", "the"]
30 if x[0] in key1 and x[1] in key2:
39 def tell_more(inp, pid):
41 key1 = ["Tell me more", "tell me more"]
47 def query_lookup(pid):
48 old = int(time.time()) - 150
49 c.execute("SELECT query FROM queries WHERE pid = (?) AND time > (?)",
54 return query_lookup(pid)
59 c.execute("""CREATE TABLE IF NOT EXISTS queries(
65 def log_query(search, pid):
66 c.execute("DELETE FROM queries WHERE pid = (?)", (str(pid),))
67 c.execute("""INSERT INTO queries
68 (time, pid, query) VALUES(?, ?, ?)""",
69 (int(time.time()), str(pid), str(search)))
72 class TallBot(discord.Client):
73 async def on_ready(self):
74 logging.info('Logged in as ', self.user)
76 async def on_message(self, message):
77 if message.author == self.user:
80 query = create_query(message.content)
81 more_info = tell_more(message.content, message.author)
83 await message.channel.send(wiki_sum(query, 2, message.author))
85 await message.channel.send(wiki_sum(more_info, 8, message.author))
89 TallBot().run(sys.argv[1])
91 if __name__ == "__main__":
92 conn = sqlite3.connect("queries.db")