Support setting Content-Type when uploading a Zarr entry

dandiapi/api/storage.py

@@ -182,15 +182,20 @@ def etag_from_blob_name(self, blob_name) -> str | None:
                 return etag[1:-1]
             return etag
 
-    def generate_presigned_put_object_url(self, blob_name: str, base64md5: str) -> str:
+    def generate_presigned_put_object_url(
+        self, blob_name: str, base64md5: str, content_type: str | None
+    ) -> str:
+        params = {
+            'Bucket': self.bucket_name,
+            'Key': blob_name,
+            'ACL': 'bucket-owner-full-control',
+            'ContentMD5': base64md5,
+        }
+        if content_type is not None:
+            params['ContentType'] = content_type
         return self.connection.meta.client.generate_presigned_url(
             ClientMethod='put_object',
-            Params={
-                'Bucket': self.bucket_name,
-                'Key': blob_name,
-                'ACL': 'bucket-owner-full-control',
-                'ContentMD5': base64md5,
-            },
+            Params=params,
             ExpiresIn=600,  # TODO: proper expiration
         )
 
@@ -243,7 +248,7 @@ def etag_from_blob_name(self, blob_name) -> str | None:
         else:
             return response.etag
 
-    def generate_presigned_put_object_url(self, blob_name: str, _: str) -> str:
+    def generate_presigned_put_object_url(self, blob_name: str, _: str, content_type: str) -> str:
         # Note: minio-py doesn't support using Content-MD5 headers
 
         # storage.client will generate URLs like `http://minio:9000/...` when running in

dandiapi/zarr/models.py

@@ -88,7 +88,9 @@ def s3_path(self, zarr_path: str) -> str:
 
     def generate_upload_urls(self, path_md5s: list[dict]):
         return [
-            self.storage.generate_presigned_put_object_url(self.s3_path(o['path']), o['base64md5'])
+            self.storage.generate_presigned_put_object_url(
+                self.s3_path(o['path']), o['base64md5'], content_type=o['content_type']
+            )
             for o in path_md5s
         ]
 

dandiapi/zarr/views/__init__.py

@@ -36,6 +36,7 @@
 class ZarrFileCreationSerializer(serializers.Serializer):
     path = serializers.CharField()
     base64md5 = serializers.CharField()
+    content_type = serializers.CharField(allow_null=True, default=None)
 
 
 class ZarrDeleteFileRequestSerializer(serializers.Serializer):