rebrandly-otel 0.1.30__tar.gz → 0.1.33__tar.gz

{rebrandly_otel-0.1.30 → rebrandly_otel-0.1.33}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rebrandly_otel
-Version: 0.1.30
+Version: 0.1.33
 Summary: Python OTEL wrapper by Rebrandly
 Home-page: https://github.com/rebrandly/rebrandly-otel-python
 Author: Antonio Romano

{rebrandly_otel-0.1.30 → rebrandly_otel-0.1.33}/rebrandly_otel.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rebrandly_otel
-Version: 0.1.30
+Version: 0.1.33
 Summary: Python OTEL wrapper by Rebrandly
 Home-page: https://github.com/rebrandly/rebrandly-otel-python
 Author: Antonio Romano

{rebrandly_otel-0.1.30 → rebrandly_otel-0.1.33}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r") as fh:
 
 setuptools.setup(
     name="rebrandly_otel",
-    version="0.1.30",
+    version="0.1.33",
     author="Antonio Romano",
     author_email="antonio@rebrandly.com",
     description="Python OTEL wrapper by Rebrandly",

{rebrandly_otel-0.1.30 → rebrandly_otel-0.1.33}/src/logs.py

@@ -18,6 +18,14 @@ from .otel_utils import *
 class RebrandlyLogger:
     """Wrapper for OpenTelemetry logging with Rebrandly-specific features."""
 
+    # Expose logging levels for convenience (compatible with standard logging)
+    DEBUG = logging.DEBUG
+    INFO = logging.INFO
+    WARNING = logging.WARNING
+    ERROR = logging.ERROR
+    CRITICAL = logging.CRITICAL
+    NOTSET = logging.NOTSET
+
     def __init__(self):
         self._logger: Optional[logging.Logger] = None
         self._provider: Optional[LoggerProvider] = None
@@ -78,6 +86,31 @@ class RebrandlyLogger:
             self._logger = logging.getLogger(get_service_name())
         return self._logger
 
+    def getLogger(self) -> logging.Logger:
+        """
+        Get the internal logger instance.
+        Alias for the logger property for consistency with standard logging API.
+        """
+        return self.logger
+
+    def setLevel(self, level: int):
+        """
+        Set the logging level for both the internal logger and OTEL handler.
+
+        Args:
+            level: Logging level (e.g., logging.INFO, logging.DEBUG, logging.WARNING)
+        """
+        # Set level on the service-specific logger using the original unbound method
+        # This avoids infinite recursion if the logger's setLevel has been monkey-patched
+        if self._logger:
+            logging.Logger.setLevel(self._logger, level)
+
+        # Set level on the OTEL handler
+        root_logger = logging.getLogger()
+        for handler in root_logger.handlers:
+            if isinstance(handler, LoggingHandler):
+                handler.setLevel(level)
+
     def force_flush(self, timeout_millis: int = 5000) -> bool:
         """
         Force flush all pending logs.

{rebrandly_otel-0.1.30 → rebrandly_otel-0.1.33}/src/pymysql_instrumentation.py

@@ -44,19 +44,24 @@ def instrument_pymysql(otel_instance, connection, options=None):
     # Get tracer from RebrandlyOTEL instance
     tracer = otel_instance.tracer
 
+    # Extract database name from connection
+    db_name = getattr(connection, 'db', None) or getattr(connection, 'database', None)
+    if db_name and isinstance(db_name, bytes):
+        db_name = db_name.decode('utf-8')
+
     # Wrap the cursor method to return instrumented cursors
     original_cursor = connection.cursor
 
     def instrumented_cursor(*args, **kwargs):
         cursor = original_cursor(*args, **kwargs)
-        return _instrument_cursor(cursor, tracer, slow_query_threshold_ms, capture_bindings)
+        return _instrument_cursor(cursor, tracer, slow_query_threshold_ms, capture_bindings, db_name)
 
     connection.cursor = instrumented_cursor
 
     return connection
 
 
-def _instrument_cursor(cursor, tracer, slow_query_threshold_ms, capture_bindings):
+def _instrument_cursor(cursor, tracer, slow_query_threshold_ms, capture_bindings, db_name=None):
     """
     Instrument a cursor's execute methods
     """
@@ -70,6 +75,7 @@ def _instrument_cursor(cursor, tracer, slow_query_threshold_ms, capture_bindings
             tracer,
             slow_query_threshold_ms,
             capture_bindings,
+            db_name,
             query,
             args,
             many=False
@@ -82,6 +88,7 @@ def _instrument_cursor(cursor, tracer, slow_query_threshold_ms, capture_bindings
             tracer,
             slow_query_threshold_ms,
             capture_bindings,
+            db_name,
             query,
             args,
             many=True
@@ -93,7 +100,7 @@ def _instrument_cursor(cursor, tracer, slow_query_threshold_ms, capture_bindings
     return cursor
 
 
-def _trace_query(func, tracer, slow_query_threshold_ms, capture_bindings, query, args, many=False):
+def _trace_query(func, tracer, slow_query_threshold_ms, capture_bindings, db_name, query, args, many=False):
     """
     Trace a query execution with OpenTelemetry
     """
@@ -112,6 +119,12 @@ def _trace_query(func, tracer, slow_query_threshold_ms, capture_bindings, query,
         span.set_attribute('db.operation.name', operation)
         span.set_attribute('db.statement', truncated_query)
 
+        # Set database name if available
+        if db_name:
+            span.set_attribute('db.name', db_name)
+        else:
+            span.set_attribute('db.name', 'unknown')
+
         # Add bindings if enabled (be cautious with sensitive data)
         if capture_bindings and args:
             if many:

{rebrandly_otel-0.1.30 → rebrandly_otel-0.1.33}/src/rebrandly_otel.py

@@ -525,3 +525,15 @@ attach_context = otel.attach_context
 detach_context = otel.detach_context
 force_flush = otel.force_flush
 shutdown = otel.shutdown
+
+# Attach logging levels to logger for convenience
+# This allows: from rebrandly_otel import logger; logger.setLevel(logger.INFO)
+import logging as _logging
+logger.DEBUG = _logging.DEBUG
+logger.INFO = _logging.INFO
+logger.WARNING = _logging.WARNING
+logger.ERROR = _logging.ERROR
+logger.CRITICAL = _logging.CRITICAL
+logger.NOTSET = _logging.NOTSET
+logger.setLevel = otel.logger.setLevel
+logger.getLogger = otel.logger.getLogger