aboutsummaryrefslogtreecommitdiff
path: root/modules/todo.py
blob: 3e0e939168b615f8293a2f852c356a7948b9fc86 (about) (plain) (blame)
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
from src import ModuleManager, utils

class Module(ModuleManager.BaseModule):
    @utils.hook("received.command.todo")
    def todo(self, event):
        """
        :help: Find out what's in your todo list
        :usage: [item number]
        """
        todo = event["user"].get_setting("todo", [])
        if event["args"]:
            if event["args_split"][0].isdigit() and int(event["args_split"][0]) > 0:
                index = int(event["args_split"][0])
                if len(todo) >= index:
                    event["stdout"].write("Todo %d: %s" % (index, todo[index-1]))
                else:
                    event["stderr"].write("You do not have that many things in your todo")
            else:
                event["stderr"].write("Please provide a number")
        else:
            todo_count = len(todo)
            event["stdout"].write("There are %d items in your todo" % todo_count)

    @utils.hook("received.command.todoadd", min_args=1)
    def todo_add(self, event):
        """
        :help: Add something to your todo list
        :usage: <description>
        """
        arg_lower = event["args"].lower()
        todo = event["user"].get_setting("todo", [])
        for item in todo:
            if item.lower() == arg_lower:
                event["stderr"].write(
                   "That is already in your todo")
                return
        todo.append(event["args"])
        event["user"].set_setting("todo", todo)
        event["stdout"].write("Saved")

    @utils.hook("received.command.tododel", min_args=1)
    def todo_del(self, event):
        """
        :help: Remove something from your todo list
        :usage: <item number>
        """
        todo = event["user"].get_setting("todo", [])
        if event["args_split"][0].isdigit() and int(event["args_split"][0]) > 0:
            index = int(event["args_split"][0])
            if len(todo) >= index:
                todo.pop(index-1)
                event["user"].set_setting("todo", todo)
                event["stdout"].write("Todo item removed")
            else:
                event["stderr"].write("You do not have that many things in "
                    "your todo")
        else:
            event["stderr"].write("Please provided a todo item number to remove")