initial commit

core/apps/companies/views/folders.py

@@ -0,0 +1,35 @@
+from django_core.mixins import BaseViewSetMixin
+from drf_spectacular.utils import extend_schema
+from rest_framework.permissions import AllowAny, IsAdminUser
+from rest_framework.viewsets import ModelViewSet
+
+from core.apps.companies.models import CompanyFolderModel
+from core.apps.companies.serializers.folders import (
+    CreateCompanyFolderSerializer,
+    ListCompanyFolderSerializer,
+    RetrieveCompanyFolderSerializer,
+    UpdateCompanyFolderSerializer,
+    DestroyCompanyFolderSerializer
+)
+
+
+@extend_schema(tags=["CompanyFolder"])
+class CompanyFolderView(BaseViewSetMixin, ModelViewSet):
+    queryset = CompanyFolderModel.objects.all()
+    serializer_class = ListCompanyFolderSerializer
+    permission_classes = [AllowAny]
+
+    action_permission_classes = {
+        "list": [IsAdminUser],
+        "retrieve": [IsAdminUser],
+        "create": [IsAdminUser],
+        "update": [IsAdminUser],
+        "destroy": [IsAdminUser],
+    }
+    action_serializer_class = {
+        "list": ListCompanyFolderSerializer,
+        "retrieve": RetrieveCompanyFolderSerializer,
+        "create": CreateCompanyFolderSerializer,
+        "update": UpdateCompanyFolderSerializer,
+        "destroy": DestroyCompanyFolderSerializer,
+    }