-logging.basicConfig(filename='goals.log', encoding='utf-8', level=logging.DEBUG)
-logging.getLogger('nio').setLevel(logging.WARNING)
-
-async def do_goal(match):
- '''
- Prefix: !goal
-
- Synax:
- !goal COMMAND ARGS
-
- Commands:
- help - show this help
- extra.help - show extended help
- add - add a goal
- task ID - add a task under the goal specified by ID
- close.goal ID - mark goal done by ID
- done ID - mark task done by ID
- list - list incomplete goals
- print GOAL_ID - print a goal and all tasks
- '''
- ex_help = '''
- add.vision - add a new vision (collection of goals)
- close.vision ID - mark vision as complete by ID
- list.all - list everything
- print.vision ID - print a vision and all goals
- set.vision GOAL_ID VISION_ID- associate GOAL_ID with VISION_ID
- visions - list incomplete visions
- list.done - list completed goals
- visionless - list goals with no set vision
- '''
-
- if match.command('setup'):
- result = Data().setup()
- return [result]
-
- if match.command('add'):
- new = ' '.join(match.args()[1:])
- result = Data().add_goal(new)
- return [f'Goal added with id {result}']
-
- if match.command('list'):
- goals = Data().get_all_goals()
- results = ['# Goals list...']
- for goal in goals:
- results.append(str(goal))
- return results
-
-
- if match.command('list.done'):
- goals = Data().get_done_goals()
- results = ['# Completed Goals...']
- for goal in goals:
- results.append(str(goal))
- return results
-
- if match.command('task'):
- new = ' '.join(match.args()[1:])
- goal_id = Data().quick_add_task(new)
- goal,tasks = Data().get_goal(goal_id)
- results = pretty_print(goal, tasks)
- return results
-
- if match.command('add.vision'):
- new = ' '.join(match.args()[1:])
- result = Data().add_vision(new)
- return [f'Vision added with id {result}']
-
- if match.command('print'):
- if len(match.args()) > 1:
- goal_id = int(match.args()[1])
- else:
- goal_id = Data()._get_pref_goal_id()
-
- logger.debug(f'Searching for goal {goal_id}...')
- goal,tasks = Data().get_goal(goal_id)
- results = pretty_print(goal, tasks)
-
- return results
-
-
- if match.command('print.vision'):
- get_id = int(match.args()[1])
- vision,goals = Data().get_vision(get_id)
- results = pretty_print(vision, goals)
- return results
-
- if match.command('close.goal'):
- goal_id = int(match.args()[1])
- vision_id = Data().complete_goal(goal_id)
- vision,goals = Data().get_vision(vision_id)
- results = pretty_print(vision, goals)
- return results
-
- if match.command('close.vision'):
- vision_id = int(match.args()[1])
- Data().complete_vision(vision_id)
- vision,goals = Data().get_vision(vision_id)
- results = pretty_print(vision, goals)
- return results
-
- if match.command('done'):
- ids = match.args()[1:]
- for task_id in ids:
- goal_id = Data().complete_task(int(task_id))
- # Print the goal of the last closed task
- goal,tasks = Data().get_goal(goal_id)
- results = pretty_print(goal, tasks)
- return results
-
- if match.command('list.all'):
- goals = Data().get_all_goals()
- results = ['# Everything is...']
- for goal in goals:
- _,tasks = Data().get_goal(goal.id)
- details = pretty_print(goal, tasks)
- results += details
- return results
-
- if match.command('visions'):
- items = Data().get_all_visions()
- results = ['# Visions']
- if not items:
- results.append('-- No current visions --')
-
- for item in items:
- results.append(item)
- return results
-
- if match.command('visionless'):
- items = Data().visionless()
- results = ['# Goals with no vision']
- if not items:
- results.append('-- No visionless goals --')
-
- for item in items:
- results.append(item)
- return results
-
- if match.command('set.vision'):
- goal_id = int(match.args()[1])
- vision_id = int(match.args()[2])
- Data().assign_vision(goal_id=goal_id, vision_id=vision_id)
- vision,goals = Data().get_vision(vision_id)
- results = pretty_print(vision, goals)
- return results
-
-
- if match.command('rename'):
- goal_id = int(match.args()[1])
- name = ' '.join(match.args()[2:])
- Data().rename_goal(goal_id=goal_id, name=name)
- goal,tasks = Data().get_goal(goal_id)
- results = pretty_print(goal, tasks)
- return results
-
- if match.command('extra.help'):
- return [ex_help]
-
- # Help text
- return [do_goal.__doc__]
-
-async def goal_bot(room, message, bot):
- match = botlib.MessageMatch(room, message, bot, PREFIX)
-
- if not match.is_not_from_this_bot() and match.prefix():
- return
-
- try:
+
+snark = [
+ 'Ye cannot get ye flask.',
+ 'Ye flask must be load bearing, or something...',
+ '''You have:
+ no tea
+ ''',
+ 'How are you gentlemen?',
+ 'All your base are belong to us.',
+ 'You have been eaten by a groo.'
+]
+
+class GoalBot():
+ def __init__(self, matrix_server,
+ bot_name,
+ bot_pwd,
+ db_file=None,
+ bot_prefix='!goal'):
+
+ creds = botlib.Creds(matrix_server, bot_name, bot_pwd)
+ self.bot = botlib.Bot(creds)
+ self.prefix = bot_prefix
+ self.data = Data(db_file)
+
+ def run(self):
+ logger.debug('Running goal bot...')
+ self.bot.run()
+
+ async def handle_message(self, room, message):
+ match = botlib.MessageMatch(room, message, self.bot, self.prefix)
+
+ if not match.is_not_from_this_bot():
+ # Let's not talk to our-self
+ return
+
+ logger.debug(f'Message: {message}')
+ if not match.prefix():
+ logger.debug(f'{message} - did not match {self.prefix}')
+
projects / matrix-goal-bot.git / blobdiff