Add invoke debug task to check db

[matrix-chore-bot.git] / app.py

diff --git a/app.py b/app.py
index fcc26524e1e055b4c0989906d08f72a825b9040d..682b96afefbe40273ef68b2ea1122f31c5adaeba 100644 (file)
--- a/app.py
+++ b/app.py
@@ -1,22 +1,28 @@
 import os
 
 import os
 

-import simplematrixbotlib as botlib
+from matrix_chore_bot.bot import ChoreBot

 
 

-from matrix_chore_bot.bot import chore_bot
+kwargs = {}
+# How to connect to Matrix
+kwargs['matrix_server'] = os.environ.get('MATRIX_SERVER')
+kwargs['bot_name'] = os.environ.get('MATRIX_BOT')
+kwargs['bot_pwd'] = os.environ.get('MATRIX_BOT_PWD')

 
 

-SERVER = os.environ.get('MATRIX_SERVER')
-BOT_NAME = os.environ.get('MATRIX_BOT')
-BOT_PWD = os.environ.get('MATRIX_BOT_PWD')
-BOT_PREFIX = os.environ.get('MATRIX_BOT_PREFIX', '!chore')
+# Prefix to listen for
+kwargs['bot_prefix'] = os.environ.get('MATRIX_BOT_PREFIX', '!chore')

 
 

-def get_bot():
-    creds = botlib.Creds(SERVER, BOT_NAME, BOT_PWD)
-    return botlib.Bot(creds)
+# Where to store data
+DATA_DIR = os.environ.get('MATRIX_DATA_DIR', '')
+DB_FILE = os.environ.get('MATRIX_DB_FILE', 'chore.db')
+kwargs['db_file'] = os.path.join(DATA_DIR, DB_FILE)

 
 

-bot = get_bot()
+print(kwargs)

 
 

-@bot.listener.on_message_event
+chore_bot = ChoreBot(**kwargs) 
+
+@chore_bot.bot.listener.on_message_event

 async def do_bot_stuff(room, message):
 async def do_bot_stuff(room, message):

-    await chore_bot(room, message, bot, BOT_PREFIX)
+    '''Pass messages to our bot'''
+    await chore_bot.handle_message(room, message)

 
 

-bot.run()
+chore_bot.run()