view and model notif and news

backend/account/serializers.py

@@ -32,3 +32,32 @@ class PushSubscriptionSerializer(serializers.ModelSerializer):
     class Meta:
         model = PushSubscription
         fields = ('endpoint', 'keys')
+
+
+class NewsSerializer(serializers.ModelSerializer):
+    notif_type = serializers.SerializerMethodField()
+
+    class Meta:
+        model = NewsModel
+        fields = ['title', 'content', 'image', 'created_at', 'notif_type']
+
+    def get_notif_type(self, obj):
+        return 'NEWS'
+
+class UserNotifSerializer(serializers.ModelSerializer):
+    notif_type = serializers.SerializerMethodField()
+
+    class Meta:
+        model = UserNotificationModel
+        fields = ['title', 'content', 'image', 'created_at', 'notif_type', 'is_read']
+
+    def get_notif_type(self, obj):
+        return 'USER_NOTIF'
+    
+
+class UnifiedNotifSerializer(serializers.Serializer):
+    def to_representation(self, instance):
+        if isinstance(instance, NewsModel):
+            return NewsSerializer(instance).data
+        elif isinstance(instance, UserNotificationModel):
+            return UserNotifSerializer(instance).data