From 2503a904c8555fdec691b0194083649ab7d8f7bf Mon Sep 17 00:00:00 2001 From: Setyo Nugroho Date: Fri, 8 Jul 2022 14:51:08 +0700 Subject: [PATCH] read or unread notification --- api/views.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/api/views.py b/api/views.py index abe2733..a02630c 100644 --- a/api/views.py +++ b/api/views.py @@ -389,3 +389,21 @@ class NotificationViewSet(viewsets.ModelViewSet): serializer = NotificationSerializer(notification) return Response(serializer.data) + + @action(detail=True, methods=['GET']) + def set_read(self, request, pk): + notification = Notification.objects.filter(id=pk).first() + notification.is_read = True + + serializer = NotificationSerializer(notification) + + return Response(serializer.data) + + @action(detail=True, methods=['GET']) + def set_unread(self, request, pk): + notification = Notification.objects.filter(id=pk).first() + notification.is_read = False + + serializer = NotificationSerializer(notification) + + return Response(serializer.data)