from plugin import BasePlugin import curses import common import timed_events class Plugin(BasePlugin): def init(self): self.add_command('remind', self.command_remind, "Usage: /remind <time in seconds> <todo>\nReminder: remind you of <todo> every <time> seconds..", self.completion_remind) self.add_command('done', self.command_done, "Usage: /done <id>\nDone: Stop reminding you do the task identified by <id>", self.completion_done) self.add_command('tasks', self.command_tasks, "Usage: /tasks\nTasks: List all the current tasks and their ids.", None) self.tasks = {} self.count = 0 for option in self.config.options(self.__module__): id, secs = option.split(',') id = int(id) if id > self.count: self.count = id value = self.config.get(option, '') self.tasks[id] = (int(secs), value) self.config.remove_section(self.__module__) self.config.add_section(self.__module__) if self.tasks: self.count += 1 self.command_tasks('', nocommand=True) def command_remind(self, arg): args = common.shell_split(arg) if len(args) < 2: return time = common.parse_str_to_secs(args[0]) if not time: return self.tasks[self.count] = (time, args[1]) timed_event = timed_events.DelayedEvent(time, self.remind, self.count) self.core.add_timed_event(timed_event) self.core.information('Task %s added: %s every %s.' % (self.count, args[1], common.parse_secs_to_str(time)), 'Info') self.count += 1 def completion_remind(self, the_input): txt = the_input.get_text() args = common.shell_split(txt) n = len(args) if txt.endswith(' '): n += 1 if n == 2: return the_input.auto_completion(["60", "5m", "15m", "30m", "1h", "10h", "1d"], '') def completion_done(self, the_input): return the_input.auto_completion(["%s" % key for key in self.tasks], '') def command_done(self, arg="0"): try: id = int(arg) except: return if not id in self.tasks: return self.core.information('Task %s: %s [DONE]' % (id, self.tasks[id][1]), 'Info') del self.tasks[id] def command_tasks(self, arg, nocommand=None): if nocommand: s = 'The following tasks were loaded:\n' else: s = 'The following tasks are active:\n' for key in self.tasks: s += 'Task %s: %s every %s.\n' % (key, repr(self.tasks[key][1]), common.parse_secs_to_str(self.tasks[key][0])) if s: self.core.information(s, 'Info') def remind(self, id=0): if not id in self.tasks: return self.core.information('Task %s: %s' % (id, self.tasks[id][1]), 'Info') if self.config.get('beep', '') == 'true': curses.beep() timed_event = timed_events.DelayedEvent(self.tasks[id][0], self.remind, id) self.core.add_timed_event(timed_event) def cleanup(self): if self.tasks: self.config.remove_section(self.__module__) self.config.add_section(self.__module__) for task in self.tasks: self.config.set('%s,%s' % (task, self.tasks[task][0]), self.tasks[task][1]) self.config.write()