78 lines
3.0 KiB
Python
78 lines
3.0 KiB
Python
from typing import cast
|
|
|
|
from django_core.mixins import BaseViewSetMixin # type: ignore
|
|
from django.db import transaction
|
|
from drf_spectacular.utils import extend_schema
|
|
from rest_framework.decorators import action # type: ignore
|
|
from rest_framework.permissions import AllowAny, IsAdminUser # type: ignore
|
|
from rest_framework.viewsets import ModelViewSet # type: ignore
|
|
from rest_framework.generics import GenericAPIView # type: ignore
|
|
from rest_framework.request import HttpRequest # type: ignore
|
|
from rest_framework.response import Response # type: ignore
|
|
from rest_framework import status # type: ignore
|
|
from core.apps.companies.permissions.folders import IsFolderOwner
|
|
|
|
from core.utils.views import BaseApiViewMixin
|
|
from core.apps.contracts.serializers.contracts import CreateContractSerializer
|
|
from core.apps.companies.models import CompanyFolderModel
|
|
from core.apps.companies.serializers.folders import (
|
|
CreateCompanyFolderSerializer,
|
|
ListCompanyFolderSerializer,
|
|
RetrieveCompanyFolderSerializer,
|
|
UpdateCompanyFolderSerializer,
|
|
DestroyCompanyFolderSerializer
|
|
)
|
|
|
|
|
|
######################################################################
|
|
# Crud
|
|
######################################################################
|
|
@extend_schema(tags=["CompanyFolder"])
|
|
class CompanyFolderCrudViewSet(BaseViewSetMixin, ModelViewSet):
|
|
queryset = CompanyFolderModel.objects.all()
|
|
permission_classes = [AllowAny]
|
|
|
|
action_permission_classes = { # type: ignore
|
|
"list": [IsAdminUser],
|
|
"retrieve": [IsAdminUser],
|
|
"create": [IsAdminUser],
|
|
"update": [IsAdminUser],
|
|
"destroy": [IsAdminUser],
|
|
}
|
|
action_serializer_class = { # type: ignore
|
|
"list": ListCompanyFolderSerializer,
|
|
"retrieve": RetrieveCompanyFolderSerializer,
|
|
"create": CreateCompanyFolderSerializer,
|
|
"update": UpdateCompanyFolderSerializer,
|
|
"destroy": DestroyCompanyFolderSerializer,
|
|
}
|
|
|
|
|
|
######################################################################
|
|
# /company-folders/<uuid:pk>/contracts
|
|
######################################################################
|
|
class ContractFolderApiView(BaseApiViewMixin, GenericAPIView): # type: ignore
|
|
queryset = CompanyFolderModel.objects.all()
|
|
permission_classes = [IsFolderOwner]
|
|
serializer_class = CreateContractSerializer
|
|
|
|
method_permission_classes = {
|
|
"post": [IsFolderOwner]
|
|
}
|
|
method_serializer_class = {
|
|
"post": CreateContractSerializer
|
|
}
|
|
|
|
@extend_schema(
|
|
summary="Create Contract For Folder",
|
|
description="Create Contract For Folder",
|
|
)
|
|
def post(self, request: HttpRequest, *args: object, **kwargs: object) -> Response:
|
|
with transaction.atomic():
|
|
folder = cast(CompanyFolderModel, self.get_object())
|
|
ser = cast(CreateContractSerializer, self.get_serializer(data=request.data)) # type: ignore
|
|
ser.is_valid(raise_exception=True)
|
|
contract = ser.save()
|
|
folder.contracts.add(contract)
|
|
return Response(ser.data, status.HTTP_201_CREATED)
|
|
|