datadog-checks-base 37.13.0__py2.py3-none-any.whl → 37.16.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.13.0"
+__version__ = "37.16.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

@@ -29,31 +29,32 @@ from typing import (  # noqa: F401
 import lazy_loader
 
 from datadog_checks.base.agent import AGENT_RUNNING, aggregator, datadog_agent
+from datadog_checks.base.config import is_affirmative
+from datadog_checks.base.constants import ServiceCheck
+from datadog_checks.base.errors import ConfigurationError
+from datadog_checks.base.types import (
+    AgentConfigType,
+    Event,
+    ExternalTagType,
+    InitConfigType,
+    InstanceType,
+    ProxySettings,
+    ServiceCheckStatus,
+)
+from datadog_checks.base.utils.agent.utils import should_profile_memory
+from datadog_checks.base.utils.common import ensure_bytes, to_native_string
+from datadog_checks.base.utils.fips import enable_fips
 from datadog_checks.base.utils.format import json
+from datadog_checks.base.utils.tagging import GENERIC_TAGS
+from datadog_checks.base.utils.tracing import traced_class
 
-from ..config import is_affirmative
-from ..constants import ServiceCheck
-from ..errors import ConfigurationError
-from ..types import (
-    AgentConfigType,  # noqa: F401
-    Event,  # noqa: F401
-    ExternalTagType,  # noqa: F401
-    InitConfigType,  # noqa: F401
-    InstanceType,  # noqa: F401
-    ProxySettings,  # noqa: F401
-    ServiceCheckStatus,  # noqa: F401
-)
-from ..utils.agent.utils import should_profile_memory
-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
+    from datadog_checks.base.log import CheckLoggingAdapter, init_logging
 
 else:
-    from ..stubs.log import CheckLoggingAdapter, init_logging
+    from datadog_checks.base.stubs.log import CheckLoggingAdapter, init_logging
 
 init_logging()
 
@@ -146,12 +147,12 @@ class AgentCheck(object):
     # a mapping type, then each key will be considered a `name` and will be sent with its (str) value.
     METADATA_TRANSFORMERS = None
 
-    FIRST_CAP_RE = re.compile(br'(.)([A-Z][a-z]+)')
-    ALL_CAP_RE = re.compile(br'([a-z0-9])([A-Z])')
-    METRIC_REPLACEMENT = re.compile(br'([^a-zA-Z0-9_.]+)|(^[^a-zA-Z]+)')
-    TAG_REPLACEMENT = re.compile(br'[,\+\*\-/()\[\]{}\s]')
-    MULTIPLE_UNDERSCORE_CLEANUP = re.compile(br'__+')
-    DOT_UNDERSCORE_CLEANUP = re.compile(br'_*\._*')
+    FIRST_CAP_RE = re.compile(rb'(.)([A-Z][a-z]+)')
+    ALL_CAP_RE = re.compile(rb'([a-z0-9])([A-Z])')
+    METRIC_REPLACEMENT = re.compile(rb'([^a-zA-Z0-9_.]+)|(^[^a-zA-Z]+)')
+    TAG_REPLACEMENT = re.compile(rb'[,\+\*\-/()\[\]{}\s]')
+    MULTIPLE_UNDERSCORE_CLEANUP = re.compile(rb'__+')
+    DOT_UNDERSCORE_CLEANUP = re.compile(rb'_*\._*')
 
     # allows to set a limit on the number of metric name and tags combination
     # this check can send per run. This is useful for checks that have an unbounded
@@ -1145,10 +1146,10 @@ class AgentCheck(object):
         And substitute illegal metric characters
         """
         name = ensure_bytes(name)
-        metric_name = self.FIRST_CAP_RE.sub(br'\1_\2', name)
-        metric_name = self.ALL_CAP_RE.sub(br'\1_\2', metric_name).lower()
-        metric_name = self.METRIC_REPLACEMENT.sub(br'_', metric_name)
-        return self.DOT_UNDERSCORE_CLEANUP.sub(br'.', metric_name).strip(b'_')
+        metric_name = self.FIRST_CAP_RE.sub(rb'\1_\2', name)
+        metric_name = self.ALL_CAP_RE.sub(rb'\1_\2', metric_name).lower()
+        metric_name = self.METRIC_REPLACEMENT.sub(rb'_', metric_name)
+        return self.DOT_UNDERSCORE_CLEANUP.sub(rb'.', metric_name).strip(b'_')
 
     def warning(self, warning_message, *args, **kwargs):
         # type: (str, *Any, **Any) -> None
@@ -1242,10 +1243,10 @@ class AgentCheck(object):
             if prefix is not None:
                 prefix = self.convert_to_underscore_separated(prefix)
         else:
-            name = self.METRIC_REPLACEMENT.sub(br'_', metric)
-            name = self.DOT_UNDERSCORE_CLEANUP.sub(br'.', name).strip(b'_')
+            name = self.METRIC_REPLACEMENT.sub(rb'_', metric)
+            name = self.DOT_UNDERSCORE_CLEANUP.sub(rb'.', name).strip(b'_')
 
-        name = self.MULTIPLE_UNDERSCORE_CLEANUP.sub(br'_', name)
+        name = self.MULTIPLE_UNDERSCORE_CLEANUP.sub(rb'_', name)
 
         if prefix is not None:
             name = ensure_bytes(prefix) + b"." + name
@@ -1261,9 +1262,9 @@ class AgentCheck(object):
         """
         if isinstance(tag, str):
             tag = tag.encode('utf-8', 'ignore')
-        tag = self.TAG_REPLACEMENT.sub(br'_', tag)
-        tag = self.MULTIPLE_UNDERSCORE_CLEANUP.sub(br'_', tag)
-        tag = self.DOT_UNDERSCORE_CLEANUP.sub(br'.', tag).strip(b'_')
+        tag = self.TAG_REPLACEMENT.sub(rb'_', tag)
+        tag = self.MULTIPLE_UNDERSCORE_CLEANUP.sub(rb'_', tag)
+        tag = self.DOT_UNDERSCORE_CLEANUP.sub(rb'.', tag).strip(b'_')
         return to_native_string(tag)
 
     def check(self, instance):
@@ -1286,7 +1287,7 @@ class AgentCheck(object):
             self._clear_diagnosis()
             # Ignore check initializations if running in a separate process
             if is_affirmative(self.instance.get('process_isolation', self.init_config.get('process_isolation', False))):
-                from ..utils.replay.execute import run_with_isolation
+                from datadog_checks.base.utils.replay.execute import run_with_isolation
 
                 run_with_isolation(self, aggregator, datadog_agent)
             else:
@@ -1301,7 +1302,7 @@ class AgentCheck(object):
                 instance = copy.deepcopy(self.instances[0])
 
                 if 'set_breakpoint' in self.init_config:
-                    from ..utils.agent.debug import enter_pdb
+                    from datadog_checks.base.utils.agent.debug import enter_pdb
 
                     enter_pdb(self.check, line=self.init_config['set_breakpoint'], args=(instance,))
                 elif self.should_profile_memory():
@@ -1456,7 +1457,7 @@ class AgentCheck(object):
 
     def profile_memory(self, func, namespaces=None, args=(), kwargs=None, extra_tags=None):
         # type: (Callable[..., Any], Optional[Sequence[str]], Sequence[Any], Optional[Dict[str, Any]], Optional[List[str]]) -> None  # noqa: E501
-        from ..utils.agent.memory import profile_memory
+        from datadog_checks.base.utils.agent.memory import profile_memory
 
         if namespaces is None:
             namespaces = self.check_id.split(':', 1)
@@ -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/checks/kube_leader/base_check.py

@@ -1,7 +1,8 @@
 # (C) Datadog, Inc. 2018-present
 # All rights reserved
 # Licensed under a 3-clause BSD style license (see LICENSE)
-from .. import AgentCheck
+from datadog_checks.base.checks import AgentCheck
+
 from .mixins import KubeLeaderElectionMixin
 
 

datadog_checks/base/checks/kube_leader/mixins.py

@@ -2,8 +2,8 @@
 # All rights reserved
 # Licensed under a 3-clause BSD style license (see LICENSE)
 from datadog_checks.base.agent import datadog_agent
+from datadog_checks.base.checks import AgentCheck
 
-from .. import AgentCheck
 from .record import ElectionRecordAnnotation, ElectionRecordLease
 
 # Import lazily to reduce memory footprint

datadog_checks/base/checks/kubelet_base/base.py

@@ -4,8 +4,8 @@
 import json
 from datetime import datetime, timedelta, timezone
 
-from ...utils.date import parse_rfc3339
-from .. import AgentCheck
+from datadog_checks.base.checks import AgentCheck
+from datadog_checks.base.utils.date import parse_rfc3339
 
 try:
     from datadog_agent import get_config

datadog_checks/base/checks/network.py

@@ -33,7 +33,7 @@ class NetworkCheck(AgentCheck):
         try:
             statuses = self._check(instance)
         except Exception:
-            self.log.exception(u"Failed to run instance '%s'.", instance.get('name', u""))
+            self.log.exception("Failed to run instance '%s'.", instance.get('name', ""))
         else:
             if isinstance(statuses, tuple):
                 # Assume the check only returns one service check

datadog_checks/base/checks/openmetrics/base_check.py

@@ -5,9 +5,10 @@ from copy import deepcopy
 
 import requests
 
-from ...errors import CheckException
-from ...utils.tracing import traced_class
-from .. import AgentCheck
+from datadog_checks.base.checks import AgentCheck
+from datadog_checks.base.errors import CheckException
+from datadog_checks.base.utils.tracing import traced_class
+
 from .mixins import OpenMetricsScraperMixin
 
 STANDARD_FIELDS = [

datadog_checks/base/checks/openmetrics/mixins.py

@@ -14,13 +14,12 @@ import requests
 from prometheus_client.samples import Sample
 
 from datadog_checks.base.agent import datadog_agent
-
-from ...config import is_affirmative
-from ...errors import CheckException
-from ...utils.common import to_native_string
-from ...utils.http import RequestsWrapper
-from .. import AgentCheck
-from ..libs.prometheus import text_fd_to_metric_families
+from datadog_checks.base.checks import AgentCheck
+from datadog_checks.base.checks.libs.prometheus import text_fd_to_metric_families
+from datadog_checks.base.config import is_affirmative
+from datadog_checks.base.errors import CheckException
+from datadog_checks.base.utils.common import to_native_string
+from datadog_checks.base.utils.http import RequestsWrapper
 
 
 class OpenMetricsScraperMixin(object):

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

@@ -6,9 +6,10 @@ from contextlib import contextmanager
 
 from requests.exceptions import RequestException
 
-from ....errors import ConfigurationError
-from ....utils.tracing import traced_class
-from ... import AgentCheck
+from datadog_checks.base.checks import AgentCheck
+from datadog_checks.base.errors import ConfigurationError
+from datadog_checks.base.utils.tracing import traced_class
+
 from .scraper import OpenMetricsScraper
 
 

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

@@ -3,7 +3,7 @@
 # Licensed under a 3-clause BSD style license (see LICENSE)
 from collections import ChainMap
 
-from ....utils.functions import no_op
+from datadog_checks.base.utils.functions import no_op
 
 
 class LabelAggregator:

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

@@ -0,0 +1,8 @@
+# (C) Datadog, Inc. 2025-present
+# All rights reserved
+# Licensed under a 3-clause BSD style license (see LICENSE)
+
+from . import decorators
+from .base_scraper import OpenMetricsCompatibilityScraper, OpenMetricsScraper
+
+__all__ = ["OpenMetricsScraper", "OpenMetricsCompatibilityScraper", "decorators"]

datadog_checks/base/checks/openmetrics/v2/{scraper.py → scraper/base_scraper.py}

@@ -4,25 +4,26 @@
 import fnmatch
 import inspect
 import re
+from collections.abc import Generator
 from copy import copy, deepcopy
 from itertools import chain
 from math import isinf, isnan
 from typing import List  # noqa: F401
 
+from prometheus_client import Metric
 from prometheus_client.openmetrics.parser import text_fd_to_metric_families as parse_openmetrics
 from prometheus_client.parser import text_fd_to_metric_families as parse_prometheus
 from requests.exceptions import ConnectionError
 
 from datadog_checks.base.agent import datadog_agent
-
-from ....config import is_affirmative
-from ....constants import ServiceCheck
-from ....errors import ConfigurationError
-from ....utils.functions import no_op, return_true
-from ....utils.http import RequestsWrapper
-from .first_scrape_handler import first_scrape_handler
-from .labels import LabelAggregator, get_label_normalizer
-from .transform import MetricTransformer
+from datadog_checks.base.checks.openmetrics.v2.first_scrape_handler import first_scrape_handler
+from datadog_checks.base.checks.openmetrics.v2.labels import LabelAggregator, get_label_normalizer
+from datadog_checks.base.checks.openmetrics.v2.transform import MetricTransformer
+from datadog_checks.base.config import is_affirmative
+from datadog_checks.base.constants import ServiceCheck
+from datadog_checks.base.errors import ConfigurationError
+from datadog_checks.base.utils.functions import no_op, return_true
+from datadog_checks.base.utils.http import RequestsWrapper
 
 
 class OpenMetricsScraper:
@@ -238,19 +239,21 @@ class OpenMetricsScraper:
         """
         runtime_data = {'flush_first_value': bool(self.flush_first_value), 'static_tags': self.static_tags}
 
-        # 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):
+        for metric in self.yield_metrics(runtime_data):
             transformer = self.metric_transformer.get(metric)
             if transformer is None:
                 continue
 
             transformer(metric, self.generate_sample_data(metric), runtime_data)
 
+    def yield_metrics(self, runtime_data: dict) -> Generator[Metric]:
+        if self.target_info:
+            consume_method = self.consume_metrics_w_target_info
+        else:
+            consume_method = self.consume_metrics
+
+        yield from consume_method(runtime_data)
+
     def scrape(self):
         try:
             self._scrape()

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

@@ -0,0 +1,48 @@
+# (C) Datadog, Inc. 2025-present
+# All rights reserved
+# Licensed under a 3-clause BSD style license (see LICENSE)
+
+from __future__ import annotations
+
+from functools import partial
+from typing import TYPE_CHECKING, Any
+
+from .base_scraper import OpenMetricsScraper
+
+if TYPE_CHECKING:
+    from collections.abc import Generator
+
+    from prometheus_client.metrics_core import Metric
+
+
+class WithHttpCodeClass(OpenMetricsScraper):
+    """
+    Scraper decorator that parses the HTTP status code from the metric and adds a new tag named
+    `code_class` to the metric.
+
+    The HTTP status code is parsed and a new tag named `code_class` is added to the metric
+    stating whether the status code is in the 1xx, 2xx, 3xx, 4xx, or 5xx range.
+    """
+
+    def __init__(self, scraper: OpenMetricsScraper, http_status_tag: str):
+        self.scraper = scraper
+        self.http_status_tag = http_status_tag
+        self.decorated_methods = {"yield_metrics": self.yield_metrics}
+
+    def __getattr__(self, name: str) -> Any:
+        return self.decorated_methods.get(name, getattr(self.scraper, name))
+
+    def _add_http_code_class(self, metric: Metric, http_status_tag: str) -> Metric:
+        for sample in metric.samples:
+            if (
+                (code := sample.labels.get(http_status_tag))
+                and isinstance(code, str)
+                and len(code) == 3
+                and code.isdigit()
+            ):
+                sample.labels["code_class"] = f"{code[0]}xx"
+        return metric
+
+    def yield_metrics(self, runtime_data: dict[str, Any]) -> Generator[Metric]:
+        add_http_code_class_func = partial(self._add_http_code_class, http_status_tag=self.http_status_tag)
+        yield from map(add_http_code_class_func, self.scraper.yield_metrics(runtime_data))

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

@@ -4,7 +4,8 @@
 import re
 from copy import deepcopy
 
-from ....config import is_affirmative
+from datadog_checks.base.config import is_affirmative
+
 from . import transformers
 
 DEFAULT_METRIC_TYPE = 'native'

datadog_checks/base/checks/openmetrics/v2/transformers/histogram.py

@@ -1,8 +1,8 @@
 # (C) Datadog, Inc. 2020-present
 # All rights reserved
 # Licensed under a 3-clause BSD style license (see LICENSE)
-from ..labels import canonicalize_numeric_label
-from ..utils import decumulate_histogram_buckets
+from datadog_checks.base.checks.openmetrics.v2.labels import canonicalize_numeric_label
+from datadog_checks.base.checks.openmetrics.v2.utils import decumulate_histogram_buckets
 
 
 def get_histogram(check, metric_name, modifiers, global_options):

datadog_checks/base/checks/openmetrics/v2/transformers/service_check.py

@@ -1,7 +1,7 @@
 # (C) Datadog, Inc. 2020-present
 # All rights reserved
 # Licensed under a 3-clause BSD style license (see LICENSE)
-from .....constants import ServiceCheck
+from datadog_checks.base.constants import ServiceCheck
 
 
 def get_service_check(check, metric_name, modifiers, global_options):

datadog_checks/base/checks/openmetrics/v2/transformers/temporal_percent.py

@@ -1,8 +1,8 @@
 # (C) Datadog, Inc. 2020-present
 # All rights reserved
 # Licensed under a 3-clause BSD style license (see LICENSE)
-from .....utils.common import total_time_to_temporal_percent
-from .....utils.constants import TIME_UNITS
+from datadog_checks.base.utils.common import total_time_to_temporal_percent
+from datadog_checks.base.utils.constants import TIME_UNITS
 
 
 def get_temporal_percent(check, metric_name, modifiers, global_options):

datadog_checks/base/checks/openmetrics/v2/transformers/time_elapsed.py

@@ -1,7 +1,7 @@
 # (C) Datadog, Inc. 2020-present
 # All rights reserved
 # Licensed under a 3-clause BSD style license (see LICENSE)
-from .....utils.time import get_timestamp
+from datadog_checks.base.utils.time import get_timestamp
 
 
 def get_time_elapsed(check, metric_name, modifiers, global_options):

datadog_checks/base/checks/prometheus/base_check.py

@@ -1,9 +1,10 @@
 # (C) Datadog, Inc. 2018-present
 # All rights reserved
 # Licensed under a 3-clause BSD style license (see LICENSE)
-from ...errors import CheckException
-from ...utils.common import to_native_string
-from .. import AgentCheck
+from datadog_checks.base.checks import AgentCheck
+from datadog_checks.base.errors import CheckException
+from datadog_checks.base.utils.common import to_native_string
+
 from .mixins import PrometheusScraperMixin
 
 

datadog_checks/base/checks/prometheus/mixins.py

@@ -11,11 +11,11 @@ from math import isinf, isnan
 import requests
 from google.protobuf.internal.decoder import _DecodeVarint32  # pylint: disable=E0611,E0401
 
-from ...config import is_affirmative
-from ...utils.http import RequestsWrapper
-from ...utils.prometheus import metrics_pb2
-from .. import AgentCheck
-from ..libs.prometheus import text_fd_to_metric_families
+from datadog_checks.base.checks import AgentCheck
+from datadog_checks.base.checks.libs.prometheus import text_fd_to_metric_families
+from datadog_checks.base.config import is_affirmative
+from datadog_checks.base.utils.http import RequestsWrapper
+from datadog_checks.base.utils.prometheus import metrics_pb2
 
 
 class PrometheusFormat:
@@ -561,7 +561,7 @@ class PrometheusScraperMixin(object):
             headers['Accept-Encoding'] = 'gzip'
         if pFormat == PrometheusFormat.PROTOBUF:
             headers['accept'] = (
-                'application/vnd.google.protobuf; ' 'proto=io.prometheus.client.MetricFamily; ' 'encoding=delimited'
+                'application/vnd.google.protobuf; proto=io.prometheus.client.MetricFamily; encoding=delimited'
             )
         handler = self.get_http_handler(endpoint, instance)
         if (
@@ -569,7 +569,7 @@ class PrometheusScraperMixin(object):
             and not handler.ignore_tls_warning
             and not is_affirmative(handler.options.get('ssl_verify', True))
         ):
-            self.log.debug(u'An unverified HTTPS request is being made to %s', endpoint)
+            self.log.debug('An unverified HTTPS request is being made to %s', endpoint)
 
         try:
             response = handler.get(endpoint, extra_headers=headers, stream=False)

datadog_checks/base/checks/prometheus/prometheus_base.py

@@ -2,8 +2,9 @@
 # All rights reserved
 # Licensed under a 3-clause BSD style license (see LICENSE)
 
-from ...utils.common import to_native_string
-from .. import AgentCheck
+from datadog_checks.base.checks import AgentCheck
+from datadog_checks.base.utils.common import to_native_string
+
 from .mixins import PrometheusScraperMixin
 
 # Prometheus check is a parent class providing a structure and some helpers

datadog_checks/base/checks/win/winpdh_base.py

@@ -6,8 +6,8 @@ from typing import Callable, Dict, List, Optional, Tuple  # noqa: F401
 
 import win32wnet
 
-from ... import AgentCheck, is_affirmative
-from ...utils.containers import hash_mutable
+from datadog_checks.base import AgentCheck, is_affirmative
+from datadog_checks.base.utils.containers import hash_mutable
 
 try:
     from .winpdh import DATA_TYPE_DOUBLE, DATA_TYPE_INT, WinPDHCounter