From e4e2db2a19d3af9a11fb9746b141186470506818 Mon Sep 17 00:00:00 2001 From: Wesley Lima Date: Wed, 18 Oct 2023 14:09:21 -0400 Subject: [PATCH] fix: OPTIC-182: Ensuring Task view always has task attribute no matter the method called --- label_studio/tasks/api.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/label_studio/tasks/api.py b/label_studio/tasks/api.py index 066874738ea..0e88f15f27e 100644 --- a/label_studio/tasks/api.py +++ b/label_studio/tasks/api.py @@ -148,6 +148,10 @@ class TaskAPI(generics.RetrieveUpdateDestroyAPIView): DELETE=all_permissions.tasks_delete, ) + def initial(self, request, *args, **kwargs): + self.task = self.get_object() + return super().initial(request, *args, **kwargs) + @staticmethod def prefetch(queryset): return queryset.prefetch_related( @@ -176,8 +180,6 @@ def get_retrieve_serializer_context(self, request): } def get(self, request, pk): - self.task = self.get_object() - context = self.get_retrieve_serializer_context(request) context['project'] = project = self.task.project