cognite-extractor-utils 7.4.0__tar.gz → 7.4.2__tar.gz

{cognite_extractor_utils-7.4.0 → cognite_extractor_utils-7.4.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cognite-extractor-utils
-Version: 7.4.0
+Version: 7.4.2
 Summary: Utilities for easier development of extractors for CDF
 Home-page: https://github.com/cognitedata/python-extractor-utils
 License: Apache-2.0
@@ -17,7 +17,7 @@ Provides-Extra: experimental
 Requires-Dist: arrow (>=1.0.0,<2.0.0)
 Requires-Dist: azure-identity (>=1.14.0,<2.0.0)
 Requires-Dist: azure-keyvault-secrets (>=4.7.0,<5.0.0)
-Requires-Dist: cognite-sdk (>=7.54.17,<8.0.0)
+Requires-Dist: cognite-sdk (>=7.58.4,<8.0.0)
 Requires-Dist: dacite (>=1.6.0,<2.0.0)
 Requires-Dist: decorator (>=5.1.1,<6.0.0)
 Requires-Dist: httpx (>=0.27.0,<0.28.0)

{cognite_extractor_utils-7.4.0 → cognite_extractor_utils-7.4.2}/cognite/extractorutils/__init__.py

@@ -16,5 +16,5 @@
 Cognite extractor utils is a Python package that simplifies the development of new extractors.
 """
 
-__version__ = "7.3.0"
+__version__ = "7.4.2"
 from .base import Extractor

{cognite_extractor_utils-7.4.0 → cognite_extractor_utils-7.4.2}/cognite/extractorutils/uploader/files.py

@@ -264,7 +264,7 @@ class IOFileUploadQueue(AbstractUploadQueue):
 
     def _upload_bytes(self, size: int, file: BinaryIO, meta_or_apply: FileMetadataOrCogniteExtractorFile) -> None:
         meta_or_apply, url = self._upload_empty(meta_or_apply)
-        resp = self._httpx_client.send(self._get_file_upload_request(url, file, size))
+        resp = self._httpx_client.send(self._get_file_upload_request(url, file, size, meta_or_apply.mime_type))
         resp.raise_for_status()
 
     def _upload_multipart(self, size: int, file: BinaryIO, meta_or_apply: FileMetadataOrCogniteExtractorFile) -> None:
@@ -281,7 +281,9 @@ class IOFileUploadQueue(AbstractUploadQueue):
 
         for url in upload_urls:
             chunks.next_chunk()
-            resp = self._httpx_client.send(self._get_file_upload_request(url, chunks, len(chunks)))
+            resp = self._httpx_client.send(
+                self._get_file_upload_request(url, chunks, len(chunks), meta_or_apply.mime_type)
+            )
             resp.raise_for_status()
 
         completed_headers = (
@@ -400,7 +402,9 @@ class IOFileUploadQueue(AbstractUploadQueue):
             self.files_queued.inc()
             self.queue_size.set(self.upload_queue_size)
 
-    def _get_file_upload_request(self, url_str: str, stream: BinaryIO, size: int) -> Request:
+    def _get_file_upload_request(
+        self, url_str: str, stream: BinaryIO, size: int, mime_type: Optional[str] = None
+    ) -> Request:
         url = URL(url_str)
         headers = Headers(self._httpx_client.headers)
         headers.update(
@@ -412,6 +416,9 @@ class IOFileUploadQueue(AbstractUploadQueue):
             }
         )
 
+        if mime_type is not None:
+            headers.update({"Content-Type": mime_type})
+
         return Request(
             method="PUT",
             url=url,

{cognite_extractor_utils-7.4.0 → cognite_extractor_utils-7.4.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "cognite-extractor-utils"
-version = "7.4.0"
+version = "7.4.2"
 description = "Utilities for easier development of extractors for CDF"
 authors = ["Mathias Lohne <mathias.lohne@cognite.com>"]
 license = "Apache-2.0"
@@ -58,7 +58,7 @@ exclude = "tests/*"
 
 [tool.poetry.dependencies]
 python = "^3.9.0"
-cognite-sdk = "^7.54.17"
+cognite-sdk = "^7.58.4"
 prometheus-client = ">0.7.0, <=1.0.0"
 arrow = "^1.0.0"
 pyyaml = ">=5.3.0, <7"