datadog-checks-base 37.5.0__py2.py3-none-any.whl → 37.6.1__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.5.0"
+__version__ = "37.6.1"

datadog_checks/base/checks/openmetrics/v2/labels.py

@@ -1,21 +1,27 @@
 # (C) Datadog, Inc. 2020-present
 # All rights reserved
 # Licensed under a 3-clause BSD style license (see LICENSE)
+from collections import ChainMap
+
 from ....utils.functions import no_op
 
 
 class LabelAggregator:
     def __init__(self, check, config):
         share_labels = config.get('share_labels', {})
-        if not isinstance(share_labels, dict):
-            raise TypeError('Setting `share_labels` must be a mapping')
-        elif not share_labels:
+        self.target_info = config.get('target_info', False)
+        self.target_info_labels = {}
+
+        self._validate_type(share_labels, dict, "Setting `share_labels` must be a mapping")
+        self._validate_type(self.target_info, bool, "Setting `target_info` must be a boolean")
+
+        if not share_labels and not self.target_info:
             self.populate = no_op
             return
 
         self.cache_shared_labels = config.get('cache_shared_labels', True)
         self.shared_labels_cached = False
-
+        self.info_metric = {'target_info': {}}
         self.metric_config = {}
         for metric, config in share_labels.items():
             data = self.metric_config[metric] = {}
@@ -31,6 +37,7 @@ class LabelAggregator:
                     raise TypeError(f'Option `values` for metric `{metric}` of setting `share_labels` must be an array')
 
                 allowed_values = set()
+
                 for i, value in enumerate(values, 1):
                     value = str(value)
 
@@ -72,34 +79,32 @@ class LabelAggregator:
 
         self.unconditional_labels = {}
 
+    def _validate_type(self, value, expected_type, error_message):
+        if not isinstance(value, expected_type):
+            raise TypeError(error_message)
+
     def __call__(self, metrics):
         if self.cache_shared_labels:
             if self.shared_labels_cached:
                 yield from metrics
             else:
-                metric_config = self.metric_config.copy()
+                metric_config, target_info_metric = self.copy_configs()
 
                 for metric in metrics:
-                    if metric_config and metric.name in metric_config:
-                        self.collect(metric, metric_config.pop(metric.name))
-
+                    self.process_metric(metric, metric_config, target_info_metric)
                     yield metric
 
                 self.shared_labels_cached = True
         else:
             try:
-                metric_config = self.metric_config.copy()
-
-                # Cache every encountered metric until the desired labels have been collected
+                metric_config, target_info_metric = self.copy_configs()
                 cached_metrics = []
 
                 for metric in metrics:
-                    if metric.name in metric_config:
-                        self.collect(metric, metric_config.pop(metric.name))
-
+                    self.process_metric(metric, metric_config, target_info_metric)
                     cached_metrics.append(metric)
 
-                    if not metric_config:
+                    if not (metric_config or target_info_metric):
                         break
 
                 yield from cached_metrics
@@ -108,6 +113,24 @@ class LabelAggregator:
                 self.label_sets.clear()
                 self.unconditional_labels.clear()
 
+    def copy_configs(self):
+        return self.metric_config.copy(), self.info_metric.copy()
+
+    def process_metric(self, metric, *configs):
+        """
+        Collects labels from shared_labels + target_info metrics
+        """
+        for config in configs:
+            if config and metric.name in config:
+                self.collect(metric, config.pop(metric.name))
+
+    def process_target_info(self, metric):
+        """
+        Updates cached target info metrics
+        """
+        if metric.samples[0].labels != self.target_info_labels:
+            self.target_info_labels = metric.samples[0].labels
+
     def collect(self, metric, config):
         allowed_values = config.get('values')
 
@@ -147,13 +170,28 @@ class LabelAggregator:
                         if label in labels:
                             self.unconditional_labels[label] = value
             else:
-                for sample in self.allowed_samples(metric, allowed_values):
-                    for label, value in sample.labels.items():
-                        self.unconditional_labels[label] = value
+                # Store target_info metric labels to be applied to other metrics in payload
+                if metric.name == 'target_info':
+                    self.target_info_labels.update(
+                        {
+                            label: value
+                            for sample in self.allowed_samples(metric, allowed_values)
+                            for label, value in sample.labels.items()
+                        }
+                    )
+                else:
+                    # Store shared labels in a seperate attribute
+                    self.unconditional_labels.update(
+                        {
+                            label: value
+                            for sample in self.allowed_samples(metric, allowed_values)
+                            for label, value in sample.labels.items()
+                        }
+                    )
 
     def populate(self, labels):
         label_set = frozenset(labels.items())
-        labels.update(self.unconditional_labels)
+        labels.update(ChainMap(self.unconditional_labels, self.target_info_labels))
 
         for matching_label_set, shared_labels in self.label_sets:
             # Check for subset without incurring the cost of a `.issubset` lookup and call

datadog_checks/base/checks/openmetrics/v2/scraper.py

@@ -58,6 +58,7 @@ class OpenMetricsScraper:
 
         # Parse the configuration
         self.endpoint = config['openmetrics_endpoint']
+        self.target_info = config.get('target_info', False)
 
         self.metric_transformer = MetricTransformer(self.check, config)
         self.label_aggregator = LabelAggregator(self.check, config)
@@ -238,7 +239,13 @@ class OpenMetricsScraper:
         """
         runtime_data = {'flush_first_value': self.flush_first_value, 'static_tags': self.static_tags}
 
-        for metric in self.consume_metrics(runtime_data):
+        # Determine which consume method to use based on target_info config
+        if self.target_info:
+            consume_method = self.consume_metrics_w_target_info
+        else:
+            consume_method = self.consume_metrics
+
+        for metric in consume_method(runtime_data):
             transformer = self.metric_transformer.get(metric)
             if transformer is None:
                 continue
@@ -248,23 +255,52 @@ class OpenMetricsScraper:
         self.flush_first_value = True
 
     def consume_metrics(self, runtime_data):
+        """
+        Yield the processed metrics and filter out excluded metrics, without checking for target_info metrics.
+        """
+
+        metric_parser = self.parse_metrics()
+
+        if not self.flush_first_value and self.use_process_start_time:
+            metric_parser = first_scrape_handler(metric_parser, runtime_data, datadog_agent.get_process_start_time())
+        if self.label_aggregator.configured:
+            metric_parser = self.label_aggregator(metric_parser)
+
+        for metric in metric_parser:
+            # Skip excluded metrics
+            if metric.name in self.exclude_metrics or (
+                self.exclude_metrics_pattern is not None and self.exclude_metrics_pattern.search(metric.name)
+            ):
+                self.submit_telemetry_number_of_ignored_metric_samples(metric)
+                continue
+
+            yield metric
+
+    def consume_metrics_w_target_info(self, runtime_data):
         """
         Yield the processed metrics and filter out excluded metrics.
+        Additionally, handle target_info metrics.
         """
 
         metric_parser = self.parse_metrics()
+
         if not self.flush_first_value and self.use_process_start_time:
             metric_parser = first_scrape_handler(metric_parser, runtime_data, datadog_agent.get_process_start_time())
         if self.label_aggregator.configured:
             metric_parser = self.label_aggregator(metric_parser)
 
         for metric in metric_parser:
+            # Skip excluded metrics
             if metric.name in self.exclude_metrics or (
                 self.exclude_metrics_pattern is not None and self.exclude_metrics_pattern.search(metric.name)
             ):
                 self.submit_telemetry_number_of_ignored_metric_samples(metric)
                 continue
 
+            # Process target_info metrics
+            if metric.name == 'target_info':
+                self.label_aggregator.process_target_info(metric)
+
             yield metric
 
     def parse_metrics(self):
@@ -312,7 +348,6 @@ class OpenMetricsScraper:
         """
         Yield a sample of processed data.
         """
-
         label_normalizer = get_label_normalizer(metric.type)
 
         for sample in metric.samples:

{datadog_checks_base-37.5.0.dist-info → datadog_checks_base-37.6.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datadog-checks-base
-Version: 37.5.0
+Version: 37.6.1
 Summary: The Datadog Check Toolkit
 Project-URL: Source, https://github.com/DataDog/integrations-core
 Author-email: Datadog <packages@datadoghq.com>
@@ -13,16 +13,16 @@ Classifier: License :: OSI Approved :: BSD License
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Topic :: System :: Monitoring
 Provides-Extra: db
-Requires-Dist: mmh3==5.0.1; 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.0; extra == 'deps'
+Requires-Dist: cachetools==5.5.1; extra == 'deps'
 Requires-Dist: cryptography==43.0.1; extra == 'deps'
 Requires-Dist: ddtrace==2.10.6; extra == 'deps'
 Requires-Dist: jellyfish==1.1.3; extra == 'deps'
 Requires-Dist: prometheus-client==0.21.1; extra == 'deps'
 Requires-Dist: protobuf==5.29.3; extra == 'deps'
-Requires-Dist: pydantic==2.10.5; extra == 'deps'
+Requires-Dist: pydantic==2.10.6; extra == 'deps'
 Requires-Dist: python-dateutil==2.9.0.post0; extra == 'deps'
 Requires-Dist: pywin32==308; (sys_platform == 'win32') and extra == 'deps'
 Requires-Dist: pyyaml==6.0.2; extra == 'deps'
@@ -34,7 +34,7 @@ Requires-Dist: uptime==3.0.1; 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.36.1; extra == 'http'
+Requires-Dist: botocore==1.36.16; extra == 'http'
 Requires-Dist: oauthlib==3.2.2; extra == 'http'
 Requires-Dist: pyjwt==2.10.1; extra == 'http'
 Requires-Dist: pyopenssl==24.2.1; extra == 'http'
@@ -43,9 +43,9 @@ Requires-Dist: requests-kerberos==0.15.0; extra == 'http'
 Requires-Dist: requests-ntlm==1.3.0; extra == 'http'
 Requires-Dist: requests-oauthlib==2.0.0; extra == 'http'
 Provides-Extra: json
-Requires-Dist: orjson==3.10.14; extra == 'json'
+Requires-Dist: orjson==3.10.15; extra == 'json'
 Provides-Extra: kube
-Requires-Dist: kubernetes==31.0.0; extra == 'kube'
+Requires-Dist: kubernetes==32.0.0; extra == 'kube'
 Requires-Dist: requests-oauthlib==2.0.0; extra == 'kube'
 Description-Content-Type: text/markdown
 

{datadog_checks_base-37.5.0.dist-info → datadog_checks_base-37.6.1.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=5AzFfItx9URe8lFWie_oC3dNfFosQs7be2xVIkuGnko,138
+datadog_checks/base/__about__.py,sha256=L56e6SJ2L24q3EfI391-ib0_nBXjubulAGFIAmNCTgA,138
 datadog_checks/base/__init__.py,sha256=rSTDo6-2p_RX6I6wnuXrNMglaatoLRyZFTE5wbHZY8s,1466
 datadog_checks/base/agent.py,sha256=nX9x_BYYizRKGNYfXq5z7S0FZ9xcX_wd2tuxpGe3_8k,350
 datadog_checks/base/config.py,sha256=qcAA4X9sXQZRdwQe8DgiGd2980VBp1SQA0d695tX_tU,604
@@ -38,9 +38,9 @@ datadog_checks/base/checks/openmetrics/mixins.py,sha256=bkrE_cQ6peW147BACY-G0tHy
 datadog_checks/base/checks/openmetrics/v2/__init__.py,sha256=P81FBhJLA9MGpz4L7o_7C-pvEwd4A_8jWFWQZ1jg0Eg,115
 datadog_checks/base/checks/openmetrics/v2/base.py,sha256=UXytG0ze00rXU13VB0BM_ZK_Y_qC6wfxA-OidFm_MyY,4096
 datadog_checks/base/checks/openmetrics/v2/first_scrape_handler.py,sha256=In-tZXMzieSdL2OXJlIAFi78KbQVwuJnGP0iaLH9974,941
-datadog_checks/base/checks/openmetrics/v2/labels.py,sha256=0F0ISh03wS2vVsjbns3FrYDn6gvxj0Xv-2fkA2Hh9D8,7310
+datadog_checks/base/checks/openmetrics/v2/labels.py,sha256=JAIlxAtzQAFoEj1d9IYLZCGt-2M3-ClyyGjZ-VzKNQY,8907
 datadog_checks/base/checks/openmetrics/v2/metrics.py,sha256=uDVZOHHRWMW8YUQZBY9jCLzecN5DueRNjImxBm1K82k,2189
-datadog_checks/base/checks/openmetrics/v2/scraper.py,sha256=d4PcJM_b7wkHUBQ5b_IZo_VW_07jQj_gwTgfDrkp1BA,23279
+datadog_checks/base/checks/openmetrics/v2/scraper.py,sha256=nj7dmKuHeoI_A8bWUzZh16Q3r94-27dV7FepvdOnv9I,24733
 datadog_checks/base/checks/openmetrics/v2/transform.py,sha256=qv7Y1yvqWm-269mGU7Oq0it8x3VkWUZU2m2d3q-VJZ0,8377
 datadog_checks/base/checks/openmetrics/v2/utils.py,sha256=tpk3htJAz_KwCRqFs2CTjajHkLCs_2TbGdBp514rWOQ,3565
 datadog_checks/base/checks/openmetrics/v2/transformers/__init__.py,sha256=hPZn8WLvQfrzLSaJxtFbYYAtK1dt8IQXSULL79srZjs,493
@@ -200,6 +200,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.5.0.dist-info/METADATA,sha256=GZWGd2TVn8CrN5OELUHpk6gXVM5OejdzkZnkcHXfpWc,3590
-datadog_checks_base-37.5.0.dist-info/WHEEL,sha256=tkmg4JIqwd9H8mL30xA7crRmoStyCtGp0VWshokd1Jc,105
-datadog_checks_base-37.5.0.dist-info/RECORD,,
+datadog_checks_base-37.6.1.dist-info/METADATA,sha256=iMylQRYC88icbKrkMES-dTg1SzDEhgsInJI-6c_Vh0o,3591
+datadog_checks_base-37.6.1.dist-info/WHEEL,sha256=tkmg4JIqwd9H8mL30xA7crRmoStyCtGp0VWshokd1Jc,105
+datadog_checks_base-37.6.1.dist-info/RECORD,,