109 lines
3.8 KiB
Python
109 lines
3.8 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: /reminder <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
|
|
if args[0].endswith('d'):
|
|
modifier = 'd'
|
|
elif args[0].endswith('h'):
|
|
modifier = 'h'
|
|
elif args[0].endswith('m'):
|
|
modifier = 'm'
|
|
else:
|
|
modifier = None
|
|
try:
|
|
if modifier:
|
|
time = int(args[0][:-1])
|
|
else:
|
|
time = int(args[0])
|
|
except:
|
|
return
|
|
|
|
if modifier == 'd':
|
|
time = time * 86400
|
|
elif modifier == 'h':
|
|
time = time * 3600
|
|
elif modifier == 'm':
|
|
time = time * 60
|
|
|
|
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 seconds.' % (self.count, args[1], 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 seconds.\n' % (key, repr(self.tasks[key][1]), 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()
|