aws-lambda-powertools 3.10.1a6__tar.gz → 3.10.1a8__tar.gz

{aws_lambda_powertools-3.10.1a6 → aws_lambda_powertools-3.10.1a8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: aws_lambda_powertools
-Version: 3.10.1a6
+Version: 3.10.1a8
 Summary: Powertools for AWS Lambda (Python) is a developer toolkit to implement Serverless best practices and increase developer velocity.
 License: MIT
 Keywords: aws_lambda_powertools,aws,tracing,logging,lambda,powertools,feature_flags,idempotency,middleware

{aws_lambda_powertools-3.10.1a6 → aws_lambda_powertools-3.10.1a8}/aws_lambda_powertools/event_handler/api_gateway.py

@@ -12,7 +12,7 @@ from enum import Enum
 from functools import partial
 from http import HTTPStatus
 from pathlib import Path
-from typing import TYPE_CHECKING, Any, Callable, Generic, Literal, Mapping, Match, Pattern, Sequence, TypeVar, cast
+from typing import TYPE_CHECKING, Any, Generic, Literal, Match, Pattern, TypeVar, cast
 
 from typing_extensions import override
 
@@ -59,6 +59,9 @@ from aws_lambda_powertools.utilities.data_classes import (
 )
 from aws_lambda_powertools.utilities.data_classes.common import BaseProxyEvent
 
+if TYPE_CHECKING:
+    from collections.abc import Callable, Mapping, Sequence
+
 logger = logging.getLogger(__name__)
 
 _DYNAMIC_ROUTE_PATTERN = r"(<\w+>)"
@@ -68,6 +71,7 @@ _UNSAFE_URI = r"%<> \[\]{}|^"
 _NAMED_GROUP_BOUNDARY_PATTERN = rf"(?P\1[{_SAFE_URI}{_UNSAFE_URI}\\w]+)"
 _DEFAULT_OPENAPI_RESPONSE_DESCRIPTION = "Successful Response"
 _ROUTE_REGEX = "^{}$"
+_JSON_DUMP_CALL = partial(json.dumps, separators=(",", ":"), cls=Encoder)
 
 ResponseEventT = TypeVar("ResponseEventT", bound=BaseProxyEvent)
 ResponseT = TypeVar("ResponseT")
@@ -830,7 +834,7 @@ class ResponseBuilder(Generic[ResponseEventT]):
     def __init__(
         self,
         response: Response,
-        serializer: Callable[[Any], str] = partial(json.dumps, separators=(",", ":"), cls=Encoder),
+        serializer: Callable[[Any], str] = _JSON_DUMP_CALL,
         route: Route | None = None,
     ):
         self.response = response
@@ -1600,7 +1604,6 @@ class ApiGatewayResolver(BaseRouter):
         response_validation_error_http_code: HTTPStatus | int | None,
         enable_validation: bool,
     ) -> HTTPStatus:
-
         if response_validation_error_http_code and not enable_validation:
             msg = "'response_validation_error_http_code' cannot be set when enable_validation is False."
             raise ValueError(msg)
@@ -1609,7 +1612,6 @@ class ApiGatewayResolver(BaseRouter):
             not isinstance(response_validation_error_http_code, HTTPStatus)
             and response_validation_error_http_code is not None
         ):
-
             try:
                 response_validation_error_http_code = HTTPStatus(response_validation_error_http_code)
             except ValueError:
@@ -1723,8 +1725,9 @@ class ApiGatewayResolver(BaseRouter):
         security = security or self.openapi_config.security
         openapi_extensions = openapi_extensions or self.openapi_config.openapi_extensions
 
+        from pydantic.json_schema import GenerateJsonSchema
+
         from aws_lambda_powertools.event_handler.openapi.compat import (
-            GenerateJsonSchema,
             get_compat_model_name_map,
             get_definitions,
         )

{aws_lambda_powertools-3.10.1a6 → aws_lambda_powertools-3.10.1a8}/aws_lambda_powertools/event_handler/appsync.py

@@ -3,13 +3,15 @@ from __future__ import annotations
 import asyncio
 import logging
 import warnings
-from typing import TYPE_CHECKING, Any, Callable
+from typing import TYPE_CHECKING, Any
 
 from aws_lambda_powertools.event_handler.graphql_appsync.exceptions import InvalidBatchResponse, ResolverNotFoundError
 from aws_lambda_powertools.event_handler.graphql_appsync.router import Router
 from aws_lambda_powertools.utilities.data_classes import AppSyncResolverEvent
 
 if TYPE_CHECKING:
+    from collections.abc import Callable
+
     from aws_lambda_powertools.utilities.typing import LambdaContext
 
 from aws_lambda_powertools.warnings import PowertoolsUserWarning

{aws_lambda_powertools-3.10.1a6 → aws_lambda_powertools-3.10.1a8}/aws_lambda_powertools/event_handler/bedrock_agent.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 import json
-from typing import TYPE_CHECKING, Any, Callable
+from typing import TYPE_CHECKING, Any
 
 from typing_extensions import override
 
@@ -14,6 +14,7 @@ from aws_lambda_powertools.event_handler.api_gateway import (
 from aws_lambda_powertools.event_handler.openapi.constants import DEFAULT_API_VERSION, DEFAULT_OPENAPI_VERSION
 
 if TYPE_CHECKING:
+    from collections.abc import Callable
     from http import HTTPStatus
     from re import Match
 
@@ -109,11 +110,11 @@ class BedrockAgentResolver(ApiGatewayResolver):
         tags: list[str] | None = None,
         operation_id: str | None = None,
         include_in_schema: bool = True,
+        openapi_extensions: dict[str, Any] | None = None,
         deprecated: bool = False,
         custom_response_validation_http_code: int | HTTPStatus | None = None,
         middlewares: list[Callable[..., Any]] | None = None,
     ) -> Callable[[Callable[..., Any]], Callable[..., Any]]:
-        openapi_extensions = None
         security = None
 
         return super().get(
@@ -150,11 +151,11 @@ class BedrockAgentResolver(ApiGatewayResolver):
         tags: list[str] | None = None,
         operation_id: str | None = None,
         include_in_schema: bool = True,
+        openapi_extensions: dict[str, Any] | None = None,
         deprecated: bool = False,
         custom_response_validation_http_code: int | HTTPStatus | None = None,
         middlewares: list[Callable[..., Any]] | None = None,
     ):
-        openapi_extensions = None
         security = None
 
         return super().post(
@@ -191,11 +192,11 @@ class BedrockAgentResolver(ApiGatewayResolver):
         tags: list[str] | None = None,
         operation_id: str | None = None,
         include_in_schema: bool = True,
+        openapi_extensions: dict[str, Any] | None = None,
         deprecated: bool = False,
         custom_response_validation_http_code: int | HTTPStatus | None = None,
         middlewares: list[Callable[..., Any]] | None = None,
     ):
-        openapi_extensions = None
         security = None
 
         return super().put(
@@ -232,11 +233,11 @@ class BedrockAgentResolver(ApiGatewayResolver):
         tags: list[str] | None = None,
         operation_id: str | None = None,
         include_in_schema: bool = True,
+        openapi_extensions: dict[str, Any] | None = None,
         deprecated: bool = False,
         custom_response_validation_http_code: int | HTTPStatus | None = None,
         middlewares: list[Callable] | None = None,
     ):
-        openapi_extensions = None
         security = None
 
         return super().patch(
@@ -273,11 +274,11 @@ class BedrockAgentResolver(ApiGatewayResolver):
         tags: list[str] | None = None,
         operation_id: str | None = None,
         include_in_schema: bool = True,
+        openapi_extensions: dict[str, Any] | None = None,
         deprecated: bool = False,
         custom_response_validation_http_code: int | HTTPStatus | None = None,
         middlewares: list[Callable[..., Any]] | None = None,
     ):
-        openapi_extensions = None
         security = None
 
         return super().delete(
@@ -324,6 +325,7 @@ class BedrockAgentResolver(ApiGatewayResolver):
         license_info: License | None = None,
         security_schemes: dict[str, SecurityScheme] | None = None,
         security: list[dict[str, list[str]]] | None = None,
+        openapi_extensions: dict[str, Any] | None = None,
     ) -> str:
         """
         Returns the OpenAPI schema as a JSON serializable dict.
@@ -364,8 +366,6 @@ class BedrockAgentResolver(ApiGatewayResolver):
         """
         from aws_lambda_powertools.event_handler.openapi.compat import model_json
 
-        openapi_extensions = None
-
         schema = super().get_openapi_schema(
             title=title,
             version=version,

{aws_lambda_powertools-3.10.1a6 → aws_lambda_powertools-3.10.1a8}/aws_lambda_powertools/event_handler/graphql_appsync/_registry.py

@@ -1,7 +1,10 @@
 from __future__ import annotations
 
 import logging
-from typing import Any, Callable
+from typing import TYPE_CHECKING, Any
+
+if TYPE_CHECKING:
+    from collections.abc import Callable
 
 logger = logging.getLogger(__name__)
 

{aws_lambda_powertools-3.10.1a6 → aws_lambda_powertools-3.10.1a8}/aws_lambda_powertools/event_handler/graphql_appsync/base.py

@@ -1,7 +1,10 @@
 from __future__ import annotations
 
 from abc import ABC, abstractmethod
-from typing import Callable
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from collections.abc import Callable
 
 
 class BaseRouter(ABC):

{aws_lambda_powertools-3.10.1a6 → aws_lambda_powertools-3.10.1a8}/aws_lambda_powertools/event_handler/graphql_appsync/router.py

@@ -1,11 +1,13 @@
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Callable
+from typing import TYPE_CHECKING
 
 from aws_lambda_powertools.event_handler.graphql_appsync._registry import ResolverRegistry
 from aws_lambda_powertools.event_handler.graphql_appsync.base import BaseRouter
 
 if TYPE_CHECKING:
+    from collections.abc import Callable
+
     from aws_lambda_powertools.utilities.data_classes.appsync_resolver_event import AppSyncResolverEvent
     from aws_lambda_powertools.utilities.typing.lambda_context import LambdaContext
 

{aws_lambda_powertools-3.10.1a6 → aws_lambda_powertools-3.10.1a8}/aws_lambda_powertools/event_handler/lambda_function_url.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Callable, Pattern
+from typing import TYPE_CHECKING, Pattern
 
 from aws_lambda_powertools.event_handler.api_gateway import (
     ApiGatewayResolver,
@@ -8,6 +8,7 @@ from aws_lambda_powertools.event_handler.api_gateway import (
 )
 
 if TYPE_CHECKING:
+    from collections.abc import Callable
     from http import HTTPStatus
 
     from aws_lambda_powertools.event_handler import CORSConfig

{aws_lambda_powertools-3.10.1a6 → aws_lambda_powertools-3.10.1a8}/aws_lambda_powertools/event_handler/middlewares/openapi_validation.py

@@ -365,7 +365,7 @@ def _validate_field(
     """
     Validate a field, and append any errors to the existing_errors list.
     """
-    validated_value, errors = field.validate(value, value, loc=loc)
+    validated_value, errors = field.validate(value=value, loc=loc)
 
     if isinstance(errors, list):
         processed_errors = _regenerate_error_with_loc(errors=errors, loc_prefix=())

{aws_lambda_powertools-3.10.1a6 → aws_lambda_powertools-3.10.1a8}/aws_lambda_powertools/event_handler/openapi/compat.py

@@ -1,36 +1,30 @@
 # mypy: ignore-errors
-# flake8: noqa
 from __future__ import annotations
 
 from collections import deque
-from copy import copy
+from collections.abc import Mapping, Sequence
 
 # MAINTENANCE: remove when deprecating Pydantic v1. Mypy doesn't handle two different code paths that import different
 # versions of a module, so we need to ignore errors here.
-
 from dataclasses import dataclass, is_dataclass
-from enum import Enum
-from typing import TYPE_CHECKING, Any, Deque, FrozenSet, List, Mapping, Sequence, Set, Tuple, Union
-
-from typing_extensions import Annotated, Literal, get_origin, get_args
-
-from pydantic import BaseModel, create_model
-from pydantic.fields import FieldInfo
+from typing import TYPE_CHECKING, Any, Deque, FrozenSet, List, Set, Tuple, Union
 
-from aws_lambda_powertools.event_handler.openapi.types import COMPONENT_REF_PREFIX, UnionType
-
-from pydantic import TypeAdapter, ValidationError
+from pydantic import BaseModel, TypeAdapter, ValidationError, create_model
 
 # Importing from internal libraries in Pydantic may introduce potential risks, as these internal libraries
 # are not part of the public API and may change without notice in future releases.
 # We use this for forward reference, as it allows us to handle forward references in type annotations.
 from pydantic._internal._typing_extra import eval_type_lenient
-from pydantic.fields import FieldInfo
 from pydantic._internal._utils import lenient_issubclass
-from pydantic.json_schema import GenerateJsonSchema, JsonSchemaValue
 from pydantic_core import PydanticUndefined, PydanticUndefinedType
+from typing_extensions import Annotated, Literal, get_args, get_origin
+
+from aws_lambda_powertools.event_handler.openapi.types import UnionType
 
 if TYPE_CHECKING:
+    from pydantic.fields import FieldInfo
+    from pydantic.json_schema import GenerateJsonSchema, JsonSchemaValue
+
     from aws_lambda_powertools.event_handler.openapi.types import IncEx, ModelNameMap
 
 Undefined = PydanticUndefined
@@ -119,7 +113,10 @@ class ModelField:
         )
 
     def validate(
-        self, value: Any, values: dict[str, Any] = {}, *, loc: tuple[int | str, ...] = ()
+        self,
+        value: Any,
+        *,
+        loc: tuple[int | str, ...] = (),
     ) -> tuple[Any, list[dict[str, Any]] | None]:
         try:
             return (self._type_adapter.validate_python(value, from_attributes=True), None)
@@ -184,7 +181,8 @@ def copy_field_info(*, field_info: FieldInfo, annotation: Any) -> FieldInfo:
 
 def get_missing_field_error(loc: tuple[str, ...]) -> dict[str, Any]:
     error = ValidationError.from_exception_data(
-        "Field required", [{"type": "missing", "loc": loc, "input": {}}]
+        "Field required",
+        [{"type": "missing", "loc": loc, "input": {}}],
     ).errors()[0]
     error["input"] = None
     return error
@@ -308,7 +306,7 @@ def value_is_sequence(value: Any) -> bool:
 
 def _annotation_is_complex(annotation: type[Any] | None) -> bool:
     return (
-        lenient_issubclass(annotation, (BaseModel, Mapping))  # TODO: UploadFile
+        lenient_issubclass(annotation, (BaseModel, Mapping))  # Keep it to UploadFile
         or _annotation_is_sequence(annotation)
         or is_dataclass(annotation)
     )

{aws_lambda_powertools-3.10.1a6 → aws_lambda_powertools-3.10.1a8}/aws_lambda_powertools/event_handler/openapi/dependant.py

@@ -2,7 +2,7 @@ from __future__ import annotations
 
 import inspect
 import re
-from typing import TYPE_CHECKING, Any, Callable, ForwardRef, cast
+from typing import TYPE_CHECKING, Any, ForwardRef, cast
 
 from aws_lambda_powertools.event_handler.openapi.compat import (
     ModelField,
@@ -27,6 +27,8 @@ from aws_lambda_powertools.event_handler.openapi.params import (
 from aws_lambda_powertools.event_handler.openapi.types import OpenAPIResponse, OpenAPIResponseContentModel
 
 if TYPE_CHECKING:
+    from collections.abc import Callable
+
     from pydantic import BaseModel
 
 """

{aws_lambda_powertools-3.10.1a6 → aws_lambda_powertools-3.10.1a8}/aws_lambda_powertools/event_handler/openapi/encoders.py

@@ -8,7 +8,7 @@ from enum import Enum
 from pathlib import Path, PurePath
 from re import Pattern
 from types import GeneratorType
-from typing import TYPE_CHECKING, Any, Callable
+from typing import TYPE_CHECKING, Any
 from uuid import UUID
 
 from pydantic import BaseModel
@@ -17,6 +17,8 @@ from pydantic.types import SecretBytes, SecretStr
 from aws_lambda_powertools.event_handler.openapi.compat import _model_dump
 
 if TYPE_CHECKING:
+    from collections.abc import Callable
+
     from aws_lambda_powertools.event_handler.openapi.types import IncEx
 
 from aws_lambda_powertools.event_handler.openapi.exceptions import SerializationError

{aws_lambda_powertools-3.10.1a6 → aws_lambda_powertools-3.10.1a8}/aws_lambda_powertools/event_handler/openapi/exceptions.py

@@ -1,4 +1,5 @@
-from typing import Any, Literal, Sequence
+from collections.abc import Sequence
+from typing import Any, Literal
 
 
 class ValidationException(Exception):

{aws_lambda_powertools-3.10.1a6 → aws_lambda_powertools-3.10.1a8}/aws_lambda_powertools/event_handler/openapi/models.py

@@ -36,7 +36,6 @@ class OpenAPIExtensions(BaseModel):
     @model_validator(mode="before")
     def serialize_openapi_extension_v2(self):
         if isinstance(self, dict) and self.get("openapi_extensions"):
-
             openapi_extension_value = self.get("openapi_extensions")
 
             for extension_key in openapi_extension_value:

{aws_lambda_powertools-3.10.1a6 → aws_lambda_powertools-3.10.1a8}/aws_lambda_powertools/event_handler/openapi/params.py

@@ -2,7 +2,7 @@ from __future__ import annotations
 
 import inspect
 from enum import Enum
-from typing import TYPE_CHECKING, Any, Callable, Literal
+from typing import TYPE_CHECKING, Any, Literal
 
 from pydantic import BaseConfig
 from pydantic.fields import FieldInfo
@@ -20,6 +20,8 @@ from aws_lambda_powertools.event_handler.openapi.compat import (
 )
 
 if TYPE_CHECKING:
+    from collections.abc import Callable
+
     from aws_lambda_powertools.event_handler.openapi.models import Example
     from aws_lambda_powertools.event_handler.openapi.types import CacheKey
 

{aws_lambda_powertools-3.10.1a6 → aws_lambda_powertools-3.10.1a8}/aws_lambda_powertools/event_handler/openapi/types.py

@@ -1,9 +1,10 @@
 from __future__ import annotations
 
 import types
-from typing import TYPE_CHECKING, Any, Callable, Dict, Set, Type, TypedDict, Union
+from typing import TYPE_CHECKING, Any, Dict, Set, Type, TypedDict, Union
 
 if TYPE_CHECKING:
+    from collections.abc import Callable
     from enum import Enum
 
     from pydantic import BaseModel

{aws_lambda_powertools-3.10.1a6 → aws_lambda_powertools-3.10.1a8}/aws_lambda_powertools/event_handler/util.py

@@ -1,6 +1,9 @@
 from __future__ import annotations
 
-from typing import Any, Dict, List, Mapping
+from typing import TYPE_CHECKING, Any
+
+if TYPE_CHECKING:
+    from collections.abc import Mapping
 
 
 class _FrozenDict(dict):
@@ -18,7 +21,7 @@ class _FrozenDict(dict):
         return hash(frozenset(self.keys()))
 
 
-class _FrozenListDict(List[Dict[str, List[str]]]):
+class _FrozenListDict(list[dict[str, list[str]]]):
     """
     Freezes a list of dictionaries containing lists of strings.
 

{aws_lambda_powertools-3.10.1a6 → aws_lambda_powertools-3.10.1a8}/aws_lambda_powertools/event_handler/vpc_lattice.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Callable, Pattern
+from typing import TYPE_CHECKING, Pattern
 
 from aws_lambda_powertools.event_handler.api_gateway import (
     ApiGatewayResolver,
@@ -8,6 +8,7 @@ from aws_lambda_powertools.event_handler.api_gateway import (
 )
 
 if TYPE_CHECKING:
+    from collections.abc import Callable
     from http import HTTPStatus
 
     from aws_lambda_powertools.event_handler import CORSConfig

{aws_lambda_powertools-3.10.1a6 → aws_lambda_powertools-3.10.1a8}/aws_lambda_powertools/logging/logger.py

@@ -242,10 +242,6 @@ class Logger:
         buffer_config: LoggerBufferConfig | None = None,
         **kwargs,
     ) -> None:
-
-        # Used in case of sampling
-        self.initial_log_level = self._determine_log_level(level)
-
         self.service = resolve_env_var_choice(
             choice=service,
             env=os.getenv(constants.SERVICE_NAME_ENV, "service_undefined"),
@@ -285,6 +281,9 @@ class Logger:
         if self._buffer_config:
             self._buffer_cache = LoggerBufferCache(max_size_bytes=self._buffer_config.max_bytes)
 
+        # Used in case of sampling
+        self.initial_log_level = self._determine_log_level(level)
+
         self._init_logger(
             formatter_options=formatter_options,
             log_level=level,
@@ -1047,6 +1046,20 @@ class Logger:
                     stacklevel=2,
                 )
 
+            # Check if buffer level is less verbose than ALC
+            if (
+                hasattr(self, "_buffer_config")
+                and self._buffer_config
+                and logging.getLevelName(lambda_log_level)
+                > logging.getLevelName(self._buffer_config.buffer_at_verbosity)
+            ):
+                warnings.warn(
+                    "Advanced Logging Controls (ALC) Log Level is less verbose than Log Buffering Log Level. "
+                    "Buffered logs will be filtered by ALC",
+                    PowertoolsUserWarning,
+                    stacklevel=2,
+                )
+
         # AWS Lambda Advanced Logging Controls takes precedence over Powertools log level and we use this
         if lambda_log_level:
             return lambda_log_level
@@ -1133,6 +1146,7 @@ class Logger:
         Handles special first invocation buffering and migration of log records
         between different tracer contexts.
         """
+
         # Determine tracer ID, defaulting to first invoke marker
         tracer_id = get_tracer_id()
 
@@ -1180,6 +1194,7 @@ class Logger:
         Any exceptions from underlying logging or buffer mechanisms
         will be propagated to caller
         """
+
         tracer_id = get_tracer_id()
 
         # Flushing log without a tracer id? Return
@@ -1191,6 +1206,21 @@ class Logger:
         if not buffer:
             return
 
+        if not self._buffer_config:
+            return
+
+        # Check ALC level against buffer level
+        lambda_log_level = self._get_aws_lambda_log_level()
+        if lambda_log_level:
+            # Check if buffer level is less verbose than ALC
+            if logging.getLevelName(lambda_log_level) > logging.getLevelName(self._buffer_config.buffer_at_verbosity):
+                warnings.warn(
+                    "Advanced Logging Controls (ALC) Log Level is less verbose than Log Buffering Log Level. "
+                    "Some logs might be missing",
+                    PowertoolsUserWarning,
+                    stacklevel=2,
+                )
+
         # Process log records
         for log_line in buffer:
             self._create_and_flush_log_record(log_line)

{aws_lambda_powertools-3.10.1a6 → aws_lambda_powertools-3.10.1a8}/aws_lambda_powertools/shared/version.py

@@ -1,3 +1,3 @@
 """Exposes version constant to avoid circular dependencies."""
 
-VERSION = "3.10.1a6"
+VERSION = "3.10.1a8"

{aws_lambda_powertools-3.10.1a6 → aws_lambda_powertools-3.10.1a8}/aws_lambda_powertools/utilities/batch/base.py

@@ -296,8 +296,7 @@ class BasePartialBatchProcessor(BasePartialProcessor):  # noqa
 
         if self._entire_batch_failed() and self.raise_on_entire_batch_failure:
             raise BatchProcessingError(
-                msg=f"All records failed processing. {len(self.exceptions)} individual errors logged "
-                f"separately below.",
+                msg=f"All records failed processing. {len(self.exceptions)} individual errors logged separately below.",
                 child_exceptions=self.exceptions,
             )
 

{aws_lambda_powertools-3.10.1a6 → aws_lambda_powertools-3.10.1a8}/aws_lambda_powertools/utilities/data_classes/code_pipeline_job_event.py

@@ -314,7 +314,6 @@ class CodePipelineJobEvent(DictWrapper):
         # So we are using if/else instead.
 
         if self.data.encryption_key:
-
             encryption_key_id = self.data.encryption_key.get_id
             encryption_key_type = self.data.encryption_key.get_type
             if encryption_key_type == "KMS":

{aws_lambda_powertools-3.10.1a6 → aws_lambda_powertools-3.10.1a8}/aws_lambda_powertools/utilities/data_classes/kafka_event.py

@@ -37,14 +37,25 @@ class KafkaEventRecord(DictWrapper):
         return self["timestampType"]
 
     @property
-    def key(self) -> str:
-        """The raw (base64 encoded) Kafka record key."""
-        return self["key"]
+    def key(self) -> str | None:
+        """
+        The raw (base64 encoded) Kafka record key.
+
+        This key is optional; if not provided,
+        a round-robin algorithm will be used to determine
+        the partition for the message.
+        """
+
+        return self.get("key")
 
     @property
-    def decoded_key(self) -> bytes:
-        """Decode the base64 encoded key as bytes."""
-        return base64.b64decode(self.key)
+    def decoded_key(self) -> bytes | None:
+        """
+        Decode the base64 encoded key as bytes.
+
+        If the key is not provided, this will return None.
+        """
+        return None if self.key is None else base64.b64decode(self.key)
 
     @property
     def value(self) -> str:

{aws_lambda_powertools-3.10.1a6 → aws_lambda_powertools-3.10.1a8}/aws_lambda_powertools/utilities/data_classes/transfer_family_event.py

@@ -39,7 +39,6 @@ class TransferFamilyAuthorizer(DictWrapper):
 
 
 class TransferFamilyAuthorizerResponse:
-
     def _build_authentication_response(
         self,
         role_arn: str,
@@ -51,7 +50,6 @@ class TransferFamilyAuthorizerResponse:
         user_uid: int | None = None,
         public_keys: str | None = None,
     ) -> dict[str, Any]:
-
         response: dict[str, Any] = {}
 
         if home_directory_type == "PATH":

{aws_lambda_powertools-3.10.1a6 → aws_lambda_powertools-3.10.1a8}/aws_lambda_powertools/utilities/data_masking/provider/base.py

@@ -82,7 +82,6 @@ class BaseProvider:
         masking_rules: dict | None = None,
         **kwargs,
     ) -> Any:
-
         result: Any = DATA_MASKING_STRING
 
         if not any([dynamic_mask, custom_mask, regex_pattern, mask_format, masking_rules]):

{aws_lambda_powertools-3.10.1a6 → aws_lambda_powertools-3.10.1a8}/aws_lambda_powertools/utilities/idempotency/serialization/dataclass.py

@@ -38,7 +38,6 @@ class DataclassSerializer(BaseIdempotencyModelSerializer):
 
     @classmethod
     def instantiate(cls, model_type: Any) -> BaseIdempotencySerializer:
-
         model_type = get_actual_type(model_type=model_type)
 
         if model_type is None:

{aws_lambda_powertools-3.10.1a6 → aws_lambda_powertools-3.10.1a8}/aws_lambda_powertools/utilities/idempotency/serialization/pydantic.py

@@ -35,7 +35,6 @@ class PydanticSerializer(BaseIdempotencyModelSerializer):
 
     @classmethod
     def instantiate(cls, model_type: Any) -> BaseIdempotencySerializer:
-
         model_type = get_actual_type(model_type=model_type)
 
         if model_type is None:

{aws_lambda_powertools-3.10.1a6 → aws_lambda_powertools-3.10.1a8}/aws_lambda_powertools/utilities/parser/models/kafka.py

@@ -1,5 +1,5 @@
 from datetime import datetime
-from typing import Dict, List, Literal, Type, Union
+from typing import Dict, List, Literal, Optional, Type, Union
 
 from pydantic import BaseModel, field_validator
 
@@ -14,12 +14,16 @@ class KafkaRecordModel(BaseModel):
     offset: int
     timestamp: datetime
     timestampType: str
-    key: bytes
+    key: Optional[bytes] = None
     value: Union[str, Type[BaseModel]]
     headers: List[Dict[str, bytes]]
 
-    # Added type ignore to keep compatibility between Pydantic v1 and v2
-    _decode_key = field_validator("key")(base64_decode)  # type: ignore[type-var, unused-ignore]
+    # key is optional; only decode if not None
+    @field_validator("key", mode="before")
+    def decode_key(cls, value):
+        if value is not None:
+            return base64_decode(value)
+        return value
 
     @field_validator("value", mode="before")
     def data_base64_decode(cls, value):