diff --git a/bot/modules/todo/Tasklist.py b/bot/modules/todo/Tasklist.py index 74df8d64..dc8737a5 100644 --- a/bot/modules/todo/Tasklist.py +++ b/bot/modules/todo/Tasklist.py @@ -121,7 +121,7 @@ class Tasklist: self.active[(member.id, channel.id)] = self @classmethod - def fetch_or_create(cls, member, channel): + def fetch_or_create(cls, ctx, flags, member, channel): tasklist = cls.active.get((member.id, channel.id), None) return tasklist if tasklist is not None else cls(member, channel) diff --git a/bot/modules/todo/commands.py b/bot/modules/todo/commands.py index 9932c6d4..78ceba5c 100644 --- a/bot/modules/todo/commands.py +++ b/bot/modules/todo/commands.py @@ -11,7 +11,7 @@ from .Tasklist import Tasklist name="todo", desc="Display and edit your personal To-Do list.", group="Productivity", - flags=('add==', 'delete==', 'check==', 'uncheck==', 'edit==') + flags=('add==', 'delete==', 'check==', 'uncheck==', 'edit==', 'text') ) @in_guild() async def cmd_todo(ctx, flags): @@ -69,7 +69,7 @@ async def cmd_todo(ctx, flags): return # TODO: Custom module, with pre-command hooks - tasklist = Tasklist.fetch_or_create(ctx.author, ctx.ch) + tasklist = Tasklist.fetch_or_create(ctx, flags, ctx.author, ctx.ch) keys = { 'add': (('add', ), tasklist.parse_add),