36 lines
1.2 KiB
Python
36 lines
1.2 KiB
Python
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,
|
|
}
|