turbo-lambda 0.6.2__tar.gz → 0.7.1__tar.gz

{turbo_lambda-0.6.2 → turbo_lambda-0.7.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: turbo-lambda
-Version: 0.6.2
+Version: 0.7.1
 Summary: Turbo Lambda Description
 Author: Sam Mosleh
 Author-email: Sam Mosleh <sam.mosleh.d@gmail.com>

{turbo_lambda-0.6.2 → turbo_lambda-0.7.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "turbo-lambda"
-version = "0.6.2"
+version = "0.7.1"
 description = "Turbo Lambda Description"
 readme = "README.md"
 authors = [{ name = "Sam Mosleh", email = "sam.mosleh.d@gmail.com" }]

{turbo_lambda-0.6.2 → turbo_lambda-0.7.1}/src/turbo_lambda/decorators.py

@@ -2,7 +2,8 @@ import inspect
 import logging
 from collections.abc import Callable
 from concurrent.futures import ThreadPoolExecutor
-from contextlib import AbstractContextManager
+from contextlib import AbstractContextManager, ContextDecorator
+from contextlib import suppress as contextlib_suppress
 from functools import wraps
 from types import TracebackType
 from typing import Any, Protocol, overload
@@ -12,7 +13,6 @@ from opentelemetry.trace import format_span_id, format_trace_id, get_current_spa
 
 from turbo_lambda import schemas
 from turbo_lambda.errors import (
-    GeneralError,
     RequestValidationError,
     general_error_to_gateway_response,
 )
@@ -126,6 +126,7 @@ def request_logger_handler[ResponseT](
         log_level=logging.DEBUG,
         log_message="request",
         log_exceptions=True,
+        excluded_fields=("context",),
     )
     @wraps(func)
     def handler(
@@ -166,6 +167,7 @@ def gateway_handler[RequestT: pydantic.BaseModel](
         log_level=logging.DEBUG,
         log_message="request",
         log_exceptions=True,
+        excluded_fields=("context",),
         result_extractor=result_extractor,
     )
     @error_transformer_handler(general_error_to_gateway_response)
@@ -199,18 +201,10 @@ def parallel_sqs_handler[RequestT](
         ).annotation
 
         def single_record_processor(
-            rec: schemas.SqsRecordModel[schemas.OnErrorNone[RequestT]],
+            rec: schemas.SqsRecordModel[RequestT],
         ) -> schemas.LambdaCheckpointItem | None:
-            if rec.body is None:
-                logger.warning(
-                    "sqs_message_ignored",
-                    extra={"message_id": rec.message_id},
-                )
-                return None
             try:
                 func(rec.body)
-            except GeneralError:
-                pass
             except Exception:
                 return schemas.LambdaCheckpointItem(item_identifier=rec.message_id)
             return None
@@ -218,8 +212,19 @@ def parallel_sqs_handler[RequestT](
         def wrapper(
             event: schemas.SqsEvent[schemas.OnErrorNone[request_type]],  # type: ignore[valid-type]
         ) -> schemas.LambdaCheckpointResponse:
+            ignored_messages = [
+                rec.message_id for rec in event.records if rec.body is None
+            ]
+            if ignored_messages:
+                logger.warning(
+                    "sqs_message_ignored",
+                    extra={"message_ids": ignored_messages},
+                )
             with ThreadPoolExecutor(max_workers=max_workers) as executor:
-                responses = executor.map(single_record_processor, event.records)
+                responses = executor.map(
+                    single_record_processor,
+                    [rec for rec in event.records if rec.body is not None],
+                )
             return schemas.LambdaCheckpointResponse(
                 batch_item_failures=[item for item in responses if item is not None]
             )
@@ -247,3 +252,7 @@ class CachedContextManager[T]:
 
     def __call__(self) -> T:
         return self._value
+
+
+class suppress(contextlib_suppress, ContextDecorator):  # noqa: N801
+    pass

{turbo_lambda-0.6.2 → turbo_lambda-0.7.1}/src/turbo_lambda/log.py

@@ -4,7 +4,9 @@ import datetime
 import inspect
 import logging
 import time
+import uuid
 from collections.abc import Callable, Generator, Iterable
+from enum import Enum
 from functools import wraps
 from typing import Any, overload
 
@@ -38,6 +40,8 @@ def _json_custom_default(value: Any) -> Any:
             return value.model_dump(mode="json")
         case datetime.datetime() | datetime.date():
             return value.isoformat()
+        case uuid.UUID() | Enum():
+            return str(value)
         case set():
             return list(value)
         case _:
@@ -63,7 +67,7 @@ def log_after_call[**P, T](
     log_level: int = logging.INFO,
     log_message: str = "call",
     log_exceptions: bool = False,
-    excluded_fields: Iterable[str] = ("self", "context"),
+    excluded_fields: Iterable[str] = ("self",),
     result_extractor: None = None,
 ) -> Callable[[Callable[P, T]], Callable[P, T]]: ...
 
@@ -74,7 +78,7 @@ def log_after_call[**P, T](
     log_level: int = logging.INFO,
     log_message: str = "call",
     log_exceptions: bool = False,
-    excluded_fields: Iterable[str] = ("self", "context"),
+    excluded_fields: Iterable[str] = ("self",),
     result_extractor: Callable[[T], dict[str, Any]],
 ) -> Callable[[Callable[P, T]], Callable[P, T]]: ...
 
@@ -84,7 +88,7 @@ def log_after_call[**P, T](  # noqa: PLR0913
     log_level: int = logging.INFO,
     log_message: str = "call",
     log_exceptions: bool = False,
-    excluded_fields: Iterable[str] = ("self", "context"),
+    excluded_fields: Iterable[str] = ("self",),
     result_extractor: Callable[[T], dict[str, Any]] | None = None,
 ) -> Callable[P, T] | Callable[[Callable[P, T]], Callable[P, T]]:
     def decorator(func: Callable[P, T]) -> Callable[P, T]:

{turbo_lambda-0.6.2 → turbo_lambda-0.7.1}/src/turbo_lambda/schemas.py

@@ -46,6 +46,10 @@ class LambdaContextProtocol(Protocol):
     aws_request_id: str
 
 
+class GatewayEventHeaders[HeaderT](pydantic.BaseModel):
+    headers: HeaderT
+
+
 class GatewayEventPathParameters[ParamT](pydantic.BaseModel):
     path_parameters: Annotated[ParamT, pydantic.Field(alias="pathParameters")]
 
@@ -541,3 +545,8 @@ class AuthorizerResponse(pydantic.BaseModel):
         AuthorizerPolicyDocument, pydantic.Field(serialization_alias="policyDocument")
     ]
     context: dict[str, str]
+
+
+class AuthorizerResponseV2(pydantic.BaseModel):
+    is_authorized: Annotated[bool, pydantic.Field(alias="isAuthorized")]
+    context: dict[str, str] = {}