otel-utils 0.1.16__tar.gz → 0.1.18__tar.gz

{otel_utils-0.1.16 → otel_utils-0.1.18}/PKG-INFO

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

{otel_utils-0.1.16 → otel_utils-0.1.18}/pyproject.toml

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

{otel_utils-0.1.16 → otel_utils-0.1.18}/src/otel_utils/logging.py

@@ -1,3 +1,4 @@
+import json
 import logging
 import platform
 from contextlib import contextmanager
@@ -7,6 +8,25 @@ from typing import Any, Dict, Optional
 from opentelemetry import trace
 
 
+class JsonFormatter(logging.Formatter):
+    def format(self, record):
+        log_data = {}
+
+        if hasattr(record, 'structured') and record.structured:
+            for key, value in record.__dict__.items():
+                if key not in (
+                        'args', 'exc_info', 'exc_text', 'msg', 'message', 'levelname',
+                        'levelno', 'pathname', 'filename', 'module', 'stack_info',
+                        'lineno', 'funcName', 'created', 'msecs', 'relativeCreated',
+                        'name', 'thread', 'threadName', 'processName', 'process'
+                ):
+                    log_data[key] = value
+
+        log_data['message'] = record.getMessage()
+
+        return json.dumps(log_data)
+
+
 class StructuredLogger:
     """
     Logger that produces structured logs with tracing context.
@@ -24,6 +44,11 @@ class StructuredLogger:
             **(default_attributes or {})
         }
 
+        if not self.logger.handlers or not any(isinstance(h.formatter, JsonFormatter) for h in self.logger.handlers):
+            handler = logging.StreamHandler()
+            handler.setFormatter(JsonFormatter())
+            self.logger.addHandler(handler)
+
     def _get_trace_context(self) -> Dict[str, str]:
         """
         Gets the current tracing context if it exists.
@@ -53,31 +78,24 @@ class StructuredLogger:
             "logger.name": self.logger.name
         }
 
-        additional_info = []
-        if kwargs.get("error_type"):
-            additional_info.append(f"type={kwargs['error_type']}")
-            log_attributes["error.type"] = kwargs["error_type"]
-        if kwargs.get("error_message"):
-            additional_info.append(f"message={kwargs['error_message']}")
-            log_attributes["error.message"] = kwargs["error_message"]
-        if kwargs.get("operation"):
-            additional_info.append(f"operation={kwargs['operation']}")
-            log_attributes["operation"] = kwargs["operation"]
-
-        remaining_attrs = {
-            k: v for k, v in kwargs.items()
-            if k not in ["error_type", "error_message", "operation"]
-        }
-        if remaining_attrs:
-            log_attributes["attributes"] = remaining_attrs
+        if trace_context:
+            log_attributes["trace_id"] = trace_context.get("trace_id")
+            log_attributes["span_id"] = trace_context.get("span_id")
+
+        operation = kwargs.pop("operation", None)
+        if operation:
+            log_attributes["operation"] = operation
+
+        status = kwargs.pop("status", None)
+        if status:
+            log_attributes["status"] = status
 
-        additional_str = f" - {', '.join(additional_info)}" if additional_info else ""
-        trace_str = f"[trace_id={trace_context.get('trace_id', '0')}]" if trace_context else ""
-        log_message = f"{trace_str} {message}{additional_str}"
+        if kwargs:
+            log_attributes["context"] = kwargs
 
         self.logger.log(
             level,
-            log_message,
+            message,
             extra={
                 "structured": True,
                 "otel.name": "log",