notifications API

notifications/views.py

@@ -3,18 +3,17 @@ notifications views file
 """
 # django imports
 from django.db.models import Q
-from rest_framework import viewsets
 from rest_framework.decorators import action
 from rest_framework.permissions import IsAuthenticated
 from rest_framework.response import Response
-
+from rest_framework import viewsets, status, views
 # local imports
-from account.utils import custom_response
-from base.messages import SUCCESS_CODE
+from account.utils import custom_response, custom_error_response
+from base.messages import SUCCESS_CODE, ERROR_CODE
 from notifications.constants import TEST_NOTIFICATION
-from notifications.serializers import RegisterDevice
+from notifications.serializers import RegisterDevice, NotificationListSerailizer, ReadNotificationSerializer
 from notifications.utils import send_notification
-
+from notifications.models import Notification
 
 class NotificationViewSet(viewsets.GenericViewSet):
     """ used to do the notification actions """
@@ -40,3 +39,38 @@ class NotificationViewSet(viewsets.GenericViewSet):
         """
         send_notification.delay(TEST_NOTIFICATION, None, request.auth.payload['user_id'], {})
         return custom_response(SUCCESS_CODE["3000"])
+
+    @action(methods=['get'], detail=False, url_path='list', url_name='list',
+            serializer_class=NotificationListSerailizer)
+    def notification_list(self, request):
+        """
+        notification list
+        """
+        try:
+            queryset = Notification.objects.filter(notification_to=request.user)
+            serializer = NotificationListSerailizer(queryset, many=True)
+            return custom_response(None, serializer.data, response_status=status.HTTP_200_OK)
+        except Exception as e:
+            return custom_error_response(str(e), response_status=status.HTTP_400_BAD_REQUEST)
+
+
+class ReadNotification(views.APIView):
+    """Update notification API"""
+    serializer_class = ReadNotificationSerializer
+    model = Notification
+    permission_classes = [IsAuthenticated]
+
+    def put(self, request, format=None):
+        try:
+            notification_id = str(self.request.data.get('notification_id'))
+            notification_queryset = Notification.objects.filter(id=notification_id, notification_to=self.request.user).last()
+            if notification_queryset:
+                # use ReadNotificationSerializer serializer
+                serializer = ReadNotificationSerializer(notification_queryset, data=request.data, partial=True)
+                if serializer.is_valid():
+                    # save serializer
+                    serializer.save()
+                    return custom_response(SUCCESS_CODE['3037'], response_status=status.HTTP_200_OK)
+                return custom_error_response(serializer.errors, response_status=status.HTTP_400_BAD_REQUEST)
+        except Exception as e:
+            return custom_error_response(str(e), response_status=status.HTTP_400_BAD_REQUEST)