103 lines
3.6 KiB
Python
103 lines
3.6 KiB
Python
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='<seconds> <todo>',
|
|
help='Remind you of <todo> every <time> seconds.',
|
|
short='Remind you of a task',
|
|
completion=self.completion_remind)
|
|
self.add_command('done', self.command_done,
|
|
usage='<id>',
|
|
help='Stop reminding you do the task identified by <id>.',
|
|
short='Remove a task',
|
|
completion=self.completion_done)
|
|
self.add_command('tasks', self.command_tasks,
|
|
usage='',
|
|
help='List all the current tasks and their ids.',
|
|
short='List current tasks')
|
|
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()
|