From 81524236222e213e0dad7d2dcd5a17a676b7299f Mon Sep 17 00:00:00 2001 From: Karl Hobley Date: Thu, 6 Feb 2020 13:54:24 +0000 Subject: [PATCH] Allow task state class to be specified as an attribute --- wagtail/core/models.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/wagtail/core/models.py b/wagtail/core/models.py index 9926d62c6d..cb2e9e41ff 100644 --- a/wagtail/core/models.py +++ b/wagtail/core/models.py @@ -2633,8 +2633,14 @@ class Task(models.Model): else: return content_type.get_object_for_this_type(id=self.id) + task_state_class = None + + @classmethod + def get_task_state_class(self): + return self.task_state_class or TaskState + def start(self, workflow_state, user=None): - task_state = TaskState(workflow_state=workflow_state) + task_state = self.get_task_state_class()(workflow_state=workflow_state) task_state.status = TaskState.STATUS_IN_PROGRESS task_state.page_revision = workflow_state.page.get_latest_revision() task_state.task = self