cognite-extractor-utils 6.2.0__py3-none-any.whl → 6.2.1__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__ = "6.2.0"
+__version__ = "6.2.1"
 from .base import Extractor

cognite/extractorutils/uploader/files.py

@@ -17,12 +17,12 @@ from concurrent.futures import Future, ThreadPoolExecutor
 from io import BytesIO
 from os import PathLike
 from types import TracebackType
-from typing import Any, BinaryIO, Callable, List, Optional, Tuple, Type, Union
+from typing import BinaryIO, Callable, List, Optional, Tuple, Type, Union
 
 from requests import ConnectionError
 
 from cognite.client import CogniteClient
-from cognite.client.data_classes import Event, FileMetadata
+from cognite.client.data_classes import FileMetadata
 from cognite.client.exceptions import CogniteAPIError
 from cognite.extractorutils.uploader._base import (
     RETRIES,
@@ -64,7 +64,7 @@ class IOFileUploadQueue(AbstractUploadQueue):
     def __init__(
         self,
         cdf_client: CogniteClient,
-        post_upload_function: Optional[Callable[[List[Event]], None]] = None,
+        post_upload_function: Optional[Callable[[List[FileMetadata]], None]] = None,
         max_queue_size: Optional[int] = None,
         max_upload_interval: Optional[int] = None,
         trigger_log_level: str = "DEBUG",
@@ -128,7 +128,7 @@ class IOFileUploadQueue(AbstractUploadQueue):
             self.files_written.inc(self.upload_queue_size)
 
             try:
-                self._post_upload(self.upload_queue)
+                self._post_upload([el[0] for el in self.upload_queue])
             except Exception as e:
                 self.logger.error("Error in upload callback: %s", str(e))
             self.upload_queue.clear()
@@ -228,7 +228,7 @@ class FileUploadQueue(IOFileUploadQueue):
     def __init__(
         self,
         cdf_client: CogniteClient,
-        post_upload_function: Optional[Callable[[List[Event]], None]] = None,
+        post_upload_function: Optional[Callable[[List[FileMetadata]], None]] = None,
         max_queue_size: Optional[int] = None,
         max_upload_interval: Optional[int] = None,
         trigger_log_level: str = "DEBUG",
@@ -284,7 +284,7 @@ class BytesUploadQueue(IOFileUploadQueue):
     def __init__(
         self,
         cdf_client: CogniteClient,
-        post_upload_function: Optional[Callable[[List[Any]], None]] = None,
+        post_upload_function: Optional[Callable[[List[FileMetadata]], None]] = None,
         max_queue_size: Optional[int] = None,
         max_upload_interval: Optional[int] = None,
         trigger_log_level: str = "DEBUG",

{cognite_extractor_utils-6.2.0.dist-info → cognite_extractor_utils-6.2.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cognite-extractor-utils
-Version: 6.2.0
+Version: 6.2.1
 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-6.2.0.dist-info → cognite_extractor_utils-6.2.1.dist-info}/RECORD

@@ -1,4 +1,4 @@
-cognite/extractorutils/__init__.py,sha256=Js6RU6bgK6AajneFJr5PwjeiOv0sq8DJxBdRjZXdnBA,739
+cognite/extractorutils/__init__.py,sha256=x-wPtqhxoUAlGmkP73DLkPGwjmdssPlYGwe1TKhDAfI,739
 cognite/extractorutils/_inner_util.py,sha256=jCPLg-FfTpyLAHkhoKaKObNwbqm19d2Z11to_DYk5EU,1558
 cognite/extractorutils/base.py,sha256=8kzePjVlCIQVftDQ1t_4join9qhs-VOlB6eTpeX9zGk,16018
 cognite/extractorutils/configtools/__init__.py,sha256=fj9kH8DdisNi9mI8cKm2sz50vnzeOkJQErIGB3mTYRo,2861
@@ -15,13 +15,13 @@ cognite/extractorutils/uploader/_base.py,sha256=ed9F-qFmlIylTg8UtuxfykYUfEsG_R0P
 cognite/extractorutils/uploader/_metrics.py,sha256=J2LJXb19L_SLSJ_voNIQHYLp0pjxUKevpH1q_xKX6Hk,3247
 cognite/extractorutils/uploader/assets.py,sha256=Gqu1VJqfFcIQphsfqciDFghL_zlNhBmHU8hPfaKSyFw,5441
 cognite/extractorutils/uploader/events.py,sha256=dzqdDHRKsSUiBWqpcwJX9BpTiUGLb6abneqk1k_QmLA,5502
-cognite/extractorutils/uploader/files.py,sha256=KiNbY3tQgvroXDlv2Z-3gpo-lEqbgz-5WtrYUtRBkTw,11755
+cognite/extractorutils/uploader/files.py,sha256=u1tJDT8cCD9W9EegQWqOYevsj3BfOdFfyiO3DSWHjsc,11784
 cognite/extractorutils/uploader/raw.py,sha256=9kvHtJoZVLD-V9xcnNEkucwvaa8DPMqsXjvCWMGiw_g,6732
 cognite/extractorutils/uploader/time_series.py,sha256=ZPkz9NVCmz2vE7SkRm0-LjYi-ZKCRZ_Qe-YR1EVtGms,26315
 cognite/extractorutils/uploader_extractor.py,sha256=K8cH8CIDiFNB0fXNdUymE_MKh1GYfqxHHcRoH07CFew,7673
 cognite/extractorutils/uploader_types.py,sha256=5MKT14DUnTFVD5Nx4Zvnfp2SfaICuTKLWAFAYaZishk,1045
 cognite/extractorutils/util.py,sha256=GDDpsoOQpgoTHUWVAUjWxGdulgOQovlD8yBWp_kObp8,14234
-cognite_extractor_utils-6.2.0.dist-info/LICENSE,sha256=psuoW8kuDP96RQsdhzwOqi6fyWv0ct8CR6Jr7He_P_k,10173
-cognite_extractor_utils-6.2.0.dist-info/METADATA,sha256=Uqv5Y5mhX6uvEQO7nig0HXaqrIOI2tbGPeQ07CDIlos,5457
-cognite_extractor_utils-6.2.0.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-cognite_extractor_utils-6.2.0.dist-info/RECORD,,
+cognite_extractor_utils-6.2.1.dist-info/LICENSE,sha256=psuoW8kuDP96RQsdhzwOqi6fyWv0ct8CR6Jr7He_P_k,10173
+cognite_extractor_utils-6.2.1.dist-info/METADATA,sha256=US8F0uxofliB65ELOEfC_dPUZNuP0lqdLWOD1eUpcIo,5457
+cognite_extractor_utils-6.2.1.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+cognite_extractor_utils-6.2.1.dist-info/RECORD,,