Add option to configure security categories (#280)

CHANGELOG.md

@@ -12,6 +12,19 @@ Changes are grouped as follows
 - `Fixed` for any bug fixes.
 - `Security` in case of vulnerabilities.
 
+## [6.2.2]
+
+### Added
+
+ * Added an optional `security-categories` attribute to the `cognite` config
+   section.
+
+## [6.2.1]
+
+### Fixed
+
+ * Fixed a type hint in the `post_upload_function` for upload queues.
+
 ## [6.2.0]
 
 ### Added

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.1"
+__version__ = "6.2.2"
 from .base import Extractor

cognite/extractorutils/configtools/elements.py

@@ -261,6 +261,7 @@ class CogniteConfig:
     extraction_pipeline: Optional[EitherIdConfig]
     timeout: TimeIntervalConfig = TimeIntervalConfig("30s")
     connection: ConnectionConfig = field(default_factory=ConnectionConfig)
+    security_categories: Optional[List[int]] = None
     external_id_prefix: str = ""
     host: str = "https://api.cognitedata.com"
 

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "cognite-extractor-utils"
-version = "6.2.1"
+version = "6.2.2"
 description = "Utilities for easier development of extractors for CDF"
 authors = ["Mathias Lohne <[email protected]>"]
 license = "Apache-2.0"

schema/cognite_config.schema.json

@@ -91,6 +91,13 @@
             "description": "Extraction pipeline used for remote config and reporting status",
             "$ref": "either_id.schema.json"
         },
+        "security-categories": {
+            "description": "Security category to attach to any data written by the extractor (where relevant)",
+            "type": "array",
+            "items": {
+                "type": "integer"
+            }
+        },
         "host": {
             "type": "string",
             "default": "https://api.cognitedata.com",