cognite-extractor-utils 7.4.1__py3-none-any.whl → 7.4.2__py3-none-any.whl

cognite/extractorutils/__init__.py

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

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.1.dist-info → cognite_extractor_utils-7.4.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cognite-extractor-utils
-Version: 7.4.1
+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

{cognite_extractor_utils-7.4.1.dist-info → cognite_extractor_utils-7.4.2.dist-info}/RECORD

@@ -1,4 +1,4 @@
-cognite/extractorutils/__init__.py,sha256=xMvvSXWtbvBeI27mzSQnukO0QT4mE9pEmtOzOULAzAk,739
+cognite/extractorutils/__init__.py,sha256=ulj-H7uJEPJAUJUhJg1funj27thycop0GTNOzJTIAXc,739
 cognite/extractorutils/_inner_util.py,sha256=gmz6aqS7jDNsg8z4RHgJjMFohDLOMiaU4gMWBhg3xcE,1558
 cognite/extractorutils/base.py,sha256=q6NU2bPec3WOasVnnIFoh-aUJudVZWZ2R6emz3IRj8Q,16391
 cognite/extractorutils/configtools/__init__.py,sha256=L-daaqInIsmHcjb2forJeY0fW8tz1mlteOUo7IsWnrU,3059
@@ -20,13 +20,13 @@ cognite/extractorutils/uploader/_metrics.py,sha256=J2LJXb19L_SLSJ_voNIQHYLp0pjxU
 cognite/extractorutils/uploader/assets.py,sha256=2E90N1kxsaA6Ah4h0_r_dTVhDYY_68ItRWrHYkkltJw,5628
 cognite/extractorutils/uploader/data_modeling.py,sha256=w35Ix5mu0Cgfn4ywnDyif4VVjo04LVTlkMEevk6ztUs,3639
 cognite/extractorutils/uploader/events.py,sha256=NZP2tMoU_rh_rb-EZiUBsOT5KdNABHN4c9Oddk0OsdE,5680
-cognite/extractorutils/uploader/files.py,sha256=jEZ_QwUnXTsfQ5Xsm03j_vNWTlYBg2gmSxE3MOyoC6s,21765
+cognite/extractorutils/uploader/files.py,sha256=orBa5_8Rh60PjJkGTAbcsgnmOYORwQaiy2atOZ95x48,21983
 cognite/extractorutils/uploader/raw.py,sha256=wFjF90PFTjmByOWx_Y4_YfDJ2w2jl0EQJ2Tjx2MP2PM,6738
 cognite/extractorutils/uploader/time_series.py,sha256=HBtQdsQoIOaL-EG5lMsaY-ORwVb0kGiXG86VjE5-_Bg,26815
 cognite/extractorutils/uploader_extractor.py,sha256=E-mpVvbPg_Tk90U4S9JybV0duptJ2SXE88HB6npE3zI,7732
 cognite/extractorutils/uploader_types.py,sha256=wxfrsiKPTzG5lmoYtQsxt8Xyj-s5HnaLl8WDzJNrazg,1020
 cognite/extractorutils/util.py,sha256=T6ef5b7aYJ8yq9swQwybYaLe3YGr3hElsJQy8E-d5Rs,17469
-cognite_extractor_utils-7.4.1.dist-info/LICENSE,sha256=psuoW8kuDP96RQsdhzwOqi6fyWv0ct8CR6Jr7He_P_k,10173
-cognite_extractor_utils-7.4.1.dist-info/METADATA,sha256=KF2LYMRHr4eMaykYRGhHt8RE0mgTB-L6LzfIu_P9tYI,5476
-cognite_extractor_utils-7.4.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-cognite_extractor_utils-7.4.1.dist-info/RECORD,,
+cognite_extractor_utils-7.4.2.dist-info/LICENSE,sha256=psuoW8kuDP96RQsdhzwOqi6fyWv0ct8CR6Jr7He_P_k,10173
+cognite_extractor_utils-7.4.2.dist-info/METADATA,sha256=cd3oSr3l7x5iL8Bj2B03kRc_J8EERrqI-QHPHsyBDmg,5476
+cognite_extractor_utils-7.4.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+cognite_extractor_utils-7.4.2.dist-info/RECORD,,