modmex-lambda 0.5.1__tar.gz → 0.5.2__tar.gz

{modmex_lambda-0.5.1 → modmex_lambda-0.5.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: modmex-lambda
-Version: 0.5.1
+Version: 0.5.2
 Summary: Ultra-lightweight AWS Lambda utilities for API Gateway routing and event handling.
 Author: Modmex
 License: MIT
@@ -597,15 +597,17 @@ from modmex_lambda import Logger
 logger = Logger()
 
 
+@logger.inject_lambda_context(log_event=True)
 def lambda_handler(event, context):
-    logger.set_context(context=context, event=event)
     logger.append_keys(tenant_id="mx")
     logger.info("request received")
 ```
 
 The logger emits structured JSON, reads `LOG_LEVEL`, uses `SERVICE_NAME` or
 `AWS_LAMBDA_FUNCTION_NAME` when no service is passed, and can extract Lambda
-request IDs and API Gateway correlation IDs.
+request IDs and API Gateway correlation IDs. The `inject_lambda_context`
+decorator resets logger state by default for warm Lambda invocations, injects
+Lambda metadata, and can optionally log the incoming event.
 
 ## Tracing
 

{modmex_lambda-0.5.1 → modmex_lambda-0.5.2}/README.md

@@ -574,15 +574,17 @@ from modmex_lambda import Logger
 logger = Logger()
 
 
+@logger.inject_lambda_context(log_event=True)
 def lambda_handler(event, context):
-    logger.set_context(context=context, event=event)
     logger.append_keys(tenant_id="mx")
     logger.info("request received")
 ```
 
 The logger emits structured JSON, reads `LOG_LEVEL`, uses `SERVICE_NAME` or
 `AWS_LAMBDA_FUNCTION_NAME` when no service is passed, and can extract Lambda
-request IDs and API Gateway correlation IDs.
+request IDs and API Gateway correlation IDs. The `inject_lambda_context`
+decorator resets logger state by default for warm Lambda invocations, injects
+Lambda metadata, and can optionally log the incoming event.
 
 ## Tracing
 

{modmex_lambda-0.5.1 → modmex_lambda-0.5.2}/modmex_lambda/logging.py

@@ -7,6 +7,7 @@ import os
 import sys
 import traceback
 from datetime import datetime, timezone
+from functools import wraps
 from threading import RLock
 from typing import Any, Callable, TextIO
 
@@ -58,6 +59,28 @@ class Logger:
             self._context = context
             self._event = event
 
+    def inject_lambda_context(
+        self,
+        func: Callable[..., Any] | None = None,
+        *,
+        log_event: bool = False,
+    ) -> Callable[..., Any]:
+        def decorator(handler: Callable[..., Any]) -> Callable[..., Any]:
+            @wraps(handler)
+            def wrapper(event: dict[str, Any], context: object, *args: Any, **kwargs: Any) -> Any:
+                self.clear_state()
+                self.set_context(context=context, event=event)
+                self._append_lambda_context_keys(context)
+                if log_event:
+                    self.info("lambda event", event=event)
+                return handler(event, context, *args, **kwargs)
+
+            return wrapper
+
+        if func is None:
+            return decorator
+        return decorator(func)
+
     def append_keys(self, **kwargs: Any) -> None:
         with self._lock:
             self._persistent_keys.update(kwargs)
@@ -138,6 +161,18 @@ class Logger:
             return message % args
         return message
 
+    def _append_lambda_context_keys(self, context: object | None) -> None:
+        if context is None:
+            return
+
+        keys = {
+            "function_name": getattr(context, "function_name", None),
+            "function_memory_size": getattr(context, "memory_limit_in_mb", None),
+            "function_arn": getattr(context, "invoked_function_arn", None),
+            "function_request_id": getattr(context, "aws_request_id", None),
+        }
+        self.append_keys(**{key: value for key, value in keys.items() if value is not None})
+
     def _extract_request_id(self) -> str | None:
         if self._context is None:
             return None

{modmex_lambda-0.5.1 → modmex_lambda-0.5.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "modmex-lambda"
-version = "0.5.1"
+version = "0.5.2"
 description = "Ultra-lightweight AWS Lambda utilities for API Gateway routing and event handling."
 readme = "README.md"
 requires-python = ">=3.10,<4.0"

{modmex_lambda-0.5.1 → modmex_lambda-0.5.2}/tests/test_logging.py

@@ -11,6 +11,9 @@ from modmex_lambda.logging import Logger
 
 class Context:
     aws_request_id = "req-123"
+    function_name = "orders-handler"
+    memory_limit_in_mb = "256"
+    invoked_function_arn = "arn:aws:lambda:us-east-1:123456789012:function:orders-handler"
 
 
 class Status(Enum):
@@ -59,6 +62,47 @@ def test_logger_includes_request_id_and_correlation_id_from_context_and_event()
     assert payload["correlation_id"] == "corr-1"
 
 
+def test_logger_inject_lambda_context_sets_context_and_logs_event() -> None:
+    stream = io.StringIO()
+    logger = Logger(service="orders", stream=stream)
+
+    @logger.inject_lambda_context(log_event=True)
+    def handler(event, context):
+        logger.info("handled")
+        return {"ok": True}
+
+    result = handler({"headers": {"X-Correlation-Id": "corr-1"}, "order_id": "o-1"}, Context())
+
+    lines = [json.loads(line) for line in stream.getvalue().splitlines() if line.strip()]
+    assert result == {"ok": True}
+    assert lines[0]["message"] == "lambda event"
+    assert lines[0]["event"]["order_id"] == "o-1"
+    assert lines[0]["request_id"] == "req-123"
+    assert lines[0]["correlation_id"] == "corr-1"
+    assert lines[0]["function_name"] == "orders-handler"
+    assert lines[0]["function_memory_size"] == "256"
+    assert lines[0]["function_arn"] == "arn:aws:lambda:us-east-1:123456789012:function:orders-handler"
+    assert lines[0]["function_request_id"] == "req-123"
+    assert lines[1]["message"] == "handled"
+    assert lines[1]["request_id"] == "req-123"
+
+
+def test_logger_inject_lambda_context_clears_state_for_warm_starts() -> None:
+    stream = io.StringIO()
+    logger = Logger(service="orders", stream=stream)
+    logger.append_keys(tenant="stale")
+
+    @logger.inject_lambda_context
+    def handler(event, context):
+        logger.info("handled")
+
+    handler({"headers": {}}, Context())
+
+    payload = json.loads(stream.getvalue().strip())
+    assert "tenant" not in payload
+    assert payload["function_name"] == "orders-handler"
+
+
 def test_logger_append_keys_and_clear_state() -> None:
     stream = io.StringIO()
     logger = Logger(service="orders", stream=stream)