aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--kjvbot.py26
1 files changed, 15 insertions, 11 deletions
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()