Restart service after update

[matrix-cadbury.git] / app.py

diff --git a/app.py b/app.py
index b3b352f753155c6c3e4ed64b3280ed4473eee6b9..35b1be8ca4480a5375501675e70960fec64c9aba 100644 (file)
--- a/app.py
+++ b/app.py
@@ -1,9 +1,32 @@
+import os
+
+import simplematrixbotlib as botlib
+

 from matrix_goal_bot.bot import get_bot, goal_bot
 from matrix_goal_bot.bot import get_bot, goal_bot

+from matrix_chore_bot.bot import chore_bot
+from matrix_shop_bot.bot import shop_bot
+
+SERVER = os.environ.get('MATRIX_SERVER')
+BOT_NAME = os.environ.get('MATRIX_BOT')
+BOT_PWD = os.environ.get('MATRIX_BOT_PWD')
+
+creds = botlib.Creds(SERVER, BOT_NAME, BOT_PWD)
+bot = botlib.Bot(creds)

 
 

-bot = get_bot()

 
 @bot.listener.on_message_event
 
 @bot.listener.on_message_event

-async def do_bot_stuff(room, message):
+async def do_goal_stuff(room, message):

     await goal_bot(room, message, bot)
 
     await goal_bot(room, message, bot)
 

+
+@bot.listener.on_message_event
+async def do_chore_stuff(room, message):
+    await chore_bot(room, message, bot)
+
+
+@bot.listener.on_message_event
+async def do_shopping_stuff(room, message):
+    await shop_bot(room, message, bot)
+
+

 bot.run()
 bot.run()