app.py

   1 import os
   2
   3 from matrix_goal_bot.bot import GoalBot
   4
   5 kwargs = {}
   6 # How to connect to Matrix
   7 kwargs['matrix_server'] = os.environ.get('MATRIX_SERVER')
   8 kwargs['bot_name'] = os.environ.get('MATRIX_BOT')
   9 kwargs['bot_pwd'] = os.environ.get('MATRIX_BOT_PWD')
  10
  11 # Prefix to listen for
  12 kwargs['bot_prefix'] = os.environ.get('MATRIX_BOT_PREFIX', '!chore')
  13
  14 # Where to store data
  15 DATA_DIR = os.environ.get('MATRIX_DATA_DIR', '')
  16 DB_FILE = os.environ.get('MATRIX_DB_FILE', 'goal.db')
  17 kwargs['db_file'] = os.path.join(DATA_DIR, DB_FILE)
  18
  19 goal_bot = GoalBot(**kwargs)
  20
  21 @goal_bot.bot.listener.on_message_event
  22 async def do_bot_stuff(room, message):
  23     '''Pass messages to our bot'''
  24     await goal_bot.handle_message(room, message)
  25
  26 goal_bot.run()
  27