datadog-checks-base 37.12.0__py2.py3-none-any.whl → 37.15.0__py2.py3-none-any.whl

datadog_checks/base/__about__.py

@@ -1,4 +1,4 @@
 # (C) Datadog, Inc. 2018-present
 # All rights reserved
 # Licensed under a 3-clause BSD style license (see LICENSE)
-__version__ = "37.12.0"
+__version__ = "37.15.0"

datadog_checks/base/checks/_config_ast.py

@@ -0,0 +1,84 @@
+import ast
+from enum import Enum
+from typing import Any
+
+# This module is used to parse and modify the config AST before it is loaded.
+# It is used to handle special float values (inf, -inf, nan) and replace them with placeholders since those
+# are not valid Python literals.
+
+
+class SpecialFloatPlaceholder(str, Enum):
+    INF = '__PYTHON_INF__'
+    NEG_INF = '__PYTHON_NEG_INF__'
+    NAN = '__PYTHON_NAN__'
+
+
+class _SpecialFloatValuesTransformer(ast.NodeTransformer):
+    def visit_Name(self, node: ast.Name) -> ast.AST:
+        """
+        Processes named constants like 'inf' and 'nan'.
+        If the name is 'inf', it's replaced with a placeholder for positive infinity.
+        If the name is 'nan', it's replaced with a placeholder for Not a Number.
+        Other names are returned unchanged.
+        """
+        if node.id == 'inf':
+            return ast.Constant(value=SpecialFloatPlaceholder.INF.value)
+        elif node.id == 'nan':
+            return ast.Constant(value=SpecialFloatPlaceholder.NAN.value)
+        return node  # Leaf node, no children to visit
+
+    def visit_UnaryOp(self, node: ast.UnaryOp) -> ast.AST:
+        """
+        Processes unary operations like negation.
+        If the operation is a negation ('-') applied to the name 'inf',
+        it's replaced with a placeholder for negative infinity.
+        We can't use visit_Name for this because the constant is 'inf' and not '-inf'.
+        Other unary operations are processed as normal.
+        """
+        if isinstance(node.op, ast.USub) and isinstance(node.operand, ast.Name) and node.operand.id == 'inf':
+            return ast.Constant(value=SpecialFloatPlaceholder.NEG_INF.value)
+        return self.generic_visit(node)
+
+
+def _restore_special_floats(data: Any) -> Any:
+    """
+    Restores placeholders for special float values (inf, -inf, nan) to their actual
+    float values in a nested data structure.
+    """
+    if isinstance(data, dict):
+        return {key: _restore_special_floats(value) for key, value in data.items()}
+    elif isinstance(data, list):
+        return [_restore_special_floats(item) for item in data]
+    elif isinstance(data, str):
+        if data == SpecialFloatPlaceholder.INF.value:
+            return float('inf')
+        elif data == SpecialFloatPlaceholder.NEG_INF.value:
+            return float('-inf')
+        elif data == SpecialFloatPlaceholder.NAN.value:
+            return float('nan')
+    return data
+
+
+def parse(object_string: str) -> Any:
+    """
+    Parses a printed Python object, handling special float values (inf, -inf, nan).
+    If any error occurs, the original string is returned.
+    """
+    try:
+        if not object_string:
+            return None
+
+        # Parse the string as a Python expression
+        ast_node = ast.parse(object_string, mode='eval').body
+
+        # Replace inf/nan with placeholders
+        transformer = _SpecialFloatValuesTransformer()
+        transformed_ast_node = transformer.visit(ast_node)
+
+        # Evaluate the AST node to get the actual value.
+        data_with_placeholders = ast.literal_eval(transformed_ast_node)
+
+        # Restore placeholders to actual float values
+        return _restore_special_floats(data_with_placeholders)
+    except Exception:
+        return object_string

datadog_checks/base/checks/base.py

@@ -48,6 +48,7 @@ from ..utils.common import ensure_bytes, to_native_string
 from ..utils.fips import enable_fips
 from ..utils.tagging import GENERIC_TAGS
 from ..utils.tracing import traced_class
+from ._config_ast import parse as _parse_ast_config
 
 if AGENT_RUNNING:
     from ..log import CheckLoggingAdapter, init_logging
@@ -1488,9 +1489,4 @@ class AgentCheck(object):
         if process.returncode != 0:
             raise ValueError(f'Failed to load config: {stderr.decode()}')
 
-        decoded = stdout.strip().decode()
-        try:
-            return eval(decoded)
-        # a single, literal unquoted string
-        except Exception:
-            return decoded
+        return _parse_ast_config(stdout.strip().decode())

datadog_checks/base/utils/db/utils.py

@@ -11,8 +11,9 @@ import socket
 import threading
 import time
 from concurrent.futures.thread import ThreadPoolExecutor
+from enum import Enum, auto
 from ipaddress import IPv4Address
-from typing import Any, Callable, Dict, List, Tuple  # noqa: F401
+from typing import Any, Callable, Dict, List, Optional, Tuple, Union  # noqa: F401
 
 from cachetools import TTLCache
 
@@ -441,3 +442,128 @@ def tracked_query(check, operation, tags=None):
     yield
     elapsed_ms = (time.time() - start_time) * 1000
     check.histogram("dd.{}.operation.time".format(check.name), elapsed_ms, **stats_kwargs)
+
+
+class TagType(Enum):
+    """Enum for different types of tags"""
+
+    KEYLESS = auto()
+
+
+class TagManager:
+    """
+    Manages tags for a check. Tags are stored as a dictionary of key-value pairs
+    for key-value tags and as a list of values for keyless tags useful for easy update and deletion.
+    There's an internal cache of the tag list to avoid generating the list of tag strings
+    multiple times.
+    """
+
+    def __init__(self) -> None:
+        self._tags: Dict[Union[str, TagType], List[str]] = {}
+        self._cached_tag_list: Optional[tuple[str, ...]] = None
+        self._keyless: TagType = TagType.KEYLESS
+
+    def set_tag(self, key: Optional[str], value: str, replace: bool = False) -> None:
+        """
+        Set a tag with the given key and value.
+        If key is None or empty, the value is stored as a keyless tag.
+        Args:
+            key (str): The tag key, or None/empty for keyless tags
+            value (str): The tag value
+            replace (bool): If True, replaces all existing values for this key
+                           If False, appends the value if it doesn't exist
+        """
+        if not key:
+            key = self._keyless
+
+        if replace or key not in self._tags:
+            self._tags[key] = [value]
+            # Invalidate the cache since tags have changed
+            self._cached_tag_list = None
+        elif value not in self._tags[key]:
+            self._tags[key].append(value)
+            # Invalidate the cache since tags have changed
+            self._cached_tag_list = None
+
+    def set_tags_from_list(self, tag_list: List[str], replace: bool = False) -> None:
+        """
+        Set multiple tags from a list of strings.
+        Strings can be in "key:value" format or just "value" format.
+        Args:
+            tag_list (List[str]): List of tags in "key:value" format or just "value"
+            replace (bool): If True, replaces all existing tags with the new tags list
+        """
+        if replace:
+            self._tags.clear()
+            self._cached_tag_list = None
+
+        for tag in tag_list:
+            if ':' in tag:
+                key, value = tag.split(':', 1)
+                self.set_tag(key, value)
+            else:
+                self.set_tag(None, tag)
+
+    def delete_tag(self, key: Optional[str], value: Optional[str] = None) -> bool:
+        """
+        Delete a tag or specific value for a tag.
+        For keyless tags, use None or empty string as the key.
+        Args:
+            key (str): The tag key to delete, or None/empty for keyless tags
+            value (str, optional): If provided, only deletes this specific value for the key.
+                                 If None, deletes all values for the key.
+        Returns:
+            bool: True if something was deleted, False otherwise
+        """
+        if not key:
+            key = self._keyless
+
+        if key not in self._tags:
+            return False
+
+        if value is None:
+            # Delete the entire key
+            del self._tags[key]
+            # Invalidate the cache
+            self._cached_tag_list = None
+            return True
+        else:
+            # Delete specific value if it exists
+            if value in self._tags[key]:
+                self._tags[key].remove(value)
+                # Clean up empty lists
+                if not self._tags[key]:
+                    del self._tags[key]
+                # Invalidate the cache
+                self._cached_tag_list = None
+                return True
+        return False
+
+    def _generate_tag_strings(self, tags_dict: Dict[Union[str, TagType], List[str]]) -> tuple[str, ...]:
+        """
+        Generate a tuple of tag strings from a tags dictionary.
+        Args:
+            tags_dict (Dict[Union[str, TagType], List[str]]): Dictionary of tags to convert to strings
+        Returns:
+            tuple[str, ...]: Tuple of tag strings
+        """
+        return tuple(
+            value if key == self._keyless else f"{key}:{value}" for key, values in tags_dict.items() for value in values
+        )
+
+    def get_tags(self) -> List[str]:
+        """
+        Get a list of tag strings.
+        For key-value tags, returns "key:value" format.
+        For keyless tags, returns just the value.
+        The returned list is always sorted alphabetically.
+        Returns:
+            list: Sorted list of tag strings
+        """
+        # Return cached list if available
+        if self._cached_tag_list is not None:
+            return list(self._cached_tag_list)
+
+        # Generate and cache regular tags
+        self._cached_tag_list = self._generate_tag_strings(self._tags)
+        return list(self._cached_tag_list)

{datadog_checks_base-37.12.0.dist-info → datadog_checks_base-37.15.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datadog-checks-base
-Version: 37.12.0
+Version: 37.15.0
 Summary: The Datadog Check Toolkit
 Project-URL: Source, https://github.com/DataDog/integrations-core
 Author-email: Datadog <packages@datadoghq.com>
@@ -16,29 +16,29 @@ Provides-Extra: db
 Requires-Dist: mmh3==5.1.0; extra == 'db'
 Provides-Extra: deps
 Requires-Dist: binary==1.0.1; extra == 'deps'
-Requires-Dist: cachetools==5.5.2; extra == 'deps'
-Requires-Dist: cryptography==44.0.3; extra == 'deps'
-Requires-Dist: ddtrace==3.7.2; extra == 'deps'
+Requires-Dist: cachetools==6.0.0; extra == 'deps'
+Requires-Dist: cryptography==45.0.3; extra == 'deps'
+Requires-Dist: ddtrace==3.9.1; extra == 'deps'
 Requires-Dist: jellyfish==1.2.0; extra == 'deps'
 Requires-Dist: lazy-loader==0.4; extra == 'deps'
-Requires-Dist: prometheus-client==0.21.1; extra == 'deps'
-Requires-Dist: protobuf==6.30.2; extra == 'deps'
-Requires-Dist: pydantic==2.11.4; extra == 'deps'
+Requires-Dist: prometheus-client==0.22.1; extra == 'deps'
+Requires-Dist: protobuf==6.31.1; extra == 'deps'
+Requires-Dist: pydantic==2.11.5; extra == 'deps'
 Requires-Dist: python-dateutil==2.9.0.post0; extra == 'deps'
 Requires-Dist: pywin32==310; (sys_platform == 'win32') and extra == 'deps'
 Requires-Dist: pyyaml==6.0.2; extra == 'deps'
 Requires-Dist: requests-toolbelt==1.0.0; extra == 'deps'
 Requires-Dist: requests-unixsocket2==1.0.0; extra == 'deps'
-Requires-Dist: requests==2.32.3; extra == 'deps'
+Requires-Dist: requests==2.32.4; extra == 'deps'
 Requires-Dist: simplejson==3.20.1; extra == 'deps'
 Requires-Dist: urllib3==2.4.0; extra == 'deps'
 Requires-Dist: wrapt==1.17.2; extra == 'deps'
 Provides-Extra: http
 Requires-Dist: aws-requests-auth==0.4.3; extra == 'http'
-Requires-Dist: botocore==1.38.8; extra == 'http'
+Requires-Dist: botocore==1.38.32; extra == 'http'
 Requires-Dist: oauthlib==3.2.2; extra == 'http'
 Requires-Dist: pyjwt==2.10.1; extra == 'http'
-Requires-Dist: pyopenssl==25.0.0; extra == 'http'
+Requires-Dist: pyopenssl==25.1.0; extra == 'http'
 Requires-Dist: pysocks==1.7.1; extra == 'http'
 Requires-Dist: requests-kerberos==0.15.0; extra == 'http'
 Requires-Dist: requests-ntlm==1.3.0; extra == 'http'

{datadog_checks_base-37.12.0.dist-info → datadog_checks_base-37.15.0.dist-info}/RECORD

@@ -3,7 +3,7 @@ datadog_checks/config.py,sha256=PrAXGdlLnoV2VMQff_noSaSJJ0wg4BAiGnw7jCQLSik,196
 datadog_checks/errors.py,sha256=eFwmnrX-batIgbu-iJyseqAPNO_4rk1UuaKK89evLhg,155
 datadog_checks/log.py,sha256=orvOgMKGNEsqSTLalCAQpWP-ouorpG1A7Gn-j2mRD80,301
 datadog_checks/py.typed,sha256=la67KBlbjXN-_-DfGNcdOcjYumVpKG_Tkw-8n5dnGB4,8
-datadog_checks/base/__about__.py,sha256=0wN6vQOLTtAvgoiMmU7PlRS1klD-UO17t5Mi4SU7w_4,139
+datadog_checks/base/__about__.py,sha256=FWWR7jAyvwkUM_4VgFGv7MU2D_yjzNuII5JoID5lwPQ,139
 datadog_checks/base/__init__.py,sha256=yWegSLE-TZWIGSvAiJj9PSrUxzlOo_UVJLt2zORZ8Ek,363
 datadog_checks/base/__init__.pyi,sha256=eH8XhrtvnD6uE6FWfEyCmKwOaaLJxNolS08D6IRHZuU,995
 datadog_checks/base/agent.py,sha256=nX9x_BYYizRKGNYfXq5z7S0FZ9xcX_wd2tuxpGe3_8k,350
@@ -14,7 +14,8 @@ datadog_checks/base/log.py,sha256=gSfzYimmg0z3Dgmn8NpXuyMNjgXwmTbn0NcyUX8B1ls,62
 datadog_checks/base/types.py,sha256=anajZS0W0TsxUHJQw-JHOP2NSeuC9BisXSy9mAStlxQ,1623
 datadog_checks/base/checks/__init__.py,sha256=q7V6v-FwQWkQC1QWaVzKaPjZMaxPJHJcLd71C0uM7bA,211
 datadog_checks/base/checks/__init__.pyi,sha256=LASfm-daLNQIYe6-w0NPqBw4cl83nYIX5_B-VhV6ARo,262
-datadog_checks/base/checks/base.py,sha256=gIXjMpJPn9j6cBUA8949vPFculWB6bG-V7WvO_ce4RA,59948
+datadog_checks/base/checks/_config_ast.py,sha256=v1rAhwORF80b3kfZKhf6zXZ7S5D3A2QPUK4tSo8eo-Y,3268
+datadog_checks/base/checks/base.py,sha256=q-V8fzkJ3MMbRnhHzhE7wyuolfukXsZ_sq-Ih_PWWkw,59873
 datadog_checks/base/checks/network.py,sha256=UijP1OVBCpCNKMy6bVkNazHgy6Sdm8qCvCYGPaWuTDo,1968
 datadog_checks/base/checks/kube_leader/__init__.py,sha256=q7V6v-FwQWkQC1QWaVzKaPjZMaxPJHJcLd71C0uM7bA,211
 datadog_checks/base/checks/kube_leader/__init__.pyi,sha256=UGDywoRwmCIz3Zii1uHsp7jiFGWRdn5fFMZZxgGGlQs,398
@@ -143,7 +144,7 @@ datadog_checks/base/utils/db/statement_metrics.py,sha256=U7EtERkmFzfCtfyd3094fBa
 datadog_checks/base/utils/db/timed_cache.py,sha256=a9Ks5KKUvExB6GOATXTSCLamVtLD919Dn6HpweGKtFw,2114
 datadog_checks/base/utils/db/transform.py,sha256=kNdiBZVoZm1oNRNWIatrAfgORjbkRKhs1AgW_aQCj3I,23879
 datadog_checks/base/utils/db/types.py,sha256=OLX2Oq58JQPFBD4oqUpCLkAP7ovRGN_i1vFk1E0N8Lg,267
-datadog_checks/base/utils/db/utils.py,sha256=hTPEtamyPN4x9tCE9BuHbd6d523TPxw8DojBuJO2qf0,16934
+datadog_checks/base/utils/db/utils.py,sha256=z1CIC_VdMHHYuvoRHZKpdnk_bHj-_M07JFSPNMjIUP4,21657
 datadog_checks/base/utils/discovery/__init__.py,sha256=vPCOdsThBcBjFJRPhDm6IsZGOwk8HlvciwCe_l8dKLk,211
 datadog_checks/base/utils/discovery/__init__.pyi,sha256=ScVLU1Njj9ekZmewltb0cULI6BylssVHfn4CcPNeyr8,173
 datadog_checks/base/utils/discovery/cache.py,sha256=f9L3A7YZpZ-mpZpFIwjsa5ab9cZMGkqdetdr9EpalbI,887
@@ -217,6 +218,6 @@ datadog_checks/utils/tracing.py,sha256=HQbQakKM-Lw75MDkItaYJYipS6YO24Z_ymDVxDsx5
 datadog_checks/utils/prometheus/__init__.py,sha256=8WwXnM9g1sfS5267QYCJX_hd8MZl5kRgBgQ_SzdNdXs,161
 datadog_checks/utils/prometheus/functions.py,sha256=4vWsTGLgujHwdYZo0tlAQkqDPHofqUJM3k9eItJqERQ,197
 datadog_checks/utils/prometheus/metrics_pb2.py,sha256=xg3UdUHe4TjeR4s13LUKZ2U1WVSt6U6zjsVRG6lX6dc,173
-datadog_checks_base-37.12.0.dist-info/METADATA,sha256=C9JrO70hO0CXDtJsgKBCNXFzrZzJs-Zw8JSXPlGu5ek,4244
-datadog_checks_base-37.12.0.dist-info/WHEEL,sha256=tkmg4JIqwd9H8mL30xA7crRmoStyCtGp0VWshokd1Jc,105
-datadog_checks_base-37.12.0.dist-info/RECORD,,
+datadog_checks_base-37.15.0.dist-info/METADATA,sha256=CUQ731u-GuA4NgGfXFBBLL5wAjuvDdbsqIC3SYgbT34,4245
+datadog_checks_base-37.15.0.dist-info/WHEEL,sha256=tkmg4JIqwd9H8mL30xA7crRmoStyCtGp0VWshokd1Jc,105
+datadog_checks_base-37.15.0.dist-info/RECORD,,