Allow multipart requests to cloud storages

cvat-ai · Mar 28, 2023 · 7bf6885 · 7bf6885
1 parent d194003
commit 7bf6885
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/cvat/apps/engine/views.py b/cvat/apps/engine/views.py
@@ -2091,6 +2091,10 @@ class CloudStorageViewSet(viewsets.GenericViewSet, mixins.ListModelMixin,
     lookup_fields = {'owner': 'owner__username', 'name': 'display_name'}
     iam_organization_field = 'organization'
 
+    # Multipart support is necessary here, as CloudStorageWriteSerializer
+    # contains a file field (secret_key).
+    parser_classes = _UPLOAD_PARSER_CLASSES
+
     def get_serializer_class(self):
         if self.request.method in ('POST', 'PATCH'):
             return CloudStorageWriteSerializer

diff --git a/cvat/schema.yml b/cvat/schema.yml
@@ -518,6 +518,9 @@ paths:
                   manifests:
                   - manifest.jsonl
                 summary: Create GCS
+          multipart/form-data:
+            schema:
+              $ref: '#/components/schemas/CloudStorageWriteRequest'
         required: true
       security:
       - sessionAuth: []
@@ -647,6 +650,9 @@ paths:
                   manifests:
                   - manifest.jsonl
                 summary: Create GCS
+          multipart/form-data:
+            schema:
+              $ref: '#/components/schemas/PatchedCloudStorageWriteRequest'
       security:
       - sessionAuth: []
         csrfAuth: []