Skip to content

Commit

Permalink
OpenConceptLab/ocl_issues#1814 | fixing task get view
Browse files Browse the repository at this point in the history
  • Loading branch information
snyaggarwal committed May 3, 2024
1 parent 2b4a666 commit c8cd7ac
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion core/tasks/views.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
from drf_yasg.utils import swagger_auto_schema
from pydash import compact
from rest_framework.generics import RetrieveAPIView, DestroyAPIView
from rest_framework.generics import RetrieveAPIView, DestroyAPIView, get_object_or_404
from rest_framework.permissions import IsAuthenticated, IsAdminUser

from core.common.exceptions import Http400
Expand Down Expand Up @@ -51,6 +51,12 @@ class TaskView(AbstractTaskView, DestroyAPIView):
lookup_url_kwarg = 'task_id'
pk_field = 'id'

def get_object(self, queryset=None):
queryset = self.get_queryset()
obj = get_object_or_404(queryset, **{self.lookup_field: self.kwargs[self.lookup_url_kwarg]})
self.check_object_permissions(self.request, obj)
return obj

def perform_destroy(self, instance):
if not instance.has_access(self.request.user):
self.permission_denied(self.request)
Expand Down

0 comments on commit c8cd7ac

Please sign in to comment.