From f19389ec8c3777ad2073189c059a16e166f990e6 Mon Sep 17 00:00:00 2001 From: jesopo Date: Tue, 5 Apr 2016 14:26:46 +0100 Subject: [PATCH] added todo.py. --- modules/lastfm.py | 1 - modules/todo.py | 50 +++++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 50 insertions(+), 1 deletion(-) create mode 100644 modules/todo.py diff --git a/modules/lastfm.py b/modules/lastfm.py index e9ff681a..ed470704 100644 --- a/modules/lastfm.py +++ b/modules/lastfm.py @@ -40,7 +40,6 @@ class Module(object): "api_key": self.bot.config["lastfm-api-key"], "user": username, "format": "json"}, json=True) tags = [] - print(info_page.keys()) if "toptags" in info_page["track"]: for tag in info_page["track"]["toptags"]["tag"]: tags.append(tag["name"]) diff --git a/modules/todo.py b/modules/todo.py new file mode 100644 index 00000000..71ebe133 --- /dev/null +++ b/modules/todo.py @@ -0,0 +1,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)