From e0c4e31600d9c8f354d86830a6cf6a51c968d77d Mon Sep 17 00:00:00 2001 From: iggy <i.kepka@student.utwente.nl> Date: Thu, 27 Jul 2023 10:57:34 +0200 Subject: [PATCH] Added method get_object in class viewsets/event.py --- apps/calendar/viewsets/event.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/apps/calendar/viewsets/event.py b/apps/calendar/viewsets/event.py index da73293..e637bce 100644 --- a/apps/calendar/viewsets/event.py +++ b/apps/calendar/viewsets/event.py @@ -1,3 +1,4 @@ +from django.shortcuts import get_object_or_404 from django.utils import timezone from guardian.shortcuts import get_objects_for_user from guardian.shortcuts import Q @@ -59,6 +60,12 @@ class EventViewSet(viewsets.ModelViewSet): def get_serializer_class(self): return EventSerializer + def get_object(self): + obj = get_object_or_404(self.get_queryset(), pk=self.kwargs["id"]) # maybe instead of id, name ? + print("method for retrieving the object was used ! [TEST]") + self.check_object_permissions(self.request, obj) + return obj + def get_queryset(self): associations = get_objects_for_user(self.request.user, 'board', klass=Association) queryset = Event.objects.filter( -- GitLab