initial commit

core/apps/companies/views/__init__.py

@@ -0,0 +1,3 @@
+from .accounts import *  # noqa
+from .companies import *  # noqa
+from .folders import *  # noqa

core/apps/companies/views/accounts.py

@@ -0,0 +1,35 @@
+from django_core.mixins import BaseViewSetMixin
+from drf_spectacular.utils import extend_schema
+from rest_framework.permissions import IsAdminUser, AllowAny
+from rest_framework.viewsets import ModelViewSet
+
+from core.apps.companies.models import CompanyAccountModel
+from core.apps.companies.serializers.accounts import (
+    CreateCompanyAccountSerializer,
+    ListCompanyAccountSerializer,
+    RetrieveCompanyAccountSerializer,
+    UpdateCompanyAccountSerializer,
+    DestroyCompanyAccountSerializer,
+)
+
+
+@extend_schema(tags=["CompanyAccount"])
+class CompanyAccountView(BaseViewSetMixin, ModelViewSet):
+    queryset = CompanyAccountModel.objects.all()
+    serializer_class = ListCompanyAccountSerializer
+    permission_classes = [AllowAny]
+
+    action_permission_classes = {
+        "list": [IsAdminUser],
+        "retrieve": [IsAdminUser],
+        "create": [IsAdminUser],
+        "update": [IsAdminUser],
+        "destroy": [IsAdminUser],
+    }
+    action_serializer_class = {
+        "list": ListCompanyAccountSerializer,
+        "retrieve": RetrieveCompanyAccountSerializer,
+        "create": CreateCompanyAccountSerializer,
+        "update": UpdateCompanyAccountSerializer,
+        "destroy": DestroyCompanyAccountSerializer,
+    }

core/apps/companies/views/companies.py

@@ -0,0 +1,189 @@
+import uuid
+
+from django_core.mixins import BaseViewSetMixin  # type: ignore
+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 (  # type: ignore
+    ModelViewSet, 
+    GenericViewSet
+)
+from rest_framework.request import HttpRequest  # type: ignore
+from rest_framework.response import Response  # type: ignore
+from rest_framework import status  # type: ignore
+from rest_framework.generics import get_object_or_404  # type: ignore
+
+
+from django.contrib.auth import get_user_model
+from core.apps.companies.permissions import IsCompanyAccount
+
+from core.apps.companies.models import (
+    CompanyModel,
+    CompanyFolderModel,
+    CompanyAccountModel
+)
+from core.apps.companies.serializers import (
+    CreateCompanySerializer,
+    ListCompanySerializer,
+    RetrieveCompanySerializer,
+    UpdateCompanySerializer,
+    DestroyCompanySerializer,
+
+    RetrieveCompanyFolderSerializer,
+    RetrieveCompanyAccountSerializer,
+    CreateCompanyFolderSerializer,
+    BaseCompanyAccountSerializer,
+
+    CreateCompanyFolderFromCompanySerializer
+)
+
+from core.apps.contracts.serializers import (
+    RetrieveContractSerializer,
+    BaseContractSerializer
+)
+
+from core.apps.contracts.models import (
+    ContractModel,
+)
+
+
+UserModel = get_user_model()
+
+
+@extend_schema(tags=["Company"])
+class CompanyView(BaseViewSetMixin, ModelViewSet):
+    queryset = CompanyModel.objects.all()
+    serializer_class = ListCompanySerializer
+    permission_classes = [AllowAny]
+
+    action_permission_classes = {
+        "list": [IsAdminUser],
+        "retrieve": [IsAdminUser],
+        "create": [IsAdminUser],
+        "update": [IsAdminUser],
+        "destroy": [IsAdminUser],
+    }
+    action_serializer_class = {  # type: ignore
+        "list": ListCompanySerializer,
+        "retrieve": RetrieveCompanySerializer,
+        "create": CreateCompanySerializer,
+        "update": UpdateCompanySerializer,
+        "destroy": DestroyCompanySerializer,
+    }
+
+
+class CompanyContractViewSet(BaseViewSetMixin, GenericViewSet):
+    queryset = CompanyModel.objects.all()
+    permission_classes = [AllowAny]
+    serializer_class = BaseContractSerializer
+
+    action_permission_classes = {
+        "list_contract": [IsCompanyAccount]
+    }
+    action_serializer_class = {
+        "list_contract": RetrieveContractSerializer
+    }
+
+    #! TODO: status should be added.
+    @extend_schema(
+        summary="Company Contracts",
+        description="Get List Company Contracts"
+    )
+    @action(methods=["GET"], detail=True, url_path="contracts")
+    def list_contract(
+        self,
+        request: HttpRequest,
+        *args: object, 
+        **kwargs: object,
+    ) -> Response:
+        company = self.get_object()
+        contracts = ContractModel.objects.filter(
+            owners__legal_entity__phone=company.phone,
+        ).distinct()
+
+        folders_param = request.data.get("folders")
+        if folders_param:
+            folder_ids = folders_param.split(",")
+            contracts = contracts.filter(folders__id__in=folder_ids)
+
+        serializer = self.get_serializer(instance=contracts, many=True)  # type: ignore
+        return Response(serializer.data, status=status.HTTP_200_OK)
+
+
+class CompanyAccountViewSet(BaseViewSetMixin, GenericViewSet):
+    queryset = CompanyModel.objects.all()
+    serializer_class = BaseCompanyAccountSerializer
+    permission_classes = [AllowAny]
+
+    action_permission_classes = {
+        "list_account": [IsCompanyAccount]
+    }
+    action_serializer_class = {
+        "list_account": RetrieveCompanyAccountSerializer
+    }
+
+    @extend_schema(
+        summary="List company accounts",
+        description="List Company Accounts"
+    )
+    @action(url_path="accounts", detail=True, methods=["GET"])
+    def list_account(
+        self,
+        request: HttpRequest,
+        pk: uuid.UUID,
+        *args: object, 
+        **kwargs: object,
+    ) -> Response:
+        company = self.get_object()
+        accounts = CompanyAccountModel.objects.filter(company=company)
+        ser = self.get_serializer(instance=accounts, many=True)  # type: ignore
+        return Response(data=ser.data, status=status.HTTP_200_OK)
+
+
+class CompanyFolderViewSet(BaseViewSetMixin, GenericViewSet):
+    queryset = CompanyModel.objects.all()
+    permission_classes = [AllowAny]
+    # serializer_class = BaseCompanyFolderSerializer
+
+    action_permission_classes = {
+        "list_folder": [IsCompanyAccount],
+        "create_folder": [IsCompanyAccount],
+    }
+    action_serializer_class = {  # type: ignore
+        "list_folder": RetrieveCompanyFolderSerializer,
+        "create_folder": CreateCompanyFolderFromCompanySerializer,
+    }
+
+    @extend_schema(
+        summary="List Company Folders",
+        description="List Company Folders"
+    )
+    @action(methods=["GET"], detail=True, url_path="folders")
+    def list_folder(
+        self,
+        request: HttpRequest,
+        *args: object, 
+        **kwargs: object,
+    ) -> Response:
+        company = self.get_object()
+        folders = CompanyFolderModel.objects.filter(company=company)
+        ser = self.get_serializer(instance=folders, many=True, context={"company_id": company.pk})  # type: ignore
+        return Response(data=ser.data, status=status.HTTP_200_OK)
+    
+    @extend_schema(
+        summary="Create Folder for company",
+        description="Create Folder for company",
+    )
+    @action(url_path="folders", detail=True, methods=["POST"])
+    def create_folder(
+        self,
+        request: HttpRequest,
+        pk: uuid.UUID,
+        *args: object, 
+        **kwargs: object,
+    ) -> Response:
+        company = self.get_object()
+        data = request.data.copy() | dict(company=company.id)  # type: ignore
+        ser = CreateCompanyFolderSerializer(data=data)  # type: ignore
+        ser.is_valid(raise_exception=True)
+        return Response(data=ser.data, status=status.HTTP_201_CREATED)

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,
+    }