chore(lint): use new isort black integration

Author: heitorlessa

aws_lambda_powertools/event_handler/api_gateway.py

@@ -10,7 +10,19 @@
 from enum import Enum
 from functools import partial
 from http import HTTPStatus
-from typing import Any, Callable, Dict, List, Match, Optional, Pattern, Set, Tuple, Type, Union
+from typing import (
+    Any,
+    Callable,
+    Dict,
+    List,
+    Match,
+    Optional,
+    Pattern,
+    Set,
+    Tuple,
+    Type,
+    Union,
+)
 
 from aws_lambda_powertools.event_handler import content_types
 from aws_lambda_powertools.event_handler.exceptions import NotFoundError, ServiceError

aws_lambda_powertools/event_handler/lambda_function_url.py

@@ -1,7 +1,10 @@
 from typing import Callable, Dict, List, Optional
 
 from aws_lambda_powertools.event_handler import CORSConfig
-from aws_lambda_powertools.event_handler.api_gateway import ApiGatewayResolver, ProxyEventType
+from aws_lambda_powertools.event_handler.api_gateway import (
+    ApiGatewayResolver,
+    ProxyEventType,
+)
 from aws_lambda_powertools.utilities.data_classes import LambdaFunctionUrlEvent
 
 

aws_lambda_powertools/logging/logger.py

@@ -12,7 +12,11 @@
 from ..shared.functions import resolve_env_var_choice, resolve_truthy_env_var_choice
 from .exceptions import InvalidLoggerSamplingRateError
 from .filters import SuppressFilter
-from .formatter import RESERVED_FORMATTER_CUSTOM_KEYS, BasePowertoolsFormatter, LambdaPowertoolsFormatter
+from .formatter import (
+    RESERVED_FORMATTER_CUSTOM_KEYS,
+    BasePowertoolsFormatter,
+    LambdaPowertoolsFormatter,
+)
 from .lambda_context import build_lambda_context_model
 
 logger = logging.getLogger(__name__)

aws_lambda_powertools/tracing/extensions.py

@@ -8,7 +8,9 @@ def aiohttp_trace_config():
     TraceConfig
         aiohttp trace config
     """
-    from aws_xray_sdk.ext.aiohttp.client import aws_xray_trace_config  # pragma: no cover
+    from aws_xray_sdk.ext.aiohttp.client import (
+        aws_xray_trace_config,  # pragma: no cover
+    )
 
     aws_xray_trace_config.__doc__ = "aiohttp extension for X-Ray (aws_xray_trace_config)"  # pragma: no cover
 

aws_lambda_powertools/utilities/batch/__init__.py

@@ -13,7 +13,10 @@
     batch_processor,
 )
 from aws_lambda_powertools.utilities.batch.exceptions import ExceptionInfo
-from aws_lambda_powertools.utilities.batch.sqs import PartialSQSProcessor, sqs_batch_processor
+from aws_lambda_powertools.utilities.batch.sqs import (
+    PartialSQSProcessor,
+    sqs_batch_processor,
+)
 
 __all__ = (
     "BatchProcessor",

aws_lambda_powertools/utilities/batch/base.py

@@ -12,9 +12,16 @@
 from typing import Any, Callable, Dict, List, Optional, Tuple, Type, Union, overload
 
 from aws_lambda_powertools.middleware_factory import lambda_handler_decorator
-from aws_lambda_powertools.utilities.batch.exceptions import BatchProcessingError, ExceptionInfo
-from aws_lambda_powertools.utilities.data_classes.dynamo_db_stream_event import DynamoDBRecord
-from aws_lambda_powertools.utilities.data_classes.kinesis_stream_event import KinesisStreamRecord
+from aws_lambda_powertools.utilities.batch.exceptions import (
+    BatchProcessingError,
+    ExceptionInfo,
+)
+from aws_lambda_powertools.utilities.data_classes.dynamo_db_stream_event import (
+    DynamoDBRecord,
+)
+from aws_lambda_powertools.utilities.data_classes.kinesis_stream_event import (
+    KinesisStreamRecord,
+)
 from aws_lambda_powertools.utilities.data_classes.sqs_event import SQSRecord
 from aws_lambda_powertools.utilities.typing import LambdaContext
 
@@ -36,7 +43,9 @@ class EventType(Enum):
 # We need them as subclasses as we must access their message ID or sequence number metadata via dot notation
 if has_pydantic:
     from aws_lambda_powertools.utilities.parser.models import DynamoDBStreamRecordModel
-    from aws_lambda_powertools.utilities.parser.models import KinesisDataStreamRecord as KinesisDataStreamRecordModel
+    from aws_lambda_powertools.utilities.parser.models import (
+        KinesisDataStreamRecord as KinesisDataStreamRecordModel,
+    )
     from aws_lambda_powertools.utilities.parser.models import SqsRecordModel
 
     BatchTypeModels = Optional[

aws_lambda_powertools/utilities/data_classes/alb_event.py

@@ -1,6 +1,9 @@
 from typing import Dict, List, Optional
 
-from aws_lambda_powertools.utilities.data_classes.common import BaseProxyEvent, DictWrapper
+from aws_lambda_powertools.utilities.data_classes.common import (
+    BaseProxyEvent,
+    DictWrapper,
+)
 
 
 class ALBEventRequestContext(DictWrapper):

aws_lambda_powertools/utilities/data_classes/appsync_resolver_event.py

@@ -1,6 +1,9 @@
 from typing import Any, Dict, List, Optional, Union
 
-from aws_lambda_powertools.utilities.data_classes.common import DictWrapper, get_header_value
+from aws_lambda_powertools.utilities.data_classes.common import (
+    DictWrapper,
+    get_header_value,
+)
 
 
 def get_identity_object(identity: Optional[dict]) -> Any:

aws_lambda_powertools/utilities/data_classes/lambda_function_url_event.py

@@ -1,4 +1,6 @@
-from aws_lambda_powertools.utilities.data_classes.api_gateway_proxy_event import APIGatewayProxyEventV2
+from aws_lambda_powertools.utilities.data_classes.api_gateway_proxy_event import (
+    APIGatewayProxyEventV2,
+)
 
 
 class LambdaFunctionUrlEvent(APIGatewayProxyEventV2):

aws_lambda_powertools/utilities/data_classes/s3_object_event.py

@@ -1,6 +1,9 @@
 from typing import Dict, Optional
 
-from aws_lambda_powertools.utilities.data_classes.common import DictWrapper, get_header_value
+from aws_lambda_powertools.utilities.data_classes.common import (
+    DictWrapper,
+    get_header_value,
+)
 
 
 class S3ObjectContext(DictWrapper):

aws_lambda_powertools/utilities/feature_flags/appconfig.py

@@ -5,7 +5,11 @@
 from botocore.config import Config
 
 from aws_lambda_powertools.utilities import jmespath_utils
-from aws_lambda_powertools.utilities.parameters import AppConfigProvider, GetParameterError, TransformParameterError
+from aws_lambda_powertools.utilities.parameters import (
+    AppConfigProvider,
+    GetParameterError,
+    TransformParameterError,
+)
 
 from ... import Logger
 from .base import StoreProvider

aws_lambda_powertools/utilities/idempotency/__init__.py

@@ -2,8 +2,12 @@
 Utility for adding idempotency to lambda functions
 """
 
-from aws_lambda_powertools.utilities.idempotency.persistence.base import BasePersistenceLayer
-from aws_lambda_powertools.utilities.idempotency.persistence.dynamodb import DynamoDBPersistenceLayer
+from aws_lambda_powertools.utilities.idempotency.persistence.base import (
+    BasePersistenceLayer,
+)
+from aws_lambda_powertools.utilities.idempotency.persistence.dynamodb import (
+    DynamoDBPersistenceLayer,
+)
 
 from .idempotency import IdempotencyConfig, idempotent, idempotent_function
 

aws_lambda_powertools/utilities/idempotency/idempotency.py

@@ -11,7 +11,9 @@
 from aws_lambda_powertools.shared.types import AnyCallableT
 from aws_lambda_powertools.utilities.idempotency.base import IdempotencyHandler
 from aws_lambda_powertools.utilities.idempotency.config import IdempotencyConfig
-from aws_lambda_powertools.utilities.idempotency.persistence.base import BasePersistenceLayer
+from aws_lambda_powertools.utilities.idempotency.persistence.base import (
+    BasePersistenceLayer,
+)
 from aws_lambda_powertools.utilities.typing import LambdaContext
 
 logger = logging.getLogger(__name__)

aws_lambda_powertools/utilities/idempotency/persistence/dynamodb.py

@@ -12,7 +12,10 @@
     IdempotencyItemAlreadyExistsError,
     IdempotencyItemNotFoundError,
 )
-from aws_lambda_powertools.utilities.idempotency.persistence.base import STATUS_CONSTANTS, DataRecord
+from aws_lambda_powertools.utilities.idempotency.persistence.base import (
+    STATUS_CONSTANTS,
+    DataRecord,
+)
 
 logger = logging.getLogger(__name__)
 

aws_lambda_powertools/utilities/parser/models/__init__.py

@@ -14,11 +14,29 @@
     RequestContextV2AuthorizerJwt,
     RequestContextV2Http,
 )
-from .cloudwatch import CloudWatchLogsData, CloudWatchLogsDecode, CloudWatchLogsLogEvent, CloudWatchLogsModel
-from .dynamodb import DynamoDBStreamChangedRecordModel, DynamoDBStreamModel, DynamoDBStreamRecordModel
+from .cloudwatch import (
+    CloudWatchLogsData,
+    CloudWatchLogsDecode,
+    CloudWatchLogsLogEvent,
+    CloudWatchLogsModel,
+)
+from .dynamodb import (
+    DynamoDBStreamChangedRecordModel,
+    DynamoDBStreamModel,
+    DynamoDBStreamRecordModel,
+)
 from .event_bridge import EventBridgeModel
-from .kafka import KafkaBaseEventModel, KafkaMskEventModel, KafkaRecordModel, KafkaSelfManagedEventModel
-from .kinesis import KinesisDataStreamModel, KinesisDataStreamRecord, KinesisDataStreamRecordPayload
+from .kafka import (
+    KafkaBaseEventModel,
+    KafkaMskEventModel,
+    KafkaRecordModel,
+    KafkaSelfManagedEventModel,
+)
+from .kinesis import (
+    KinesisDataStreamModel,
+    KinesisDataStreamRecord,
+    KinesisDataStreamRecordPayload,
+)
 from .lambda_function_url import LambdaFunctionUrlModel
 from .s3 import S3Model, S3RecordModel
 from .s3_object_event import (

aws_lambda_powertools/utilities/parser/parser.py

@@ -1,7 +1,11 @@
 import logging
 from typing import Any, Callable, Dict, Optional, Type, overload
 
-from aws_lambda_powertools.utilities.parser.types import EnvelopeModel, EventParserReturnType, Model
+from aws_lambda_powertools.utilities.parser.types import (
+    EnvelopeModel,
+    EventParserReturnType,
+    Model,
+)
 
 from ...middleware_factory import lambda_handler_decorator
 from ..typing import LambdaContext

aws_lambda_powertools/utilities/typing/lambda_client_context.py

@@ -1,7 +1,9 @@
 # -*- coding: utf-8 -*-
 from typing import Any, Dict
 
-from aws_lambda_powertools.utilities.typing.lambda_client_context_mobile_client import LambdaClientContextMobileClient
+from aws_lambda_powertools.utilities.typing.lambda_client_context_mobile_client import (
+    LambdaClientContextMobileClient,
+)
 
 
 class LambdaClientContext(object):

aws_lambda_powertools/utilities/typing/lambda_context.py

@@ -1,6 +1,10 @@
 # -*- coding: utf-8 -*-
-from aws_lambda_powertools.utilities.typing.lambda_client_context import LambdaClientContext
-from aws_lambda_powertools.utilities.typing.lambda_cognito_identity import LambdaCognitoIdentity
+from aws_lambda_powertools.utilities.typing.lambda_client_context import (
+    LambdaClientContext,
+)
+from aws_lambda_powertools.utilities.typing.lambda_cognito_identity import (
+    LambdaCognitoIdentity,
+)
 
 
 class LambdaContext(object):

aws_lambda_powertools/utilities/validation/__init__.py

@@ -2,7 +2,11 @@
 Simple validator to enforce incoming/outgoing event conforms with JSON Schema
 """
 
-from .exceptions import InvalidEnvelopeExpressionError, InvalidSchemaFormatError, SchemaValidationError
+from .exceptions import (
+    InvalidEnvelopeExpressionError,
+    InvalidSchemaFormatError,
+    SchemaValidationError,
+)
 from .validator import validate, validator
 
 __all__ = [

tests/e2e/utils/infrastructure.py

@@ -10,7 +10,16 @@
 import boto3
 import pytest
 import yaml
-from aws_cdk import App, AssetStaging, BundlingOptions, CfnOutput, DockerImage, RemovalPolicy, Stack, aws_logs
+from aws_cdk import (
+    App,
+    AssetStaging,
+    BundlingOptions,
+    CfnOutput,
+    DockerImage,
+    RemovalPolicy,
+    Stack,
+    aws_logs,
+)
 from aws_cdk.aws_lambda import Code, Function, LayerVersion, Runtime, Tracing
 from filelock import FileLock
 from mypy_boto3_cloudformation import CloudFormationClient

tests/functional/data_classes/test_amazon_mq.py

@@ -1,7 +1,14 @@
 from typing import Dict
 
-from aws_lambda_powertools.utilities.data_classes.active_mq_event import ActiveMQEvent, ActiveMQMessage
-from aws_lambda_powertools.utilities.data_classes.rabbit_mq_event import BasicProperties, RabbitMessage, RabbitMQEvent
+from aws_lambda_powertools.utilities.data_classes.active_mq_event import (
+    ActiveMQEvent,
+    ActiveMQMessage,
+)
+from aws_lambda_powertools.utilities.data_classes.rabbit_mq_event import (
+    BasicProperties,
+    RabbitMessage,
+    RabbitMQEvent,
+)
 from tests.functional.utils import load_event
 
 

tests/functional/event_handler/test_lambda_function_url.py

@@ -1,4 +1,8 @@
-from aws_lambda_powertools.event_handler import LambdaFunctionUrlResolver, Response, content_types
+from aws_lambda_powertools.event_handler import (
+    LambdaFunctionUrlResolver,
+    Response,
+    content_types,
+)
 from aws_lambda_powertools.utilities.data_classes import LambdaFunctionUrlEvent
 from tests.functional.utils import load_event
 

tests/functional/feature_flags/test_feature_flags.py

@@ -3,7 +3,10 @@
 import pytest
 from botocore.config import Config
 
-from aws_lambda_powertools.utilities.feature_flags import ConfigurationStoreError, schema
+from aws_lambda_powertools.utilities.feature_flags import (
+    ConfigurationStoreError,
+    schema,
+)
 from aws_lambda_powertools.utilities.feature_flags.appconfig import AppConfigStore
 from aws_lambda_powertools.utilities.feature_flags.exceptions import StoreClientError
 from aws_lambda_powertools.utilities.feature_flags.feature_flags import FeatureFlags

tests/functional/feature_flags/test_schema_validation.py

@@ -2,7 +2,9 @@
 
 import pytest  # noqa: F401
 
-from aws_lambda_powertools.utilities.feature_flags.exceptions import SchemaValidationError
+from aws_lambda_powertools.utilities.feature_flags.exceptions import (
+    SchemaValidationError,
+)
 from aws_lambda_powertools.utilities.feature_flags.schema import (
     CONDITION_ACTION,
     CONDITION_KEY,

tests/functional/idempotency/test_idempotency.py

@@ -10,9 +10,19 @@
 from botocore import stub
 from pydantic import BaseModel
 
-from aws_lambda_powertools.utilities.data_classes import APIGatewayProxyEventV2, event_source
-from aws_lambda_powertools.utilities.idempotency import DynamoDBPersistenceLayer, IdempotencyConfig
-from aws_lambda_powertools.utilities.idempotency.base import MAX_RETRIES, IdempotencyHandler, _prepare_data
+from aws_lambda_powertools.utilities.data_classes import (
+    APIGatewayProxyEventV2,
+    event_source,
+)
+from aws_lambda_powertools.utilities.idempotency import (
+    DynamoDBPersistenceLayer,
+    IdempotencyConfig,
+)
+from aws_lambda_powertools.utilities.idempotency.base import (
+    MAX_RETRIES,
+    IdempotencyHandler,
+    _prepare_data,
+)
 from aws_lambda_powertools.utilities.idempotency.exceptions import (
     IdempotencyAlreadyInProgressError,
     IdempotencyInconsistentStateError,
@@ -21,8 +31,14 @@
     IdempotencyPersistenceLayerError,
     IdempotencyValidationError,
 )
-from aws_lambda_powertools.utilities.idempotency.idempotency import idempotent, idempotent_function
-from aws_lambda_powertools.utilities.idempotency.persistence.base import BasePersistenceLayer, DataRecord
+from aws_lambda_powertools.utilities.idempotency.idempotency import (
+    idempotent,
+    idempotent_function,
+)
+from aws_lambda_powertools.utilities.idempotency.persistence.base import (
+    BasePersistenceLayer,
+    DataRecord,
+)
 from aws_lambda_powertools.utilities.validation import envelopes, validator
 from tests.functional.idempotency.utils import (
     build_idempotency_put_item_stub,