otel-utils 0.8.0__tar.gz → 1.1.0__tar.gz

{otel_utils-0.8.0 → otel_utils-1.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: otel-utils
-Version: 0.8.0
+Version: 1.1.0
 Summary: Utilidades simplificadas para instrumentación con OpenTelemetry
 License: Proprietary
 Author: Harold Portocarrero

{otel_utils-0.8.0 → otel_utils-1.1.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "otel-utils"
-version = "0.8.0"
+version = "1.1.0"
 description = "Utilidades simplificadas para instrumentación con OpenTelemetry"
 authors = ["Harold Portocarrero <harold@getcometa.com>"]
 readme = "README.md"

{otel_utils-0.8.0 → otel_utils-1.1.0}/src/otel_utils/logging.py

@@ -43,6 +43,18 @@ class JsonFormatter(logging.Formatter):
         return json.dumps(log_record)
 
 
+class DefaultAttributesFilter(logging.Filter):
+    def __init__(self, default_attributes):
+        super().__init__()
+        self.default_attributes = default_attributes
+
+    def filter(self, record):
+        for k, v in self.default_attributes.items():
+            if not hasattr(record, k):
+                setattr(record, k, v)
+        return True
+
+
 class StructuredLogger:
     """
     Logger that produces structured logs with tracing context.
@@ -65,11 +77,9 @@ class StructuredLogger:
             handler = logging.StreamHandler()
             handler.setFormatter(JsonFormatter())
             self.logger.addHandler(handler)
+        self.logger.addFilter(DefaultAttributesFilter(self.default_attributes))
 
     def _get_trace_context(self) -> Dict[str, str]:
-        """
-        Gets the current tracing context if it exists.
-        """
         span = trace.get_current_span()
         if span.is_recording():
             ctx = span.get_span_context()
@@ -88,23 +98,15 @@ class StructuredLogger:
     ):
         operation = kwargs.pop("operation", None)
         status = kwargs.pop("status", None)
-
         context = kwargs.copy() if kwargs else None
-
-        attributes = dict(self.default_attributes)
-        attributes.update(kwargs)
-        extra_data = attributes.copy()
+        extra_data = {}
         extra_data['service_name'] = self.service_name
-
         if operation:
             extra_data["operation"] = operation
-
         if status:
             extra_data["status"] = status
-
         if context:
             extra_data["context"] = context
-
         trace_ctx = self._get_trace_context()
         if trace_ctx and not hasattr(logging.getLogRecordFactory(), "otelTraceID"):
             trace_id_hex = trace_ctx.get("trace_id")
@@ -120,7 +122,6 @@ class StructuredLogger:
                     extra_data["dd.span_id"] = span_id_hex
             extra_data["trace_id"] = trace_id_hex
             extra_data["span_id"] = span_id_hex
-
         self.logger.log(level, message, extra=extra_data)
 
     def debug(self, message: str, *args, **kwargs):
@@ -141,9 +142,6 @@ class StructuredLogger:
             operation_name: str,
             **context
     ):
-        """
-        Provides context for an operation, recording its beginning and end.
-        """
         try:
             self.info(
                 f"Iniciando {operation_name}",

{otel_utils-0.8.0 → otel_utils-1.1.0}/src/otel_utils/metrics.py

@@ -1,7 +1,7 @@
 from contextlib import contextmanager
 from typing import Optional, Dict, Any, List
 from opentelemetry import metrics
-from opentelemetry.metrics import Observation, CallbackOptions
+from opentelemetry.metrics import Observation, CallbackOptions, Counter, Histogram, _Gauge as Gauge
 import time
 
 
@@ -13,9 +13,9 @@ class Metrics:
 
     def __init__(self, service_name: str):
         self._meter = metrics.get_meter(service_name)
-        self._gauges = {}
-        self._histograms = {}
-        self._counters = {}
+        self._gauges: Dict[str, Gauge] = {}
+        self._histograms: Dict[str, Histogram] = {}
+        self._counters: Dict[str, Counter] = {}
 
     def create_histogram(
             self,
@@ -109,4 +109,14 @@ class Metrics:
                 description=description,
                 unit=unit
             )
-        return self._histograms[name]
+        return self._histograms[name]
+
+    def get_gauge(self, name: str, description: str = "", unit: str = "1"):
+        """Gets or creates a gauge."""
+        if name not in self._gauges:
+            self._gauges[name] = self._meter.create_gauge(
+                name=name,
+                description=description,
+                unit=unit
+            )
+        return self._gauges[name]