Merge branch 'dev' into feature/addotional_project_fields

Author: Toksi86

invites/serializers.py

@@ -1,6 +1,8 @@
+from django.apps import apps
 from rest_framework import serializers
 
 from invites.models import Invite
+from projects.models import Collaborator
 from projects.serializers import ProjectListSerializer
 from users.serializers import UserDetailSerializer
 
@@ -18,6 +20,48 @@ class Meta:
             "is_accepted",
         ]
 
+    def validate(self, attrs):
+        project = attrs["project"]
+        user = attrs["user"]
+
+        if project.leader_id == user.id:
+            raise serializers.ValidationError(
+                {"user": "Пользователь уже является лидером проекта."}
+            )
+
+        if Collaborator.objects.filter(project=project, user=user).exists():
+            raise serializers.ValidationError(
+                {"user": "Пользователь уже состоит в проекте."}
+            )
+
+        if Invite.objects.filter(
+            project=project, user=user, is_accepted__isnull=True
+        ).exists():
+            raise serializers.ValidationError(
+                {"user": "У пользователя уже есть активное приглашение в этот проект."}
+            )
+
+        link = project.program_links.select_related("partner_program").first()
+        if link:
+            PartnerProgramUserProfile = apps.get_model(
+                "partner_programs", "PartnerProgramUserProfile"
+            )
+            is_participant = PartnerProgramUserProfile.objects.filter(
+                user_id=user.id,
+                partner_program_id=link.partner_program_id,
+            ).exists()
+            if not is_participant:
+                raise serializers.ValidationError(
+                    {
+                        "user": (
+                            "Нельзя пригласить пользователя: проект относится к программе, "
+                            "а пользователь не является её участником."
+                        )
+                    }
+                )
+
+        return attrs
+
 
 class InviteDetailSerializer(serializers.ModelSerializer[Invite]):
     user = UserDetailSerializer(many=False, read_only=True)

projects/models.py

@@ -1,7 +1,9 @@
 from typing import Optional
 
+from django.apps import apps
 from django.contrib.auth import get_user_model
 from django.contrib.contenttypes.fields import GenericRelation
+from django.core.exceptions import ValidationError
 from django.core.validators import (
     MaxLengthValidator,
     MaxValueValidator,
@@ -310,6 +312,35 @@ class Meta:
             )
         ]
 
+    def clean(self):
+        """
+        Если проект привязан к программе, добавлять коллаборатора можно
+        только если пользователь — участник этой программы.
+        (Проект привязан максимум к одной программе.)
+        """
+        link = self.project.program_links.select_related("partner_program").first()
+        if not link:
+            return
+
+        PartnerProgramUserProfile = apps.get_model(
+            "partner_programs",
+            "PartnerProgramUserProfile",
+        )
+
+        is_participant = PartnerProgramUserProfile.objects.filter(
+            user_id=self.user_id,
+            partner_program_id=link.partner_program_id,
+        ).exists()
+
+        if not is_participant:
+            raise ValidationError(
+                "Пользователь не является участником программы, к которой относится проект."
+            )
+
+    def save(self, *args, **kwargs):
+        self.full_clean()
+        return super().save(*args, **kwargs)
+
 
 class ProjectNews(models.Model):
     """

projects/permissions.py

@@ -1,10 +1,10 @@
 from datetime import datetime, timedelta
 
 from django.utils import timezone
-from rest_framework.exceptions import PermissionDenied
+from rest_framework.exceptions import PermissionDenied, ValidationError
 from rest_framework.permissions import SAFE_METHODS, BasePermission
 
-from partner_programs.models import PartnerProgramUserProfile
+from partner_programs.models import PartnerProgram, PartnerProgramUserProfile
 from projects.models import Project
 
 
@@ -189,3 +189,24 @@ def has_object_permission(self, request, view, obj):
             and request.user.is_authenticated
             and obj.project.leader_id == request.user.id
         )
+
+
+class CanBindProjectToProgram(BasePermission):
+    message = "Привязать проект к программе может только её участник (или менеджер)."
+
+    def has_permission(self, request, view):
+        program_id = (request.data or {}).get("partner_program_id")
+        if not program_id:
+            return True
+
+        try:
+            program = PartnerProgram.objects.get(pk=program_id)
+        except PartnerProgram.DoesNotExist:
+            raise ValidationError({"partner_program_id": "Программа не найдена."})
+
+        if program.is_manager(request.user):
+            return True
+
+        return PartnerProgramUserProfile.objects.filter(
+            user=request.user, partner_program=program
+        ).exists()

projects/serializers.py

@@ -412,7 +412,7 @@ def validate(self, data):
 
         if project.leader != request.user:
             raise serializers.ValidationError(
-                "Только лидер проекта может дублировать его в программу."
+                {"error": "Только лидер проекта может дублировать его в программу."}
             )
 
         try:

projects/views.py

@@ -33,6 +33,7 @@
 from projects.models import Achievement, Collaborator, Project, ProjectGoal, ProjectNews
 from projects.pagination import ProjectNewsPagination, ProjectsPagination
 from projects.permissions import (
+    CanBindProjectToProgram,
     HasInvolvementInProjectOrReadOnly,
     IsNewsAuthorIsProjectLeaderOrReadOnly,
     IsProjectLeader,
@@ -661,7 +662,7 @@ def patch(self, request, project_pk: int, user_to_leader_pk: int) -> Response:
 
 
 class DuplicateProjectView(APIView):
-    permission_classes = [IsAuthenticated]
+    permission_classes = [IsAuthenticated, CanBindProjectToProgram]
 
     @swagger_auto_schema(
         request_body=ProjectDuplicateRequestSerializer,