1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
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()
|
