changing code structure

core/apps/evaluation/views/document.py

@@ -1,51 +1,51 @@
+# django
 from django.shortcuts import get_object_or_404
 from django_core.mixins import BaseViewSetMixin
+
+# swagger
 from drf_spectacular.utils import extend_schema, OpenApiParameter
+
+# rest framework
 from rest_framework.exceptions import NotFound, PermissionDenied
 from rest_framework.permissions import AllowAny
 from rest_framework.viewsets import ReadOnlyModelViewSet, ModelViewSet
 from rest_framework.parsers import FormParser, MultiPartParser
 from rest_framework.decorators import action
 from rest_framework.response import Response
+
+# filters
 from django_filters.rest_framework import DjangoFilterBackend
 
 from core.apps.evaluation.models import DocumentModel, ValuationDocumentModel, AutoEvaluationModel
-from core.apps.evaluation.serializers.document import (
-    CreateDocumentSerializer,
-    CreateValuationdocumentSerializer,
-    ListDocumentSerializer,
-    ListValuationdocumentSerializer,
-    RetrieveDocumentSerializer,
-    RetrieveValuationdocumentSerializer,
-)
+from core.apps.evaluation.serializers import document as serializers
 
 
 @extend_schema(tags=["ValuationDocument"])
 class ValuationDocumentView(BaseViewSetMixin, ReadOnlyModelViewSet):
     queryset = ValuationDocumentModel.objects.all()
-    serializer_class = ListValuationdocumentSerializer
+    serializer_class = serializers.ListValuationdocumentSerializer
     permission_classes = [AllowAny]
 
     action_permission_classes = {}
     action_serializer_class = {
-        "list": ListValuationdocumentSerializer,
-        "retrieve": RetrieveValuationdocumentSerializer,
-        "create": CreateValuationdocumentSerializer,
+        "list": serializers.ListValuationdocumentSerializer,
+        "retrieve": serializers.RetrieveValuationdocumentSerializer,
+        "create": serializers.CreateValuationdocumentSerializer,
     }
 
 
 @extend_schema(tags=["Document"])
 class DocumentView(BaseViewSetMixin, ModelViewSet):
     queryset = DocumentModel.objects.all()
-    serializer_class = ListDocumentSerializer
+    serializer_class = serializers.ListDocumentSerializer
     permission_classes = [AllowAny]
     parser_classes = [FormParser, MultiPartParser]
     filter_backends = [DjangoFilterBackend]
     action_permission_classes = {}
     action_serializer_class = {
-        "list": ListDocumentSerializer,
-        "retrieve": RetrieveDocumentSerializer,
-        "create": CreateDocumentSerializer,
+        "list": serializers.ListDocumentSerializer,
+        "retrieve": serializers.RetrieveDocumentSerializer,
+        "create": serializers.CreateDocumentSerializer,
     }
 
     @extend_schema(
@@ -83,9 +83,9 @@ class DocumentView(BaseViewSetMixin, ModelViewSet):
                 documents = documents.filter(category_id=category_id)
             page = self.paginate_queryset(documents)
             if page is not None:
-                serializer = ListDocumentSerializer(page, many=True, context={"request": request})
+                serializer = serializers.ListDocumentSerializer(page, many=True, context={"request": request})
                 return self.get_paginated_response(serializer.data)
-            serializer = ListDocumentSerializer(documents, many=True, context={"request": request})
+            serializer = serializers.ListDocumentSerializer(documents, many=True, context={"request": request})
             return Response(serializer.data)
         except AutoEvaluationModel.DoesNotExist:
             raise NotFound("Auto evaluation not found")