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

cognite/extractorutils/configtools/__init__.py

@@ -90,6 +90,7 @@ from cognite.extractorutils.exceptions import InvalidConfigError
 from .elements import (
     AuthenticatorConfig,
     BaseConfig,
+    CastableInt,
     CertificateConfig,
     CogniteConfig,
     ConfigType,
@@ -99,6 +100,7 @@ from .elements import (
     LocalStateStoreConfig,
     LoggingConfig,
     MetricsConfig,
+    PortNumber,
     RawDestinationConfig,
     RawStateStoreConfig,
     StateStoreConfig,

cognite/extractorutils/configtools/elements.py

@@ -744,3 +744,43 @@ class IgnorePattern:
             _logger.warning("'options' is preferred over 'flags' as this may be removed in a future release")
             self.options = self.flags
             self.flags = None
+
+
+class CastableInt(int):
+    """
+    Represents an integer in a config schema. Difference from regular int is that the
+    value if this type can be  either a string or an integer in the yaml file.
+    """
+
+    def __new__(cls, value: Any) -> "CastableInt":
+        """
+        Returns value as is if it's int. If it's str or bytes try to convert to int.
+        Raises ValueError if conversion is unsuccessful or value is of not supported type.
+
+        Type check is required to avoid unexpected behaviour, such as implictly casting booleans,
+        floats and other types supported by standard int.
+        """
+
+        if not isinstance(value, (int, str, bytes)):
+            raise ValueError(f"CastableInt cannot be created form value {value!r} of type {type(value)!r}.")
+
+        return super().__new__(cls, value)
+
+
+class PortNumber(CastableInt):
+    """
+    A subclass of int to be used in config schemas. It represents a valid port number (0 to 65535) and allows the value
+    to be of either str or int type. If the value is not a valid port number raises a ValueError at instantiation.
+    """
+
+    def __new__(cls, value: Any) -> "PortNumber":
+        """
+        Try to convert the `value` to int. If successful, check if it's within a valid range for a port number.
+        Raises ValueError if conversion to int or validation is unsuccessful.
+        """
+        value = super().__new__(cls, value)
+
+        if not (0 <= value <= 65535):
+            raise ValueError(f"Port number must be between 0 and 65535. Got: {value}.")
+
+        return value

cognite/extractorutils/configtools/loaders.py

@@ -36,8 +36,10 @@ from cognite.client import CogniteClient
 from cognite.extractorutils.configtools._util import _to_snake_case
 from cognite.extractorutils.configtools.elements import (
     BaseConfig,
+    CastableInt,
     ConfigType,
     IgnorePattern,
+    PortNumber,
     TimeIntervalConfig,
     _BaseConfig,
 )
@@ -224,7 +226,7 @@ def _load_yaml(
         config = dacite.from_dict(
             data=config_dict,
             data_class=config_type,
-            config=dacite.Config(strict=True, cast=[Enum, TimeIntervalConfig, Path]),
+            config=dacite.Config(strict=True, cast=[Enum, TimeIntervalConfig, Path, CastableInt, PortNumber]),
         )
     except dacite.UnexpectedDataError as e:
         unknowns = [f'"{k.replace("_", "-") if case_style == "hyphen" else k}"' for k in e.keys]

{cognite_extractor_utils-7.4.8.dist-info → cognite_extractor_utils-7.4.9.dist-info}/METADATA

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

@@ -1,10 +1,10 @@
-cognite/extractorutils/__init__.py,sha256=u3-6p5s1eOo5EQ3ORueamuE8fTrlXi5_W7FYr3SsYTE,739
+cognite/extractorutils/__init__.py,sha256=Cxd38BeR0YorUxKXaS9dz_FKoKa2trhrZJFhiHh0EuY,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
+cognite/extractorutils/configtools/__init__.py,sha256=YEpFGJoza23eM8Zj5DqqUj7sEstERV_QYsN6Nw4dKCg,3092
 cognite/extractorutils/configtools/_util.py,sha256=WdZptkZz_vkn1M4Vqwb39Gb1wxTOe_MNJXWHzOtwv50,4797
-cognite/extractorutils/configtools/elements.py,sha256=Kl36zBOPnwxHuBlpQTpoexxSlDx99WDQK0FlUl2ca_8,24749
-cognite/extractorutils/configtools/loaders.py,sha256=_hfGBJo-AAPEsCCwp9pb6OvgZXa0v0vIpZYwQHWUuM0,18204
+cognite/extractorutils/configtools/elements.py,sha256=6HWhh4jVS-v7Wzl9j3D-Gbli7tEtVidnkIT-eQKCOAE,26314
+cognite/extractorutils/configtools/loaders.py,sha256=TEBW9SuvYY-sCq6is0nYSFGM4kcwUdomqcuK9Dv0GbQ,18262
 cognite/extractorutils/configtools/validators.py,sha256=5or-lg8UHNzl7KfZpbVx8b40835RZwHbUQlMsUCAs4w,1053
 cognite/extractorutils/exceptions.py,sha256=1PgvW1FrgVnuNtkwC0RTvG1-FZp1qmBuYrY1AWW-BJc,1188
 cognite/extractorutils/metrics.py,sha256=01ZMRbDisXPxrfCSyTSEkXMsslzmZwEqw18fuu9okdc,15509
@@ -38,7 +38,7 @@ cognite/extractorutils/uploader/time_series.py,sha256=HBtQdsQoIOaL-EG5lMsaY-ORwV
 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.8.dist-info/LICENSE,sha256=psuoW8kuDP96RQsdhzwOqi6fyWv0ct8CR6Jr7He_P_k,10173
-cognite_extractor_utils-7.4.8.dist-info/METADATA,sha256=nW6GyH7MtYw_I_RIiuHW8Dcm5FmOhX3H_w2JdQ_WW0g,5598
-cognite_extractor_utils-7.4.8.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-cognite_extractor_utils-7.4.8.dist-info/RECORD,,
+cognite_extractor_utils-7.4.9.dist-info/LICENSE,sha256=psuoW8kuDP96RQsdhzwOqi6fyWv0ct8CR6Jr7He_P_k,10173
+cognite_extractor_utils-7.4.9.dist-info/METADATA,sha256=I5rq1WJzAnrchQKwS75G30Xx_qZkJ-ZDYP9w_8RZ2jw,5598
+cognite_extractor_utils-7.4.9.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+cognite_extractor_utils-7.4.9.dist-info/RECORD,,