From 3fbc3b1fcb76f44e59cad718379ee75826ae515c Mon Sep 17 00:00:00 2001 From: Conatum Date: Wed, 11 May 2022 22:04:43 +0300 Subject: [PATCH] (tasklist): Pass context to factory. --- bot/modules/todo/Tasklist.py | 2 +- bot/modules/todo/commands.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) 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),