matrix_goal_bot/bot.py

   1 from datetime import datetime as dt
   2 import logging
   3 import os
   4
   5 import simplematrixbotlib as botlib
   6
   7 from .db import Data, pretty_print
   8
   9 SERVER = os.environ.get('MATRIX_SERVER')
  10 BOT_NAME = os.environ.get('GOAL_BOT')
  11 BOT_PWD = os.environ.get('GOAL_BOT_PWD')
  12
  13 def get_bot():
  14     creds = botlib.Creds(SERVER, BOT_NAME, BOT_PWD)
  15     return botlib.Bot(creds)
  16
  17 DATE_FORMAT = '%Y-%m-%d'
  18
  19 def today():
  20     return dt.now().strftime(DATE_FORMAT)
  21
  22 PREFIX = '!goal'
  23
  24 logger = logging.getLogger(__name__)
  25 logging.basicConfig(filename='goals.log', encoding='utf-8', level=logging.DEBUG)
  26 logging.getLogger('nio').setLevel(logging.WARNING)
  27
  28 async def do_goal(match):
  29     '''
  30     Prefix: !goal
  31
  32     Synax:
  33         !goal COMMAND ARGS
  34
  35     Commands:
  36       help - show this help
  37       add - add a goal
  38       add-task ID - add a task under the goal specified by ID
  39       add-vision - add a new vision (collection of goals)
  40       close-goal ID - mark goal done by ID
  41       close-vision ID - mark vision as complete by ID
  42       done ID - mark task done by ID
  43       list - list incomplete goals
  44       list-all - list everything
  45       list-done - list completed goals
  46       print GOAL_ID - print a goal and all tasks
  47       print-vision ID - print a vision and all goals
  48       set-vision GOAL_ID VISION_ID- associate GOAL_ID with VISION_ID
  49       visions - list incomplete visions
  50       visionless - list goals with no set vision
  51     '''
  52
  53     if match.command('setup'):
  54         result = Data().setup()
  55         return [result]
  56
  57     if match.command('add'):
  58         new = ' '.join(match.args()[1:])
  59         result = Data().add_goal(new)
  60         return [f'Goal added with id {result}']
  61
  62     if match.command('list'):
  63         goals = Data().get_all_goals()
  64         results = ['# Goals list...']
  65         for goal in goals:
  66             results.append(str(goal))
  67         return results
  68
  69
  70     if match.command('list.done'):
  71         goals = Data().get_done_goals()
  72         results = ['# Completed Goals...']
  73         for goal in goals:
  74             results.append(str(goal))
  75         return results
  76
  77     if match.command('add.task'):
  78         new = ' '.join(match.args()[1:])
  79         result = Data().quick_add_task(new)
  80         goal,tasks = Data().get_goal(goal_id)
  81         results = pretty_print(goal, tasks)
  82         return results
  83
  84     if match.command('add.vision'):
  85         new = ' '.join(match.args()[1:])
  86         result = Data().add_vision(new)
  87         return [f'Vision added with id {result}']
  88
  89     if match.command('print'):
  90         if len(match.args()) > 1:
  91             goal_id = int(match.args()[1])
  92         else:
  93             goal_id = Data()._get_pref_goal_id()
  94
  95         logger.debug(f'Searching for goal {goal_id}...')
  96         goal,tasks = Data().get_goal(goal_id)
  97         results = pretty_print(goal, tasks)
  98
  99         return results
 100
 101
 102     if match.command('print.vision'):
 103         get_id = int(match.args()[1])
 104         vision,goals = Data().get_vision(get_id)
 105         results = pretty_print(vision, goals)
 106         return results
 107
 108     if match.command('close.goal'):
 109         goal_id = int(match.args()[1])
 110         vision_id = Data().complete_goal(goal_id)
 111         vision,goals = Data().get_vision(vision_id)
 112         results = pretty_print(vision, goals)
 113         return results
 114
 115     if match.command('close.vision'):
 116         vision_id = int(match.args()[1])
 117         Data().complete_vision(vision_id)
 118         vision,goals = Data().get_vision(vision_id)
 119         results = pretty_print(vision, goals)
 120         return results
 121
 122     if match.command('done'):
 123         ids = match.args()[1:]
 124         for task_id in ids:
 125             goal_id = Data().complete_task(int(task_id))
 126         # Print the goal of the last closed task
 127         goal,tasks = Data().get_goal(goal_id)
 128         results = pretty_print(goal, tasks)
 129         return results
 130
 131     if match.command('list.all'):
 132         goals = Data().get_all_goals()
 133         results = ['# Everything is...']
 134         for goal in goals:
 135             _,tasks = Data().get_goal(goal.id)
 136             details = pretty_print(goal, tasks)
 137             results += details
 138         return results
 139
 140     if match.command('visions'):
 141         items = Data().get_all_visions()
 142         results = ['# Visions']
 143         if not items:
 144             results.append('-- No current visions --')
 145
 146         for item in items:
 147             results.append(item)
 148         return results
 149
 150     if match.command('visions'):
 151         items = Data().visionless()
 152         results = ['# Goals with no vision']
 153         if not items:
 154             results.append('-- No visionless goals --')
 155
 156         for item in items:
 157             results.append(item)
 158         return results
 159
 160     if match.command('set.vision'):
 161         goal_id = int(match.args()[1])
 162         vision_id = int(match.args()[2])
 163         Data().assign_vision(goal_id=goal_id, vision_id=vision_id)
 164         vision,goals = Data().get_vision(vision_id)
 165         results = pretty_print(vision, goals)
 166         return results
 167
 168
 169     if match.command('rename'):
 170         goal_id = int(match.args()[1])
 171         name = ' '.join(match.args()[2:])
 172         Data().rename_goal(goal_id=goal_id, name=name)
 173         goal,tasks = Data().get_goal(goal_id)
 174         results = pretty_print(goal, tasks)
 175         return results
 176
 177     # Help text
 178     return [do_goal.__doc__]
 179
 180 async def goal_bot(room, message, bot):
 181     match = botlib.MessageMatch(room, message, bot, PREFIX)
 182
 183     if not match.is_not_from_this_bot() and match.prefix():
 184         return
 185
 186     try:
 187         if match.prefix():
 188             result_lines = await do_goal(match)
 189             for line in result_lines:
 190                 await bot.api.send_text_message(
 191                     room.room_id, str(line))
 192     except Exception as ex:
 193         line ='Ye cannot get ye flask.'
 194         await bot.api.send_text_message(
 195                     room.room_id, str(line))
 196         await bot.api.send_text_message(
 197                     room.room_id, str(ex))
 198
 199