Added ProjectNewsDetail view

Author: yakser

projects/permissions.py

@@ -1,19 +1,23 @@
 from rest_framework.permissions import BasePermission, SAFE_METHODS
 
+from projects.models import Project
+
 
 class IsProjectLeaderOrReadOnlyForNonDrafts(BasePermission):
     """
     Allows access to update only to project leader.
     """
 
     def has_permission(self, request, view) -> bool:
+        # fixme:
         if request.method in SAFE_METHODS or (request.user and request.user.id):
             return True
         return False
 
     def has_object_permission(self, request, view, obj):
         if (request.method in SAFE_METHODS and not obj.draft) or (
             obj.leader == request.user
+            # fixme:
             or obj.collaborator_set.filter(user=request.user).exists()
             or obj.invite_set.filter(user=request.user).exists()
         ):
@@ -61,3 +65,25 @@ def has_object_permission(self, request, view, obj):
             ):
                 return True
             return False
+
+
+class IsNewsAuthorIsProjectLeader(BasePermission):
+    """
+    Allows access to update project news only to leader.
+    """
+
+    def has_permission(self, request, view) -> bool:
+        try:
+            project = Project.objects.get(pk=view.kwargs["project_pk"])
+            if request.method in SAFE_METHODS or (request.user == project.leader):
+                return True
+        except Project.DoesNotExist:
+            pass
+        return False
+
+    def has_object_permission(self, request, view, obj):
+        if (
+            request.method in SAFE_METHODS and not obj.project.draft
+        ) or obj.project.leader == request.user:
+            return True
+        return False

projects/serializers.py

@@ -238,3 +238,34 @@ class Meta:
             "views_count",
             "likes_count",
         ]
+
+
+class ProjectNewsDetailSerializer(serializers.ModelSerializer):
+    likes_count = serializers.SerializerMethodField()
+    project_name = serializers.SerializerMethodField()
+    project_image_address = serializers.SerializerMethodField()
+
+    def get_project_name(self, obj):
+        return obj.project.name
+
+    def get_project_image_address(self, obj):
+        return obj.project.image_address
+
+    def get_likes_count(self, obj):
+        obj_type = ContentType.objects.get_for_model(obj)
+        return User.objects.filter(
+            likes__content_type=obj_type, likes__object_id=obj.id
+        ).count()
+
+    class Meta:
+        model = ProjectNews
+        fields = [
+            "id",
+            "project_name",
+            "project_image_address",
+            "text",
+            "datetime_created",
+            "datetime_updated",
+            "views_count",
+            "likes_count",
+        ]

projects/urls.py

@@ -12,6 +12,7 @@
     ProjectRecommendedUsers,
     SetLikeOnProject,
     ProjectNewsList,
+    ProjectNewsDetail,
 )
 
 app_name = "projects"
@@ -20,6 +21,7 @@
     path("", ProjectList.as_view()),
     path("<int:pk>/like/", SetLikeOnProject.as_view()),
     path("<int:pk>/news/", ProjectNewsList.as_view()),
+    path("<int:project_pk>/news/<int:pk>/", ProjectNewsDetail.as_view()),
     path("<int:pk>/collaborators/", ProjectCollaborators.as_view()),
     path("<int:pk>/", ProjectDetail.as_view()),
     path("<int:pk>/recommended_users", ProjectRecommendedUsers.as_view()),

projects/views.py

@@ -15,6 +15,7 @@
     IsProjectLeaderOrReadOnlyForNonDrafts,
     HasInvolvementInProjectOrReadOnly,
     IsProjectLeader,
+    IsNewsAuthorIsProjectLeader,
 )
 from projects.serializers import (
     ProjectDetailSerializer,
@@ -23,6 +24,7 @@
     AchievementDetailSerializer,
     ProjectCollaboratorSerializer,
     ProjectNewsListSerializer,
+    ProjectNewsDetailSerializer,
 )
 from users.models import LikesOnProject
 from users.serializers import UserListSerializer
@@ -227,10 +229,23 @@ def get(self, request, pk):
 
 
 class ProjectNewsList(generics.ListCreateAPIView):
-    queryset = ProjectNews.objects.all()
     serializer_class = ProjectNewsListSerializer
-    permission_classes = [IsProjectLeaderOrReadOnlyForNonDrafts]
+    permission_classes = [IsNewsAuthorIsProjectLeader]
 
     def perform_create(self, serializer):
         project = Project.objects.get(pk=self.kwargs["pk"])
         serializer.save(project=project)
+
+    def get_queryset(self):
+        project = Project.objects.get(pk=self.kwargs["pk"])
+        return ProjectNews.objects.filter(project=project)
+
+
+class ProjectNewsDetail(generics.RetrieveUpdateDestroyAPIView):
+    queryset = ProjectNews.objects.all()
+    serializer_class = ProjectNewsDetailSerializer
+    permission_classes = [IsNewsAuthorIsProjectLeader]
+
+    def get_queryset(self):
+        project = Project.objects.get(pk=self.kwargs["project_pk"])
+        return ProjectNews.objects.filter(project=project).all()