aboutsummaryrefslogtreecommitdiff
path: root/modules/todo.py
blob: 71ebe133df8ba0f037fa9d11f827697f1d8eb3e8 (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
class Module(object):
    def __init__(self, bot):
        self.bot = bot
        bot.events.on("received").on("command").on("todo").hook(
            self.todo)

    def todo(self, event):
        if len(event["args_split"]) > 1:
            action = event["args_split"][0].lower()
            arg = " ".join(event["args_split"][1:])
            arg_lower = arg.lower()
            todo = event["user"].get_setting("todo", [])
            if action == "add":
                for item in todo:
                    if item.lower() == arg_lower:
                        event["stderr"].write(
                            "That is already in your todo")
                        return
                todo.append(arg)
                event["user"].set_setting("todo", list(todo))
                event["stdout"].write("Saved")
            elif action == "remove":
                if event["args_split"][1].isdigit():
                    index = int(event["args_split"][1])
                    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")
            elif action == "show":
                if event["args_split"][1].isdigit():
                    index = int(event["args_split"][1])
                    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 todo item number to show")
        elif len(event["args_split"]) == 1:
            event["stderr"].write("Please provided an action and an argument")
        else:
            todo_count = len(event["user"].get_setting("todo", []))
            event["stdout"].write("There are %d items in your todo" % todo_count)