poezio/plugins/reminder.py
mathieui 9885203c67 Update the plugins to use the PluginAPI
Also:
- Add get_conversation_messages() to PluginAPI
- Make plugins_autoload colon-separated instead of space-separated
    (for consistency)
- Replace a JID() with a safeJID() in the uptime plugin
2013-03-08 22:53:35 +01:00

103 lines
3.6 KiB
Python

from plugin import BasePlugin
import curses
import common
import timed_events
class Plugin(BasePlugin):
def init(self):
self.api.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.api.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.api.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.api.add_timed_event(timed_event)
self.api.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.api.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.api.information(s, 'Info')
def remind(self, id=0):
if not id in self.tasks:
return
self.api.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.api.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()