Added is_user_liked field for ProjectNews serializers

Author: yakser

projects/serializers.py

@@ -2,7 +2,7 @@
 from rest_framework import serializers
 
 from core.fields import CustomListField
-from core.services import get_views_count, get_likes_count
+from core.services import get_views_count, get_likes_count, is_fan
 from industries.models import Industry
 from projects.models import Project, Achievement, Collaborator, ProjectNews
 from projects.validators import validate_project
@@ -215,6 +215,7 @@ class ProjectNewsListSerializer(serializers.ModelSerializer):
     likes_count = serializers.SerializerMethodField()
     project_name = serializers.SerializerMethodField()
     project_image_address = serializers.SerializerMethodField()
+    is_user_liked = serializers.SerializerMethodField()
 
     def get_project_name(self, obj):
         return obj.project.name
@@ -228,6 +229,12 @@ def get_views_count(self, obj):
     def get_likes_count(self, obj):
         return get_likes_count(obj)
 
+    def get_is_user_liked(self, obj):
+        user = self.context.get("user")
+        if user:
+            return is_fan(obj, user)
+        return False
+
     class Meta:
         model = ProjectNews
         fields = [
@@ -238,6 +245,7 @@ class Meta:
             "datetime_created",
             "views_count",
             "likes_count",
+            "is_user_liked",
             "files",
         ]
 
@@ -247,6 +255,7 @@ class ProjectNewsDetailSerializer(serializers.ModelSerializer):
     likes_count = serializers.SerializerMethodField()
     project_name = serializers.SerializerMethodField()
     project_image_address = serializers.SerializerMethodField()
+    is_user_liked = serializers.SerializerMethodField()
 
     def get_project_name(self, obj):
         return obj.project.name
@@ -260,6 +269,12 @@ def get_views_count(self, obj):
     def get_likes_count(self, obj):
         return get_likes_count(obj)
 
+    def get_is_user_liked(self, obj):
+        user = self.context.get("user")
+        if user:
+            return is_fan(obj, user)
+        return False
+
     class Meta:
         model = ProjectNews
         fields = [
@@ -271,5 +286,6 @@ class Meta:
             "datetime_updated",
             "views_count",
             "likes_count",
+            "is_user_liked",
             "files",
         ]

projects/views.py

@@ -244,6 +244,12 @@ def get_queryset(self):
         project = Project.objects.get(pk=self.kwargs.get("project_pk"))
         return ProjectNews.objects.filter(project=project)
 
+    def get(self, request, *args, **kwargs):
+        news = self.paginate_queryset(self.get_queryset())
+        context = {"user": request.user}
+        serializer = ProjectNewsListSerializer(news, context=context, many=True)
+        return self.get_paginated_response(serializer.data)
+
 
 class ProjectNewsDetail(generics.RetrieveUpdateDestroyAPIView):
     queryset = ProjectNews.objects.all()
@@ -257,6 +263,14 @@ def get_queryset(self):
         except Project.DoesNotExist:
             return []
 
+    def get(self, request, *args, **kwargs):
+        try:
+            news = self.get_queryset().get(pk=self.kwargs["pk"])
+            context = {"user": request.user}
+            return Response(ProjectNewsDetailSerializer(news, context=context).data)
+        except ProjectNews.DoesNotExist:
+            return Response(status=status.HTTP_404_NOT_FOUND)
+
 
 class ProjectNewsDetailSetViewed(generics.CreateAPIView):
     queryset = ProjectNews.objects.all()