diff --git a/cognite/extractorutils/__init__.py b/cognite/extractorutils/__init__.py index 6020eb50..5ba16585 100644 --- a/cognite/extractorutils/__init__.py +++ b/cognite/extractorutils/__init__.py @@ -16,5 +16,5 @@ Cognite extractor utils is a Python package that simplifies the development of new extractors. """ -__version__ = "6.1.1" +__version__ = "6.1.2" from .base import Extractor diff --git a/cognite/extractorutils/base.py b/cognite/extractorutils/base.py index edd64f14..9a943536 100644 --- a/cognite/extractorutils/base.py +++ b/cognite/extractorutils/base.py @@ -88,6 +88,7 @@ def __init__( handle_interrupts: bool = True, reload_config_interval: Optional[int] = 300, reload_config_action: ReloadConfigAction = ReloadConfigAction.DO_NOTHING, + success_message: str = "Successful shutdown", ): self.name = name self.description = description @@ -102,6 +103,7 @@ def __init__( self.handle_interrupts = handle_interrupts self.reload_config_interval = reload_config_interval self.reload_config_action = reload_config_action + self.success_message = success_message self.started = False self.configured_logger = False @@ -208,7 +210,7 @@ def _report_success(self) -> None: ExtractionPipelineRun( extpipe_external_id=self.extraction_pipeline.external_id, status="success", - message="Successful shutdown", + message=self.success_message, ) ) diff --git a/pyproject.toml b/pyproject.toml index dd0e7e11..c5f9f29b 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "cognite-extractor-utils" -version = "6.1.1" +version = "6.1.2" description = "Utilities for easier development of extractors for CDF" authors = ["Mathias Lohne "] license = "Apache-2.0"