add attach to existing logger feature

Author: psyb0t

README.md

@@ -53,6 +53,38 @@ logger.send_log(
 )
 ```
 
+### 🔌 Attaching to Existing logging.loggers
+
+Want to supercharge your existing logger with Telegram powers? We've got you covered! Use the `attach_to_logger` function to add Telegram logging to any logger instance:
+
+```python
+import logging
+from telegram_logger_client import attach_to_logger
+
+# Basic configuration for console output
+logging.basicConfig(level=logging.DEBUG)
+
+# Create your logger
+logger = logging.getLogger("your_logger_name")
+
+# Set up Telegram logging
+attach_to_logger(logger)
+
+# Use the logger as usual
+logger.info("This will be sent to Telegram Logger and output to stdout")
+```
+
+The `attach_to_logger` function is like a magic wand that adds Telegram logging capabilities to your existing logger. It takes care of all the behind-the-scenes work, so you can focus on what really matters - logging your heart out!
+
+Parameters for `attach_to_logger`:
+
+- `logger`: Your existing logger instance (required)
+- `base_url`: The URL where your Telegram Logger service lives (optional if set in environment)
+- `x_id`: Your secret Telegram Logger handshake (optional if set in environment)
+- `level`: Set a specific logging level for Telegram logging (optional)
+
+Don't forget to set your `TELEGRAM_LOGGER_BASE_URL` and `TELEGRAM_LOGGER_X_ID` environment variables, or pass them directly to `attach_to_logger`. Now go forth and log like a boss! 🚀📝
+
 ### 🖥️ Command Line Magic
 
 First, set up your secret lair:

telegram_logger_client/__init__.py

@@ -1 +1,2 @@
-from .telegram_logger_client import TelegramLoggerClient
+from ._telegram_logger_client import TelegramLoggerClient
+from ._logging_handler import attach_to_logger

telegram_logger_client/__main__.py

@@ -3,7 +3,8 @@
 import os
 import sys
 import logging
-from .telegram_logger_client import TelegramLoggerClient
+from ._telegram_logger_client import TelegramLoggerClient
+from ._constants import ENV_VAR_TELEGRAM_LOGGER_BASE_URL, ENV_VAR_TELEGRAM_LOGGER_X_ID
 
 
 def setup_logging(debug=False):
@@ -47,8 +48,8 @@ def main():
     logger.debug(f"Parsed arguments: {args}")
 
     # Get base_url and x_id from environment variables or command-line arguments
-    base_url = args.base_url or os.environ.get("TELEGRAM_LOGGER_BASE_URL")
-    x_id = args.x_id or os.environ.get("TELEGRAM_LOGGER_X_ID")
+    base_url = args.base_url or os.environ.get(ENV_VAR_TELEGRAM_LOGGER_BASE_URL)
+    x_id = args.x_id or os.environ.get(ENV_VAR_TELEGRAM_LOGGER_X_ID)
 
     logger.debug(f"Base URL: {base_url}")
     logger.debug(

telegram_logger_client/_constants.py

@@ -0,0 +1,2 @@
+ENV_VAR_TELEGRAM_LOGGER_BASE_URL = "TELEGRAM_LOGGER_BASE_URL"
+ENV_VAR_TELEGRAM_LOGGER_X_ID = "TELEGRAM_LOGGER_X_ID"

telegram_logger_client/_logging_handler.py

@@ -0,0 +1,73 @@
+import logging
+import os
+from typing import Optional
+from ._telegram_logger_client import TelegramLoggerClient
+from ._constants import (
+    ENV_VAR_TELEGRAM_LOGGER_BASE_URL,
+    ENV_VAR_TELEGRAM_LOGGER_X_ID,
+)
+
+
+class TelegramLoggerHandler(logging.Handler):
+    def __init__(self, base_url: str, x_id: str, *args, **kwargs):
+        super(TelegramLoggerHandler, self).__init__(*args, **kwargs)
+        self.client = TelegramLoggerClient(base_url, x_id)
+
+    def emit(self, record):
+        try:
+            message = record.getMessage()
+            data = {
+                "file": f"{record.pathname}:{record.lineno}",
+            }
+            if record.exc_info:
+                data["exception"] = self.formatter.formatException(record.exc_info)
+
+            self.client.send_log(
+                caller=record.name,
+                level=record.levelname.lower(),
+                message=message,
+                error=record.exc_text if record.exc_info else None,
+                request_id=getattr(record, "request_id", None),
+                trace_id=getattr(record, "trace_id", None),
+                span_id=getattr(record, "span_id", None),
+                data=data,
+            )
+        except Exception as e:
+            self.handleError(record)
+
+
+class TelegramLogFormatter(logging.Formatter):
+    def format(self, record):
+        # This formatter is now only used for console output, if needed
+        return super().format(record)
+
+
+def attach_to_logger(
+    logger: logging.Logger,
+    base_url: Optional[str] = None,
+    x_id: Optional[str] = None,
+    level: Optional[int] = None,
+):
+    """
+    Set up Telegram logging for an existing logger instance.
+    :param logger: The logger instance to modify
+    :param base_url: The base URL of the Telegram Logger service (optional)
+    :param x_id: The X-ID token for authentication (optional)
+    :param level: The logging level for the Telegram Logger handler (optional)
+    """
+    base_url = base_url or os.environ.get(ENV_VAR_TELEGRAM_LOGGER_BASE_URL)
+    x_id = x_id or os.environ.get(ENV_VAR_TELEGRAM_LOGGER_X_ID)
+
+    if not base_url or not x_id:
+        raise ValueError(
+            "Both base_url and x_id must be provided either as arguments or through environment variables."
+        )
+
+    telegram_handler = TelegramLoggerHandler(base_url, x_id)
+    telegram_handler.setFormatter(TelegramLogFormatter())
+
+    if level is not None:
+        telegram_handler.setLevel(level)
+
+    logger.addHandler(telegram_handler)
+    return logger