From 8195a27111029846370e63b3dca337ffc029ef63 Mon Sep 17 00:00:00 2001
From: Camil Staps
Date: Sun, 20 Mar 2016 21:57:58 +0100
Subject: Fix bug

---
 kjvbot.py | 26 +++++++++++++++-----------
 1 file changed, 15 insertions(+), 11 deletions(-)

(limited to 'kjvbot.py')

diff --git a/kjvbot.py b/kjvbot.py
index 83e2af8..47259be 100644
--- a/kjvbot.py
+++ b/kjvbot.py
@@ -20,7 +20,7 @@ def search(q):
         ctx = ''
     q = list(filter(lambda s: not s[0] == ':', q))
     print(ctx, q)
-    refs = check_output(['bible', '-f'] + q).decode('utf-8').splitlines()[2:22]
+    refs = check_output(['bible', '-f'] + q).decode('utf-8').splitlines()
     return map(get_verses, list(filter(lambda r: r.find(ctx) == 0, refs)))
 
 class InlineHandler(telepot.async.helper.UserHandler):
@@ -59,15 +59,19 @@ class InlineHandler(telepot.async.helper.UserHandler):
     def on_close(*args, **kwargs):
         pass
 
-TOKEN = sys.argv[1]
+def main():
+    TOKEN = sys.argv[1]
+    
+    bot = telepot.async.DelegatorBot(TOKEN, [
+        (per_inline_from_id(), create_open(InlineHandler, timeout=10)),
+    ])
+    loop = asyncio.get_event_loop()
+    
+    loop.create_task(bot.messageLoop())
+    print('Listening...')
+    
+    loop.run_forever()
 
-bot = telepot.async.DelegatorBot(TOKEN, [
-    (per_inline_from_id(), create_open(InlineHandler, timeout=10)),
-])
-loop = asyncio.get_event_loop()
-
-loop.create_task(bot.messageLoop())
-print('Listening...')
-
-loop.run_forever()
+if __name__ == '__main__':
+    main()
 
-- 
cgit v1.2.3