bitbot-3.11-fork/modules/todo.py

52 lines
2.2 KiB
Python

class Module(object):
def __init__(self, bot, events, exports):
self.bot = bot
events.on("received.command.todo").hook(self.todo,
help="Find out what's in your todo list", usage="[item number]")
events.on("received.command.todoadd").hook(self.todo_add, min_args=1,
help="Add something to your todo list", usage="<description>")
events.on("received.command.tododel").hook(self.todo_del, min_args=1,
help="Remove something from your todo list", usage="<item number>")
def todo(self, event):
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)
def todo_add(self, event):
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")
def todo_del(self, event):
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")