Merge branch 'master' of https://github.com/cognitedata/python-extractor-utils into DOG-4924-investigate-security-implications-of-dotenv-usage-in-cognite-extractor-utils

Author: Nithin Bodanapu

CHANGELOG.md

@@ -12,12 +12,18 @@ Changes are grouped as follows
 - `Fixed` for any bug fixes.
 - `Security` in case of vulnerabilities.
 
-## 7.5.12
+## 7.5.13
 
 ### Security
 
 * Disabled recursive search for .env file. The change will prevent loading environment variables hosted in directories other than the current working directory. Environment variable management is now more explicit to enhance security, and reduce potential misconfigurations.
 
+## 7.5.12
+
+## Added
+
+- Allow users to override what message to send on shutdown
+
 ## 7.5.11
 
 ### Added

cognite/extractorutils/__init__.py

@@ -16,7 +16,7 @@
 Cognite extractor utils is a Python package that simplifies the development of new extractors.
 """
 
-__version__ = "7.5.12"
+__version__ = "7.5.13"
 from .base import Extractor
 
 __all__ = ["Extractor"]

cognite/extractorutils/base.py

@@ -90,6 +90,7 @@ def __init__(
         handle_interrupts: bool = True,
         reload_config_interval: int | None = 300,
         reload_config_action: ReloadConfigAction = ReloadConfigAction.DO_NOTHING,
+        success_message: str = "Successful shutdown",
     ):
         self.name = name
         self.description = description
@@ -104,6 +105,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
@@ -218,7 +220,7 @@ def _report_success(self) -> None:
                 ExtractionPipelineRun(
                     extpipe_external_id=self.extraction_pipeline.external_id,
                     status="success",
-                    message="Successful shutdown",
+                    message=self.success_message,
                 )
             )
 

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "cognite-extractor-utils"
-version = "7.5.12"
+version = "7.5.13"
 description = "Utilities for easier development of extractors for CDF"
 authors = [
     {name = "Mathias Lohne", email = "mathias.lohne@cognite.com"}