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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: aws_lambda_powertools
-Version: 3.10.1a4
+Version: 3.10.1a6
 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/event_handler/exceptions.py

@@ -0,0 +1,73 @@
+from http import HTTPStatus
+
+
+class ServiceError(Exception):
+    """Powertools class HTTP Service Error"""
+
+    def __init__(self, status_code: int, msg: str):
+        """
+        Parameters
+        ----------
+        status_code: int
+            Http status code
+        msg: str
+            Error message
+        """
+        self.status_code = status_code
+        self.msg = msg
+
+
+class BadRequestError(ServiceError):
+    """Powertools class Bad Request Error (400)"""
+
+    def __init__(self, msg: str):
+        super().__init__(HTTPStatus.BAD_REQUEST, msg)
+
+
+class UnauthorizedError(ServiceError):
+    """Powertools class Unauthorized Error (401)"""
+
+    def __init__(self, msg: str):
+        super().__init__(HTTPStatus.UNAUTHORIZED, msg)
+
+
+class ForbiddenError(ServiceError):
+    """Powertools class Forbidden Error (403)"""
+
+    def __init__(self, msg: str):
+        super().__init__(HTTPStatus.FORBIDDEN, msg)
+
+
+class NotFoundError(ServiceError):
+    """Powertools class Not Found Error (404)"""
+
+    def __init__(self, msg: str = "Not found"):
+        super().__init__(HTTPStatus.NOT_FOUND, msg)
+
+
+class RequestTimeoutError(ServiceError):
+    """Powertools class Request Timeout Error (408)"""
+
+    def __init__(self, msg: str):
+        super().__init__(HTTPStatus.REQUEST_TIMEOUT, msg)
+
+
+class RequestEntityTooLargeError(ServiceError):
+    """Powertools class Request Entity Too Large Error (413)"""
+
+    def __init__(self, msg: str):
+        super().__init__(HTTPStatus.REQUEST_ENTITY_TOO_LARGE, msg)
+
+
+class InternalServerError(ServiceError):
+    """Powertools class Internal Server Error (500)"""
+
+    def __init__(self, message: str):
+        super().__init__(HTTPStatus.INTERNAL_SERVER_ERROR, message)
+
+
+class ServiceUnavailableError(ServiceError):
+    """Powertools class Service Unavailable Error (503)"""
+
+    def __init__(self, msg: str):
+        super().__init__(HTTPStatus.SERVICE_UNAVAILABLE, msg)

{aws_lambda_powertools-3.10.1a4 → aws_lambda_powertools-3.10.1a6}/aws_lambda_powertools/logging/buffer/functions.py

@@ -2,10 +2,11 @@ from __future__ import annotations
 
 import sys
 import time
-from typing import TYPE_CHECKING, Any, Mapping
+from typing import TYPE_CHECKING, Any
 
 if TYPE_CHECKING:
     import logging
+    from collections.abc import Mapping
 
 
 def _create_buffer_record(

{aws_lambda_powertools-3.10.1a4 → aws_lambda_powertools-3.10.1a6}/aws_lambda_powertools/logging/formatter.py

@@ -11,12 +11,14 @@ from contextlib import contextmanager
 from contextvars import ContextVar
 from datetime import datetime, timezone
 from functools import partial
-from typing import TYPE_CHECKING, Any, Callable, Generator, Iterable
+from typing import TYPE_CHECKING, Any
 
 from aws_lambda_powertools.shared import constants
 from aws_lambda_powertools.shared.functions import powertools_dev_is_set
 
 if TYPE_CHECKING:
+    from collections.abc import Callable, Generator, Iterable
+
     from aws_lambda_powertools.logging.types import LogRecord, LogStackTrace
 
 RESERVED_LOG_ATTRS = (
@@ -68,7 +70,7 @@ class BasePowertoolsFormatter(logging.Formatter, metaclass=ABCMeta):
         yield
 
     # These specific thread-safe methods are necessary to manage shared context in concurrent environments.
-    # They prevent race conditions and ensure data consistency across multiple threads.
+    # They prevent race conditions and ensure data consistency across multiple threads and logger.
     def thread_safe_append_keys(self, **additional_keys) -> None:
         raise NotImplementedError()
 
@@ -194,9 +196,10 @@ class LambdaPowertoolsFormatter(BasePowertoolsFormatter):
 
         # exception and exception_name fields can be added as extra key
         # in any log level, we try to extract and use them first
-        extracted_exception, extracted_exception_name = self._extract_log_exception(log_record=record)
+        extracted_exception, extracted_exception_name, exception_notes = self._extract_log_exception(log_record=record)
         formatted_log["exception"] = formatted_log.get("exception", extracted_exception)
         formatted_log["exception_name"] = formatted_log.get("exception_name", extracted_exception_name)
+        formatted_log["exception_notes"] = formatted_log.get("exception_notes", exception_notes)
         if self.serialize_stacktrace:
             # Generate the traceback from the traceback library
             formatted_log["stack_trace"] = self._serialize_stacktrace(log_record=record)
@@ -380,7 +383,7 @@ class LambdaPowertoolsFormatter(BasePowertoolsFormatter):
 
         return None
 
-    def _extract_log_exception(self, log_record: logging.LogRecord) -> tuple[str, str] | tuple[None, None]:
+    def _extract_log_exception(self, log_record: logging.LogRecord) -> tuple[str, str, list] | tuple[None, None, None]:
         """Format traceback information, if available
 
         Parameters
@@ -393,10 +396,12 @@ class LambdaPowertoolsFormatter(BasePowertoolsFormatter):
         log_record: tuple[str, str] | tuple[None, None]
             Log record with constant traceback info and exception name
         """
+
         if isinstance(log_record.exc_info, tuple) and hasattr(log_record.exc_info[0], "__name__"):
-            return self.formatException(log_record.exc_info), log_record.exc_info[0].__name__  # type: ignore
+            exception_notes = getattr(log_record.exc_info[1], "__notes__", None)
+            return self.formatException(log_record.exc_info), log_record.exc_info[0].__name__, exception_notes  # type: ignore
 
-        return None, None
+        return None, None, None
 
     def _extract_log_keys(self, log_record: logging.LogRecord) -> dict[str, Any]:
         """Extract and parse custom and reserved log keys

{aws_lambda_powertools-3.10.1a4 → aws_lambda_powertools-3.10.1a6}/aws_lambda_powertools/logging/formatters/datadog.py

@@ -1,10 +1,12 @@
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Any, Callable
+from typing import TYPE_CHECKING, Any
 
 from aws_lambda_powertools.logging.formatter import LambdaPowertoolsFormatter
 
 if TYPE_CHECKING:
+    from collections.abc import Callable
+
     from aws_lambda_powertools.logging.types import LogRecord
 
 

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

@@ -14,7 +14,7 @@ import random
 import sys
 import warnings
 from contextlib import contextmanager
-from typing import IO, TYPE_CHECKING, Any, Callable, Generator, Iterable, Mapping, TypeVar, cast, overload
+from typing import IO, TYPE_CHECKING, Any, TypeVar, cast, overload
 
 from aws_lambda_powertools.logging.buffer.cache import LoggerBufferCache
 from aws_lambda_powertools.logging.buffer.functions import _check_minimum_buffer_log_level, _create_buffer_record
@@ -45,6 +45,8 @@ from aws_lambda_powertools.utilities import jmespath_utils
 from aws_lambda_powertools.warnings import PowertoolsUserWarning
 
 if TYPE_CHECKING:
+    from collections.abc import Callable, Generator, Iterable, Mapping
+
     from aws_lambda_powertools.logging.buffer.config import LoggerBufferConfig
     from aws_lambda_powertools.shared.types import AnyCallableT
 

{aws_lambda_powertools-3.10.1a4 → aws_lambda_powertools-3.10.1a6}/aws_lambda_powertools/logging/utils.py

@@ -1,9 +1,11 @@
 from __future__ import annotations
 
 import logging
-from typing import TYPE_CHECKING, Callable
+from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
+    from collections.abc import Callable
+
     from aws_lambda_powertools.logging.logger import Logger
 
 PACKAGE_LOGGER = "aws_lambda_powertools"

{aws_lambda_powertools-3.10.1a4 → aws_lambda_powertools-3.10.1a6}/aws_lambda_powertools/metrics/base.py

@@ -15,7 +15,7 @@ import os
 import warnings
 from collections import defaultdict
 from contextlib import contextmanager
-from typing import TYPE_CHECKING, Any, Callable, Generator
+from typing import TYPE_CHECKING, Any
 
 from aws_lambda_powertools.metrics.exceptions import (
     MetricResolutionError,
@@ -34,6 +34,8 @@ from aws_lambda_powertools.shared import constants
 from aws_lambda_powertools.shared.functions import resolve_env_var_choice
 
 if TYPE_CHECKING:
+    from collections.abc import Callable, Generator
+
     from aws_lambda_powertools.metrics.types import MetricNameUnitResolution
 
 logger = logging.getLogger(__name__)

{aws_lambda_powertools-3.10.1a4 → aws_lambda_powertools-3.10.1a6}/aws_lambda_powertools/metrics/metrics.py

@@ -1,11 +1,13 @@
 # NOTE: keeps for compatibility
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Any, Callable
+from typing import TYPE_CHECKING, Any
 
 from aws_lambda_powertools.metrics.provider.cloudwatch_emf.cloudwatch import AmazonCloudWatchEMFProvider
 
 if TYPE_CHECKING:
+    from collections.abc import Callable
+
     from aws_lambda_powertools.metrics.base import MetricResolution, MetricUnit
     from aws_lambda_powertools.metrics.provider.cloudwatch_emf.types import CloudWatchEMFOutput
     from aws_lambda_powertools.shared.types import AnyCallableT

{aws_lambda_powertools-3.10.1a4 → aws_lambda_powertools-3.10.1a6}/aws_lambda_powertools/middleware_factory/__init__.py

@@ -3,6 +3,6 @@
     [`Middleware Factory`](../utilities/middleware_factory.md)
 """
 
-from .factory import lambda_handler_decorator
+from aws_lambda_powertools.middleware_factory.factory import lambda_handler_decorator
 
 __all__ = ["lambda_handler_decorator"]

{aws_lambda_powertools-3.10.1a4 → aws_lambda_powertools-3.10.1a6}/aws_lambda_powertools/middleware_factory/factory.py

@@ -4,7 +4,7 @@ import functools
 import inspect
 import logging
 import os
-from typing import Any, Callable
+from typing import TYPE_CHECKING, Any
 
 from aws_lambda_powertools.middleware_factory.exceptions import MiddlewareInvalidArgumentError
 from aws_lambda_powertools.shared import constants
@@ -13,6 +13,9 @@ from aws_lambda_powertools.tracing import Tracer
 
 logger = logging.getLogger(__name__)
 
+if TYPE_CHECKING:
+    from collections.abc import Callable
+
 
 # Maintenance: we can't yet provide an accurate return type without ParamSpec etc. see #1066
 def lambda_handler_decorator(decorator: Callable | None = None, trace_execution: bool | None = None) -> Callable:

{aws_lambda_powertools-3.10.1a4 → aws_lambda_powertools-3.10.1a6}/aws_lambda_powertools/shared/dynamodb_deserializer.py

@@ -1,7 +1,10 @@
 from __future__ import annotations
 
 from decimal import Clamped, Context, Decimal, Inexact, Overflow, Rounded, Underflow
-from typing import Any, Callable, Sequence
+from typing import TYPE_CHECKING, Any
+
+if TYPE_CHECKING:
+    from collections.abc import Callable, Sequence
 
 # NOTE: DynamoDB supports up to 38 digits precision
 # Therefore, this ensures our Decimal follows what's stored in the table

{aws_lambda_powertools-3.10.1a4 → aws_lambda_powertools-3.10.1a6}/aws_lambda_powertools/shared/functions.py

@@ -8,10 +8,13 @@ import re
 import warnings
 from binascii import Error as BinAsciiError
 from pathlib import Path
-from typing import Any, Generator, overload
+from typing import TYPE_CHECKING, Any, overload
 
 from aws_lambda_powertools.shared import constants
 
+if TYPE_CHECKING:
+    from collections.abc import Generator
+
 logger = logging.getLogger(__name__)
 
 

aws_lambda_powertools-3.10.1a6/aws_lambda_powertools/shared/types.py

@@ -0,0 +1,4 @@
+from collections.abc import Callable
+from typing import Any, TypeVar
+
+AnyCallableT = TypeVar("AnyCallableT", bound=Callable[..., Any])

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

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

{aws_lambda_powertools-3.10.1a4 → aws_lambda_powertools-3.10.1a6}/aws_lambda_powertools/tracing/base.py

@@ -8,11 +8,12 @@ from __future__ import annotations
 
 import abc
 from contextlib import contextmanager
-from typing import TYPE_CHECKING, Any, Generator, Sequence
+from typing import TYPE_CHECKING, Any
 
 if TYPE_CHECKING:
     import numbers
     import traceback
+    from collections.abc import Generator, Sequence
 
 
 class BaseSegment(abc.ABC):

{aws_lambda_powertools-3.10.1a4 → aws_lambda_powertools-3.10.1a6}/aws_lambda_powertools/tracing/tracer.py

@@ -6,7 +6,7 @@ import functools
 import inspect
 import logging
 import os
-from typing import TYPE_CHECKING, Any, Callable, Sequence, TypeVar, cast, overload
+from typing import TYPE_CHECKING, Any, TypeVar, cast, overload
 
 from aws_lambda_powertools.shared import constants
 from aws_lambda_powertools.shared.functions import (
@@ -19,6 +19,7 @@ from aws_lambda_powertools.shared.types import AnyCallableT
 
 if TYPE_CHECKING:
     import numbers
+    from collections.abc import Callable, Sequence
 
     from aws_lambda_powertools.tracing.base import BaseProvider, BaseSegment
 

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

@@ -14,7 +14,7 @@ import os
 import sys
 from abc import ABC, abstractmethod
 from enum import Enum
-from typing import TYPE_CHECKING, Any, Callable, Tuple, Union, overload
+from typing import TYPE_CHECKING, Any, Tuple, Union, overload
 
 from aws_lambda_powertools.shared import constants
 from aws_lambda_powertools.utilities.batch.exceptions import (
@@ -31,6 +31,8 @@ from aws_lambda_powertools.utilities.data_classes.kinesis_stream_event import (
 from aws_lambda_powertools.utilities.data_classes.sqs_event import SQSRecord
 
 if TYPE_CHECKING:
+    from collections.abc import Callable
+
     from aws_lambda_powertools.utilities.batch.types import (
         PartialItemFailureResponse,
         PartialItemFailures,

{aws_lambda_powertools-3.10.1a4 → aws_lambda_powertools-3.10.1a6}/aws_lambda_powertools/utilities/batch/decorators.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 import warnings
-from typing import TYPE_CHECKING, Any, Awaitable, Callable
+from typing import TYPE_CHECKING, Any
 
 from typing_extensions import deprecated
 
@@ -16,6 +16,8 @@ from aws_lambda_powertools.utilities.batch.exceptions import UnexpectedBatchType
 from aws_lambda_powertools.warnings import PowertoolsDeprecationWarning
 
 if TYPE_CHECKING:
+    from collections.abc import Awaitable, Callable
+
     from aws_lambda_powertools.utilities.batch.types import PartialItemFailureResponse
     from aws_lambda_powertools.utilities.typing import LambdaContext
 

{aws_lambda_powertools-3.10.1a4 → aws_lambda_powertools-3.10.1a6}/aws_lambda_powertools/utilities/data_classes/active_mq_event.py

@@ -1,11 +1,14 @@
 from __future__ import annotations
 
 from functools import cached_property
-from typing import Any, Iterator
+from typing import TYPE_CHECKING, Any
 
 from aws_lambda_powertools.utilities.data_classes.common import DictWrapper
 from aws_lambda_powertools.utilities.data_classes.shared_functions import base64_decode
 
+if TYPE_CHECKING:
+    from collections.abc import Iterator
+
 
 class ActiveMQMessage(DictWrapper):
     @property

{aws_lambda_powertools-3.10.1a4 → aws_lambda_powertools-3.10.1a6}/aws_lambda_powertools/utilities/data_classes/common.py

@@ -9,14 +9,17 @@ from __future__ import annotations
 import base64
 import json
 import warnings
+from collections.abc import Mapping
 from functools import cached_property
-from typing import TYPE_CHECKING, Any, Callable, Iterator, Mapping, overload
+from typing import TYPE_CHECKING, Any, overload
 
 from typing_extensions import deprecated
 
 from aws_lambda_powertools.warnings import PowertoolsDeprecationWarning
 
 if TYPE_CHECKING:
+    from collections.abc import Callable, Iterator
+
     from aws_lambda_powertools.shared.headers_serializer import BaseHeadersSerializer
 
 from aws_lambda_powertools.utilities.data_classes.shared_functions import (

{aws_lambda_powertools-3.10.1a4 → aws_lambda_powertools-3.10.1a6}/aws_lambda_powertools/utilities/data_classes/dynamo_db_stream_event.py

@@ -2,11 +2,14 @@ from __future__ import annotations
 
 from enum import Enum
 from functools import cached_property
-from typing import Any, Iterator
+from typing import TYPE_CHECKING, Any
 
 from aws_lambda_powertools.shared.dynamodb_deserializer import TypeDeserializer
 from aws_lambda_powertools.utilities.data_classes.common import DictWrapper
 
+if TYPE_CHECKING:
+    from collections.abc import Iterator
+
 
 class StreamViewType(Enum):
     """The type of data from the modified DynamoDB item that was captured in this stream record"""

{aws_lambda_powertools-3.10.1a4 → aws_lambda_powertools-3.10.1a6}/aws_lambda_powertools/utilities/data_classes/event_source.py

@@ -1,10 +1,12 @@
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Any, Callable
+from typing import TYPE_CHECKING, Any
 
 from aws_lambda_powertools.middleware_factory import lambda_handler_decorator
 
 if TYPE_CHECKING:
+    from collections.abc import Callable
+
     from aws_lambda_powertools.utilities.data_classes.common import DictWrapper
     from aws_lambda_powertools.utilities.typing import LambdaContext
 

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

@@ -2,10 +2,13 @@ from __future__ import annotations
 
 import base64
 from functools import cached_property
-from typing import Any, Iterator
+from typing import TYPE_CHECKING, Any
 
 from aws_lambda_powertools.utilities.data_classes.common import CaseInsensitiveDict, DictWrapper
 
+if TYPE_CHECKING:
+    from collections.abc import Iterator
+
 
 class KafkaEventRecord(DictWrapper):
     @property

{aws_lambda_powertools-3.10.1a4 → aws_lambda_powertools-3.10.1a6}/aws_lambda_powertools/utilities/data_classes/kinesis_firehose_event.py

@@ -5,11 +5,13 @@ import json
 import warnings
 from dataclasses import dataclass, field
 from functools import cached_property
-from typing import TYPE_CHECKING, Any, Callable, ClassVar, Iterator
+from typing import TYPE_CHECKING, Any, ClassVar
 
 from aws_lambda_powertools.utilities.data_classes.common import DictWrapper
 
 if TYPE_CHECKING:
+    from collections.abc import Callable, Iterator
+
     from typing_extensions import Literal
 
 

{aws_lambda_powertools-3.10.1a4 → aws_lambda_powertools-3.10.1a6}/aws_lambda_powertools/utilities/data_classes/kinesis_stream_event.py

@@ -3,13 +3,16 @@ from __future__ import annotations
 import base64
 import json
 import zlib
-from typing import Iterator
+from typing import TYPE_CHECKING
 
 from aws_lambda_powertools.utilities.data_classes.cloud_watch_logs_event import (
     CloudWatchLogsDecodedData,
 )
 from aws_lambda_powertools.utilities.data_classes.common import DictWrapper
 
+if TYPE_CHECKING:
+    from collections.abc import Iterator
+
 
 class KinesisStreamRecordPayload(DictWrapper):
     @property

{aws_lambda_powertools-3.10.1a4 → aws_lambda_powertools-3.10.1a6}/aws_lambda_powertools/utilities/data_classes/s3_batch_operation_event.py

@@ -2,7 +2,7 @@ from __future__ import annotations
 
 import warnings
 from dataclasses import dataclass, field
-from typing import Any, Iterator, Literal
+from typing import TYPE_CHECKING, Any, Literal
 from urllib.parse import unquote_plus
 
 from aws_lambda_powertools.utilities.data_classes.common import DictWrapper
@@ -11,6 +11,9 @@ from aws_lambda_powertools.utilities.data_classes.common import DictWrapper
 VALID_RESULT_CODES: tuple[str, str, str] = ("Succeeded", "TemporaryFailure", "PermanentFailure")
 RESULT_CODE_TYPE = Literal["Succeeded", "TemporaryFailure", "PermanentFailure"]
 
+if TYPE_CHECKING:
+    from collections.abc import Iterator
+
 
 @dataclass(repr=False, order=False)
 class S3BatchOperationResponseRecord:

{aws_lambda_powertools-3.10.1a4 → aws_lambda_powertools-3.10.1a6}/aws_lambda_powertools/utilities/data_classes/s3_event.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from typing import Iterator
+from typing import TYPE_CHECKING
 from urllib.parse import unquote_plus
 
 from aws_lambda_powertools.utilities.data_classes.common import DictWrapper
@@ -8,6 +8,9 @@ from aws_lambda_powertools.utilities.data_classes.event_bridge_event import (
     EventBridgeEvent,
 )
 
+if TYPE_CHECKING:
+    from collections.abc import Iterator
+
 
 class S3Identity(DictWrapper):
     @property

{aws_lambda_powertools-3.10.1a4 → aws_lambda_powertools-3.10.1a6}/aws_lambda_powertools/utilities/data_classes/ses_event.py

@@ -1,9 +1,12 @@
 from __future__ import annotations
 
-from typing import Iterator
+from typing import TYPE_CHECKING
 
 from aws_lambda_powertools.utilities.data_classes.common import DictWrapper
 
+if TYPE_CHECKING:
+    from collections.abc import Iterator
+
 
 class SESMailHeader(DictWrapper):
     @property

{aws_lambda_powertools-3.10.1a4 → aws_lambda_powertools-3.10.1a6}/aws_lambda_powertools/utilities/data_classes/sns_event.py

@@ -1,9 +1,12 @@
 from __future__ import annotations
 
-from typing import Iterator
+from typing import TYPE_CHECKING
 
 from aws_lambda_powertools.utilities.data_classes.common import DictWrapper
 
+if TYPE_CHECKING:
+    from collections.abc import Iterator
+
 
 class SNSMessageAttribute(DictWrapper):
     @property

{aws_lambda_powertools-3.10.1a4 → aws_lambda_powertools-3.10.1a6}/aws_lambda_powertools/utilities/data_classes/sqs_event.py

@@ -1,12 +1,15 @@
 from __future__ import annotations
 
 from functools import cached_property
-from typing import Any, Dict, ItemsView, Iterator, TypeVar
+from typing import TYPE_CHECKING, Any, ItemsView, Iterator, TypeVar
 
 from aws_lambda_powertools.utilities.data_classes import S3Event
 from aws_lambda_powertools.utilities.data_classes.common import DictWrapper
 from aws_lambda_powertools.utilities.data_classes.sns_event import SNSMessage
 
+if TYPE_CHECKING:
+    from collections.abc import Iterator
+
 
 class SQSRecordAttributes(DictWrapper):
     @property
@@ -86,7 +89,7 @@ class SQSMessageAttribute(DictWrapper):
         return self["dataType"]
 
 
-class SQSMessageAttributes(Dict[str, SQSMessageAttribute]):
+class SQSMessageAttributes(dict[str, SQSMessageAttribute]):
     def __getitem__(self, key: str) -> SQSMessageAttribute | None:  # type: ignore
         item = super().get(key)
         return None if item is None else SQSMessageAttribute(item)  # type: ignore

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

@@ -11,7 +11,7 @@ import functools
 import logging
 import warnings
 from copy import deepcopy
-from typing import TYPE_CHECKING, Any, Callable, Mapping, Sequence
+from typing import TYPE_CHECKING, Any
 
 from jsonpath_ng.ext import parse
 
@@ -23,6 +23,7 @@ from aws_lambda_powertools.utilities.data_masking.provider import BaseProvider
 from aws_lambda_powertools.warnings import PowertoolsUserWarning
 
 if TYPE_CHECKING:
+    from collections.abc import Callable, Mapping, Sequence
     from numbers import Number
 
 logger = logging.getLogger(__name__)

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

@@ -3,13 +3,18 @@ from __future__ import annotations
 import functools
 import json
 import re
-from typing import Any, Callable
+from typing import TYPE_CHECKING, Any
 
 from aws_lambda_powertools.utilities.data_masking.constants import DATA_MASKING_STRING
 
+if TYPE_CHECKING:
+    from collections.abc import Callable
+
 PRESERVE_CHARS = set("-_. ")
 _regex_cache = {}
 
+JSON_DUMPS_CALL = functools.partial(json.dumps, ensure_ascii=False)
+
 
 class BaseProvider:
     """
@@ -49,7 +54,7 @@ class BaseProvider:
 
     def __init__(
         self,
-        json_serializer: Callable[..., str] = functools.partial(json.dumps, ensure_ascii=False),
+        json_serializer: Callable[..., str] = JSON_DUMPS_CALL,
         json_deserializer: Callable[[str], Any] = json.loads,
     ) -> None:
         self.json_serializer = json_serializer