remplace la classe CSVImportApiView par un fichier utils dédié

api/views/diagnostic.py

@@ -13,6 +13,7 @@
 from rest_framework.exceptions import NotFound, PermissionDenied
 from rest_framework.generics import CreateAPIView, ListAPIView, UpdateAPIView
 from rest_framework.pagination import LimitOffsetPagination
+from rest_framework.views import APIView
 
 from api.exceptions import DuplicateException
 from api.permissions import (
@@ -22,7 +23,8 @@
     IsCanteenManager,
 )
 from api.serializers import DiagnosticAndCanteenSerializer, ManagerDiagnosticSerializer
-from api.views.utils import CSVImportApiView, update_change_reason_with_auth
+from api.views.utils import update_change_reason_with_auth
+from common import file_import as file_import_utils
 from common.utils import send_mail
 from data.models import Canteen, Teledeclaration
 from data.models.diagnostic import Diagnostic
@@ -92,14 +94,14 @@ def perform_update(self, serializer):
         update_change_reason_with_auth(self, diagnostic)
 
 
-class EmailDiagnosticImportFileView(CSVImportApiView):
+class EmailDiagnosticImportFileView(APIView):
     permission_classes = [IsAuthenticated]
 
     def post(self, request):
         try:
             self.file = request.data["file"]
-            super()._verify_file_size()
-            super()._verify_file_format()
+            file_import_utils.validate_file_size(self.file)
+            file_import_utils.validate_file_format(self.file)
             email = request.data.get("email", request.user.email).strip()
             context = {
                 "from": email,

api/views/diagnosticimport.py

@@ -19,7 +19,7 @@
 
 from api.permissions import IsAuthenticated
 from api.serializers import FullCanteenSerializer
-from api.views.utils import CSVImportApiView
+from common import file_import as file_import_utils
 from data.models import Canteen, ImportFailure, ImportType, Sector
 from data.models.diagnostic import Diagnostic
 from data.models.teledeclaration import Teledeclaration
@@ -30,7 +30,7 @@
 logger = logging.getLogger(__name__)
 
 
-class ImportDiagnosticsView(ABC, CSVImportApiView):
+class ImportDiagnosticsView(ABC):
     permission_classes = [IsAuthenticated]
     value_error_regex = re.compile(r"Field '(.+)' expected .+? got '(.+)'.")
     annotated_sectors = Sector.objects.annotate(name_lower=Lower("name"))
@@ -79,8 +79,8 @@ def post(self, request):
         try:
             with transaction.atomic():
                 self.file = request.data["file"]
-                super()._verify_file_size()
-                super()._verify_file_format()
+                file_import_utils.validate_file_size(self.file)
+                file_import_utils.validate_file_format(self.file)
                 self._process_file(self.file)
 
                 if self.errors:

api/views/purchaseimport.py

@@ -12,18 +12,19 @@
 from django.http import JsonResponse
 from rest_framework import status
 from rest_framework.exceptions import PermissionDenied
+from rest_framework.views import APIView
 
 from api.permissions import IsAuthenticated
 from api.serializers import PurchaseSerializer
-from api.views.utils import CSVImportApiView
+from common import file_import as file_import_utils
 from data.models import Canteen, ImportFailure, ImportType, Purchase
 
 from .utils import camelize, decode_bytes, normalise_siret
 
 logger = logging.getLogger(__name__)
 
 
-class ImportPurchasesView(CSVImportApiView):
+class ImportPurchasesView(APIView):
     permission_classes = [IsAuthenticated]
     max_error_items = 30
 
@@ -48,8 +49,8 @@ def post(self, request):
         logger.info("Purchase bulk import started")
         try:
             self.file = request.data["file"]
-            super()._verify_file_size()
-            super()._verify_file_format()
+            file_import_utils.validate_file_size(self.file)
+            file_import_utils.validate_file_format(self.file)
 
             self.file_digest = super()._get_file_digest()
             self._check_duplication()

common/file_import.py

@@ -0,0 +1,34 @@
+import hashlib
+
+from django.conf import settings
+from django.core.exceptions import ValidationError
+
+
+def get_file_size(file):
+    return file.size
+
+
+def validate_file_size(file):
+    if get_file_size(file) > settings.CSV_IMPORT_MAX_SIZE:
+        raise ValidationError(
+            f"Ce fichier est trop grand, merci d'utiliser un fichier de moins de {settings.CSV_IMPORT_MAX_SIZE_PRETTY}"
+        )
+
+
+def get_file_content_type(file):
+    return file.content_type
+
+
+def validate_file_format(file):
+    file_format = get_file_content_type(file)
+    if file_format not in ["text/csv", "text/tab-separated-values"]:
+        raise ValidationError(
+            f"Ce fichier est au format {file_format}, merci d'exporter votre fichier au format CSV et réessayer."
+        )
+
+
+def get_file_digest(file):
+    file_hash = hashlib.md5()
+    for row in file:
+        file_hash.update(row)
+    return file_hash.hexdigest()