Релизован стандартный CRUD для Колонок Доски

Author: Toksi86

kanban/serializers.py

@@ -45,3 +45,40 @@ def create(self, validated_data):
             if not board.columns.exists():
                 BoardColumn.objects.create(board=board, name="Бэклог", order=1)
         return board
+
+
+class BoardColumnSerializer(serializers.ModelSerializer):
+    board = serializers.PrimaryKeyRelatedField(queryset=Board.objects.all())
+    tasks_count = serializers.IntegerField(read_only=True)
+
+    class Meta:
+        model = BoardColumn
+        fields = (
+            "id",
+            "board",
+            "name",
+            "order",
+            "tasks_count",
+            "created_at",
+            "updated_at",
+        )
+        read_only_fields = ("id", "tasks_count", "created_at", "updated_at")
+
+    def validate_board(self, board: Board):
+        request = self.context.get("request")
+        user = getattr(request, "user", None)
+        if not user or not user.is_authenticated:
+            raise serializers.ValidationError("Требуется аутентификация")
+
+        if board.project.leader_id == user.id:
+            return board
+
+        is_member = Collaborator.objects.filter(
+            project_id=board.project_id,
+            user_id=user.id,
+        ).exists()
+        if not is_member:
+            raise serializers.ValidationError(
+                "Пользователь не является участником проекта"
+            )
+        return board

kanban/tests/test_column_api.py

@@ -0,0 +1,52 @@
+from django.urls import reverse
+from rest_framework import status
+from rest_framework.test import APIClient
+
+from kanban.models import Board, BoardColumn
+from kanban.tests.base import BaseKanbanTestCase
+from projects.models import Project
+from users.models import CustomUser
+
+
+class ColumnAPITests(BaseKanbanTestCase):
+    def setUp(self):
+        super().setUp()
+        self.client = APIClient()
+        self.client.force_authenticate(user=self.user)
+        self.url = reverse("kanban:kanban-column-list")
+
+    def test_list_columns_returns_board_columns(self):
+        response = self.client.get(self.url)
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        self.assertGreaterEqual(len(response.data), 1)
+        self.assertEqual(response.data[0]["board"], self.board.id)
+
+    def test_create_column(self):
+        payload = {"board": self.board.id, "name": "In Progress", "order": 5}
+        response = self.client.post(self.url, payload, format="json")
+        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
+        column_id = response.data["id"]
+        column = BoardColumn.objects.get(id=column_id)
+        self.assertEqual(column.name, "In Progress")
+
+    def test_cannot_create_column_in_foreign_project(self):
+        foreign_leader = CustomUser.objects.create(
+            email="foreign@example.com",
+            first_name="Foreign",
+            last_name="Leader",
+            password="pass1234",
+            birthday=self.user.birthday,
+        )
+        foreign_project = Project.objects.create(name="Foreign", leader=foreign_leader)
+        foreign_board = Board.objects.create(
+            project=foreign_project, name="Foreign Board"
+        )
+        payload = {"board": foreign_board.id, "name": "Forbidden", "order": 1}
+        response = self.client.post(self.url, payload, format="json")
+        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
+
+    def test_cannot_delete_last_column(self):
+        self.board.columns.exclude(pk=self.column.pk).delete()
+        col_url = reverse("kanban:kanban-column-detail", args=[self.column.id])
+        response = self.client.delete(col_url)
+        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)

kanban/urls.py

@@ -1,11 +1,12 @@
 from django.urls import include, path
 from rest_framework.routers import DefaultRouter
 
-from kanban.views import BoardViewSet
+from kanban.views import BoardViewSet, BoardColumnViewSet
 
 app_name = "kanban"
 
 router = DefaultRouter()
 router.register(r"boards", BoardViewSet, basename="kanban-board")
+router.register(r"columns", BoardColumnViewSet, basename="kanban-column")
 
 urlpatterns = [path("", include(router.urls))]

kanban/views.py

@@ -1,9 +1,11 @@
 from django.db.models import Q
 from drf_yasg.utils import swagger_auto_schema
 from rest_framework import permissions, viewsets
+from rest_framework.response import Response
+from rest_framework import status
 
-from kanban.models import Board
-from kanban.serializers import BoardSerializer
+from kanban.models import Board, BoardColumn
+from kanban.serializers import BoardSerializer, BoardColumnSerializer
 from projects.models import Collaborator
 
 
@@ -52,3 +54,53 @@ def update(self, request, *args, **kwargs):
     @swagger_auto_schema(tags=["Kanban Boards"])
     def destroy(self, request, *args, **kwargs):
         return super().destroy(request, *args, **kwargs)
+
+
+class BoardColumnViewSet(viewsets.ModelViewSet):
+    serializer_class = BoardColumnSerializer
+    permission_classes = [permissions.IsAuthenticated]
+
+    def get_queryset(self):
+        user = self.request.user
+        collaborator_projects = Collaborator.objects.filter(user=user).values(
+            "project_id"
+        )
+        return (
+            BoardColumn.objects.select_related("board", "board__project")
+            .filter(
+                Q(board__project__leader_id=user.id)
+                | Q(board__project_id__in=collaborator_projects)
+            )
+            .distinct()
+        )
+
+    @swagger_auto_schema(tags=["Kanban Columns"])
+    def list(self, request, *args, **kwargs):
+        return super().list(request, *args, **kwargs)
+
+    @swagger_auto_schema(tags=["Kanban Columns"])
+    def create(self, request, *args, **kwargs):
+        return super().create(request, *args, **kwargs)
+
+    @swagger_auto_schema(tags=["Kanban Columns"])
+    def retrieve(self, request, *args, **kwargs):
+        return super().retrieve(request, *args, **kwargs)
+
+    @swagger_auto_schema(tags=["Kanban Columns"])
+    def partial_update(self, request, *args, **kwargs):
+        return super().partial_update(request, *args, **kwargs)
+
+    @swagger_auto_schema(tags=["Kanban Columns"])
+    def update(self, request, *args, **kwargs):
+        return super().update(request, *args, **kwargs)
+
+    @swagger_auto_schema(tags=["Kanban Columns"])
+    def destroy(self, request, *args, **kwargs):
+        try:
+            return super().destroy(request, *args, **kwargs)
+        except Exception as exc:
+            from django.core.exceptions import ValidationError
+
+            if isinstance(exc, ValidationError):
+                return Response(exc.messages, status=status.HTTP_400_BAD_REQUEST)
+            raise