datadog-checks-base 37.12.0__py2.py3-none-any.whl → 37.13.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.13.0"

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.13.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datadog-checks-base
-Version: 37.12.0
+Version: 37.13.0
 Summary: The Datadog Check Toolkit
 Project-URL: Source, https://github.com/DataDog/integrations-core
 Author-email: Datadog <packages@datadoghq.com>
@@ -16,14 +16,14 @@ 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.8.0; 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.0; extra == 'deps'
+Requires-Dist: protobuf==6.31.0; 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'
@@ -35,10 +35,10 @@ 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.24; 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.13.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=lB4-_xYUmOUSD8CzIo0yXKB4H2tiej7dQ2fEkgCdyHw,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
@@ -143,7 +143,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 +217,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.13.0.dist-info/METADATA,sha256=fEpcawGlBAaVY_ESNFrALlVN9FGTQBSt5fUByiRAPA0,4245
+datadog_checks_base-37.13.0.dist-info/WHEEL,sha256=tkmg4JIqwd9H8mL30xA7crRmoStyCtGp0VWshokd1Jc,105
+datadog_checks_base-37.13.0.dist-info/RECORD,,