diff --git a/aws_lambda_powertools/logging/__init__.py b/aws_lambda_powertools/logging/__init__.py index 0456b202ffa..2c9532ef540 100644 --- a/aws_lambda_powertools/logging/__init__.py +++ b/aws_lambda_powertools/logging/__init__.py @@ -1,5 +1,6 @@ """Logging utility """ + from .logger import Logger __all__ = ["Logger"] diff --git a/aws_lambda_powertools/logging/logger.py b/aws_lambda_powertools/logging/logger.py index ab159061bff..1e21d6680af 100644 --- a/aws_lambda_powertools/logging/logger.py +++ b/aws_lambda_powertools/logging/logger.py @@ -345,8 +345,7 @@ def inject_lambda_context( log_event: Optional[bool] = None, correlation_id_path: Optional[str] = None, clear_state: Optional[bool] = False, - ) -> AnyCallableT: - ... + ) -> AnyCallableT: ... @overload def inject_lambda_context( @@ -355,8 +354,7 @@ def inject_lambda_context( log_event: Optional[bool] = None, correlation_id_path: Optional[str] = None, clear_state: Optional[bool] = False, - ) -> Callable[[AnyCallableT], AnyCallableT]: - ... + ) -> Callable[[AnyCallableT], AnyCallableT]: ... def inject_lambda_context( self, diff --git a/aws_lambda_powertools/metrics/__init__.py b/aws_lambda_powertools/metrics/__init__.py index b8c94478816..cafd348b8ec 100644 --- a/aws_lambda_powertools/metrics/__init__.py +++ b/aws_lambda_powertools/metrics/__init__.py @@ -1,5 +1,6 @@ """CloudWatch Embedded Metric Format utility """ + from aws_lambda_powertools.metrics.base import MetricResolution, MetricUnit, single_metric from aws_lambda_powertools.metrics.exceptions import ( MetricResolutionError, diff --git a/aws_lambda_powertools/middleware_factory/__init__.py b/aws_lambda_powertools/middleware_factory/__init__.py index 9d57d843ec2..b44d49d6987 100644 --- a/aws_lambda_powertools/middleware_factory/__init__.py +++ b/aws_lambda_powertools/middleware_factory/__init__.py @@ -1,4 +1,5 @@ """ Utilities to enhance middlewares """ + from .factory import lambda_handler_decorator __all__ = ["lambda_handler_decorator"] diff --git a/aws_lambda_powertools/shared/functions.py b/aws_lambda_powertools/shared/functions.py index 9765f55c025..ee274498260 100644 --- a/aws_lambda_powertools/shared/functions.py +++ b/aws_lambda_powertools/shared/functions.py @@ -57,18 +57,15 @@ def resolve_max_age(env: str, choice: Optional[int]) -> int: @overload -def resolve_env_var_choice(env: Optional[str], choice: float) -> float: - ... +def resolve_env_var_choice(env: Optional[str], choice: float) -> float: ... @overload -def resolve_env_var_choice(env: Optional[str], choice: str) -> str: - ... +def resolve_env_var_choice(env: Optional[str], choice: str) -> str: ... @overload -def resolve_env_var_choice(env: Optional[str], choice: Optional[str]) -> str: - ... +def resolve_env_var_choice(env: Optional[str], choice: Optional[str]) -> str: ... def resolve_env_var_choice( diff --git a/aws_lambda_powertools/tracing/__init__.py b/aws_lambda_powertools/tracing/__init__.py index f45ac1fb73e..1031ae4aec6 100644 --- a/aws_lambda_powertools/tracing/__init__.py +++ b/aws_lambda_powertools/tracing/__init__.py @@ -1,7 +1,6 @@ """Tracing utility """ - from .extensions import aiohttp_trace_config from .tracer import Tracer diff --git a/aws_lambda_powertools/tracing/tracer.py b/aws_lambda_powertools/tracing/tracer.py index 3004e5505c8..b30467188f9 100644 --- a/aws_lambda_powertools/tracing/tracer.py +++ b/aws_lambda_powertools/tracing/tracer.py @@ -345,8 +345,7 @@ def decorate(event, context, **kwargs): # see #465 @overload - def capture_method(self, method: "AnyCallableT") -> "AnyCallableT": - ... # pragma: no cover + def capture_method(self, method: "AnyCallableT") -> "AnyCallableT": ... # pragma: no cover @overload def capture_method( @@ -354,8 +353,7 @@ def capture_method( method: None = None, capture_response: Optional[bool] = None, capture_error: Optional[bool] = None, - ) -> Callable[["AnyCallableT"], "AnyCallableT"]: - ... # pragma: no cover + ) -> Callable[["AnyCallableT"], "AnyCallableT"]: ... # pragma: no cover def capture_method( self, diff --git a/aws_lambda_powertools/utilities/batch/base.py b/aws_lambda_powertools/utilities/batch/base.py index a158eafdad0..569467f2248 100644 --- a/aws_lambda_powertools/utilities/batch/base.py +++ b/aws_lambda_powertools/utilities/batch/base.py @@ -339,12 +339,15 @@ def _collect_dynamodb_failures(self): return failures @overload - def _to_batch_type(self, record: dict, event_type: EventType, model: "BatchTypeModels") -> "BatchTypeModels": - ... # pragma: no cover + def _to_batch_type( + self, + record: dict, + event_type: EventType, + model: "BatchTypeModels", + ) -> "BatchTypeModels": ... # pragma: no cover @overload - def _to_batch_type(self, record: dict, event_type: EventType) -> EventSourceDataClassTypes: - ... # pragma: no cover + def _to_batch_type(self, record: dict, event_type: EventType) -> EventSourceDataClassTypes: ... # pragma: no cover def _to_batch_type(self, record: dict, event_type: EventType, model: Optional["BatchTypeModels"] = None): if model is not None: diff --git a/aws_lambda_powertools/utilities/batch/exceptions.py b/aws_lambda_powertools/utilities/batch/exceptions.py index 89659e42bf8..a3eefbb9cea 100644 --- a/aws_lambda_powertools/utilities/batch/exceptions.py +++ b/aws_lambda_powertools/utilities/batch/exceptions.py @@ -1,6 +1,7 @@ """ Batch processing exceptions """ + from __future__ import annotations import traceback diff --git a/aws_lambda_powertools/utilities/data_classes/common.py b/aws_lambda_powertools/utilities/data_classes/common.py index 41fbe5cd0de..104993deae8 100644 --- a/aws_lambda_powertools/utilities/data_classes/common.py +++ b/aws_lambda_powertools/utilities/data_classes/common.py @@ -193,7 +193,7 @@ def get_multi_value_query_string_values( self, name: str, default_values: Optional[List[str]] = None, - ) ->List[str]: + ) -> List[str]: """Get multi-value query string parameter values by name Parameters diff --git a/aws_lambda_powertools/utilities/data_masking/base.py b/aws_lambda_powertools/utilities/data_masking/base.py index c2557dcef24..5274aac3b8a 100644 --- a/aws_lambda_powertools/utilities/data_masking/base.py +++ b/aws_lambda_powertools/utilities/data_masking/base.py @@ -85,20 +85,16 @@ def decrypt( ) @overload - def erase(self, data, fields: None) -> str: - ... + def erase(self, data, fields: None) -> str: ... @overload - def erase(self, data: list, fields: list[str]) -> list[str]: - ... + def erase(self, data: list, fields: list[str]) -> list[str]: ... @overload - def erase(self, data: tuple, fields: list[str]) -> tuple[str]: - ... + def erase(self, data: tuple, fields: list[str]) -> tuple[str]: ... @overload - def erase(self, data: dict, fields: list[str]) -> dict: - ... + def erase(self, data: dict, fields: list[str]) -> dict: ... def erase(self, data: Sequence | Mapping, fields: list[str] | None = None) -> str | list[str] | tuple[str] | dict: return self._apply_action(data=data, fields=fields, action=self.provider.erase) diff --git a/aws_lambda_powertools/utilities/data_masking/provider/kms/aws_encryption_sdk.py b/aws_lambda_powertools/utilities/data_masking/provider/kms/aws_encryption_sdk.py index bbdbb0bad6f..657a812337f 100644 --- a/aws_lambda_powertools/utilities/data_masking/provider/kms/aws_encryption_sdk.py +++ b/aws_lambda_powertools/utilities/data_masking/provider/kms/aws_encryption_sdk.py @@ -105,7 +105,6 @@ def decrypt(self, data: str, provider_options: dict | None = None, **encryption_ class KMSKeyProvider: - """ The KMSKeyProvider is responsible for assembling an AWS Key Management Service (KMS) client, a caching mechanism, and a keyring for secure key management and data encryption. diff --git a/aws_lambda_powertools/utilities/idempotency/exceptions.py b/aws_lambda_powertools/utilities/idempotency/exceptions.py index 55ec662799e..27f319a5266 100644 --- a/aws_lambda_powertools/utilities/idempotency/exceptions.py +++ b/aws_lambda_powertools/utilities/idempotency/exceptions.py @@ -2,7 +2,6 @@ Idempotency errors """ - from typing import Optional, Union from aws_lambda_powertools.utilities.idempotency.persistence.datarecord import DataRecord diff --git a/aws_lambda_powertools/utilities/idempotency/idempotency.py b/aws_lambda_powertools/utilities/idempotency/idempotency.py index 94306dfa2c5..9593655b3cd 100644 --- a/aws_lambda_powertools/utilities/idempotency/idempotency.py +++ b/aws_lambda_powertools/utilities/idempotency/idempotency.py @@ -1,6 +1,7 @@ """ Primary interface for idempotent Lambda functions utility """ + import functools import logging import os diff --git a/aws_lambda_powertools/utilities/idempotency/serialization/base.py b/aws_lambda_powertools/utilities/idempotency/serialization/base.py index 45317bd0315..ba41b23dbe4 100644 --- a/aws_lambda_powertools/utilities/idempotency/serialization/base.py +++ b/aws_lambda_powertools/utilities/idempotency/serialization/base.py @@ -1,6 +1,7 @@ """ Serialization for supporting idempotency """ + from abc import ABC, abstractmethod from typing import Any, Dict diff --git a/aws_lambda_powertools/utilities/parameters/appconfig.py b/aws_lambda_powertools/utilities/parameters/appconfig.py index d5a9b7856e4..b29d256a0a0 100644 --- a/aws_lambda_powertools/utilities/parameters/appconfig.py +++ b/aws_lambda_powertools/utilities/parameters/appconfig.py @@ -2,7 +2,6 @@ AWS App Config configuration retrieval and caching utility """ - import os from typing import TYPE_CHECKING, Any, Dict, Optional, Union diff --git a/aws_lambda_powertools/utilities/parameters/base.py b/aws_lambda_powertools/utilities/parameters/base.py index 710634636d0..5ce06589613 100644 --- a/aws_lambda_powertools/utilities/parameters/base.py +++ b/aws_lambda_powertools/utilities/parameters/base.py @@ -1,6 +1,7 @@ """ Base for Parameter providers """ + from __future__ import annotations import base64 @@ -372,8 +373,7 @@ def transform_value( transform: TransformOptions, raise_on_transform_error: bool = False, key: str = "", -) -> Dict[str, Any]: - ... +) -> Dict[str, Any]: ... @overload @@ -382,8 +382,7 @@ def transform_value( transform: TransformOptions, raise_on_transform_error: bool = False, key: str = "", -) -> Optional[Union[str, bytes, Dict[str, Any]]]: - ... +) -> Optional[Union[str, bytes, Dict[str, Any]]]: ... def transform_value( diff --git a/aws_lambda_powertools/utilities/parameters/dynamodb.py b/aws_lambda_powertools/utilities/parameters/dynamodb.py index d6c82fd3293..e008e3b62db 100644 --- a/aws_lambda_powertools/utilities/parameters/dynamodb.py +++ b/aws_lambda_powertools/utilities/parameters/dynamodb.py @@ -2,7 +2,6 @@ Amazon DynamoDB parameter retrieval and caching utility """ - from typing import TYPE_CHECKING, Dict, Optional import boto3 diff --git a/aws_lambda_powertools/utilities/parameters/secrets.py b/aws_lambda_powertools/utilities/parameters/secrets.py index dc7bc2b0053..beb4bb80846 100644 --- a/aws_lambda_powertools/utilities/parameters/secrets.py +++ b/aws_lambda_powertools/utilities/parameters/secrets.py @@ -2,7 +2,6 @@ AWS Secrets Manager parameter retrieval and caching utility """ - import os from typing import TYPE_CHECKING, Any, Dict, Optional, Union diff --git a/aws_lambda_powertools/utilities/parameters/ssm.py b/aws_lambda_powertools/utilities/parameters/ssm.py index 7df0e66c027..1be07e7c9f0 100644 --- a/aws_lambda_powertools/utilities/parameters/ssm.py +++ b/aws_lambda_powertools/utilities/parameters/ssm.py @@ -1,6 +1,7 @@ """ AWS SSM Parameter retrieval and caching utility """ + from __future__ import annotations import os @@ -553,8 +554,7 @@ def get_parameter( force_fetch: bool = False, max_age: Optional[int] = None, **sdk_options, -) -> str: - ... +) -> str: ... @overload @@ -565,8 +565,7 @@ def get_parameter( force_fetch: bool = False, max_age: Optional[int] = None, **sdk_options, -) -> dict: - ... +) -> dict: ... @overload @@ -577,8 +576,7 @@ def get_parameter( force_fetch: bool = False, max_age: Optional[int] = None, **sdk_options, -) -> Union[str, dict, bytes]: - ... +) -> Union[str, dict, bytes]: ... @overload @@ -589,8 +587,7 @@ def get_parameter( force_fetch: bool = False, max_age: Optional[int] = None, **sdk_options, -) -> bytes: - ... +) -> bytes: ... def get_parameter( @@ -683,8 +680,7 @@ def get_parameters( max_age: Optional[int] = None, raise_on_transform_error: bool = False, **sdk_options, -) -> Dict[str, str]: - ... +) -> Dict[str, str]: ... @overload @@ -697,8 +693,7 @@ def get_parameters( max_age: Optional[int] = None, raise_on_transform_error: bool = False, **sdk_options, -) -> Dict[str, dict]: - ... +) -> Dict[str, dict]: ... @overload @@ -711,8 +706,7 @@ def get_parameters( max_age: Optional[int] = None, raise_on_transform_error: bool = False, **sdk_options, -) -> Dict[str, bytes]: - ... +) -> Dict[str, bytes]: ... @overload @@ -725,8 +719,7 @@ def get_parameters( max_age: Optional[int] = None, raise_on_transform_error: bool = False, **sdk_options, -) -> Union[Dict[str, bytes], Dict[str, dict], Dict[str, str]]: - ... +) -> Union[Dict[str, bytes], Dict[str, dict], Dict[str, str]]: ... def get_parameters( @@ -825,8 +818,7 @@ def get_parameters_by_name( decrypt: Optional[bool] = None, max_age: Optional[int] = None, raise_on_error: bool = True, -) -> Dict[str, str]: - ... +) -> Dict[str, str]: ... @overload @@ -836,8 +828,7 @@ def get_parameters_by_name( decrypt: Optional[bool] = None, max_age: Optional[int] = None, raise_on_error: bool = True, -) -> Dict[str, bytes]: - ... +) -> Dict[str, bytes]: ... @overload @@ -847,8 +838,7 @@ def get_parameters_by_name( decrypt: Optional[bool] = None, max_age: Optional[int] = None, raise_on_error: bool = True, -) -> Dict[str, Dict[str, Any]]: - ... +) -> Dict[str, Dict[str, Any]]: ... @overload @@ -858,8 +848,7 @@ def get_parameters_by_name( decrypt: Optional[bool] = None, max_age: Optional[int] = None, raise_on_error: bool = True, -) -> Union[Dict[str, str], Dict[str, dict]]: - ... +) -> Union[Dict[str, str], Dict[str, dict]]: ... def get_parameters_by_name( diff --git a/aws_lambda_powertools/utilities/parser/__init__.py b/aws_lambda_powertools/utilities/parser/__init__.py index 1bc67934b13..ad19168bb29 100644 --- a/aws_lambda_powertools/utilities/parser/__init__.py +++ b/aws_lambda_powertools/utilities/parser/__init__.py @@ -1,5 +1,6 @@ """Advanced event_parser utility """ + from . import envelopes from .envelopes import BaseEnvelope from .parser import event_parser, parse diff --git a/aws_lambda_powertools/utilities/parser/parser.py b/aws_lambda_powertools/utilities/parser/parser.py index 91d2763f4ec..a9a89db8a6a 100644 --- a/aws_lambda_powertools/utilities/parser/parser.py +++ b/aws_lambda_powertools/utilities/parser/parser.py @@ -104,13 +104,11 @@ def handler(event: Order, context: LambdaContext): @overload -def parse(event: Dict[str, Any], model: Type[Model]) -> Model: - ... # pragma: no cover +def parse(event: Dict[str, Any], model: Type[Model]) -> Model: ... # pragma: no cover @overload -def parse(event: Dict[str, Any], model: Type[Model], envelope: Type[Envelope]) -> Model: - ... # pragma: no cover +def parse(event: Dict[str, Any], model: Type[Model], envelope: Type[Envelope]) -> Model: ... # pragma: no cover def parse(event: Dict[str, Any], model: Type[Model], envelope: Optional[Type[Envelope]] = None): diff --git a/aws_lambda_powertools/utilities/serialization.py b/aws_lambda_powertools/utilities/serialization.py index ef76eec70e2..cb5289ae4af 100644 --- a/aws_lambda_powertools/utilities/serialization.py +++ b/aws_lambda_powertools/utilities/serialization.py @@ -1,4 +1,5 @@ """Standalone functions to serialize/deserialize common data structures""" + import base64 import json from typing import Any, Callable diff --git a/poetry.lock b/poetry.lock index 678896ba721..1fc250f11ce 100644 --- a/poetry.lock +++ b/poetry.lock @@ -1,4 +1,4 @@ -# This file is automatically @generated by Poetry 1.8.0 and should not be changed by hand. +# This file is automatically @generated by Poetry 1.8.1 and should not be changed by hand. [[package]] name = "anyio" @@ -330,17 +330,17 @@ uvloop = ["uvloop (>=0.15.2)"] [[package]] name = "boto3" -version = "1.34.49" +version = "1.34.50" description = "The AWS SDK for Python" optional = false python-versions = ">= 3.8" files = [ - {file = "boto3-1.34.49-py3-none-any.whl", hash = "sha256:ce8d1de03024f52a1810e8d71ad4dba3a5b9bb48b35567191500e3432a9130b4"}, - {file = "boto3-1.34.49.tar.gz", hash = "sha256:96b9dc85ce8d52619b56ca7b1ac1423eaf0af5ce132904bcc8aa81396eec2abf"}, + {file = "boto3-1.34.50-py3-none-any.whl", hash = "sha256:8d709365231234bc4f0ca98fdf33a25eeebf78072853c6aa3d259f0f5cf09877"}, + {file = "boto3-1.34.50.tar.gz", hash = "sha256:290952be7899560039cb0042e8a2354f61a7dead0d0ca8bea6ba901930df0468"}, ] [package.dependencies] -botocore = ">=1.34.49,<1.35.0" +botocore = ">=1.34.50,<1.35.0" jmespath = ">=0.7.1,<2.0.0" s3transfer = ">=0.10.0,<0.11.0" @@ -349,13 +349,13 @@ crt = ["botocore[crt] (>=1.21.0,<2.0a0)"] [[package]] name = "botocore" -version = "1.34.49" +version = "1.34.50" description = "Low-level, data-driven core of boto 3." optional = false python-versions = ">= 3.8" files = [ - {file = "botocore-1.34.49-py3-none-any.whl", hash = "sha256:4ed9d7603a04b5bb5bd5de63b513bc2c8a7e8b1cd0088229c5ceb461161f43b6"}, - {file = "botocore-1.34.49.tar.gz", hash = "sha256:d89410bc60673eaff1699f3f1fdcb0e3a5e1f7a6a048c0d88c3ce5c3549433ec"}, + {file = "botocore-1.34.50-py3-none-any.whl", hash = "sha256:fda510559dbe796eefdb59561cc81be1b99afba3dee53fd23db9a3d587adc0ab"}, + {file = "botocore-1.34.50.tar.gz", hash = "sha256:33ab82cb96c4bb684f0dbafb071808e4817d83debc88b223e7d988256370c6d7"}, ] [package.dependencies] @@ -2940,13 +2940,13 @@ pbr = "*" [[package]] name = "sentry-sdk" -version = "1.40.5" +version = "1.40.6" description = "Python client for Sentry (https://sentry.io)" optional = false python-versions = "*" files = [ - {file = "sentry-sdk-1.40.5.tar.gz", hash = "sha256:d2dca2392cc5c9a2cc9bb874dd7978ebb759682fe4fe889ee7e970ee8dd1c61e"}, - {file = "sentry_sdk-1.40.5-py2.py3-none-any.whl", hash = "sha256:d188b407c9bacbe2a50a824e1f8fb99ee1aeb309133310488c570cb6d7056643"}, + {file = "sentry-sdk-1.40.6.tar.gz", hash = "sha256:f143f3fb4bb57c90abef6e2ad06b5f6f02b2ca13e4060ec5c0549c7a9ccce3fa"}, + {file = "sentry_sdk-1.40.6-py2.py3-none-any.whl", hash = "sha256:becda09660df63e55f307570e9817c664392655a7328bbc414b507e9cb874c67"}, ] [package.dependencies] @@ -2972,7 +2972,7 @@ huey = ["huey (>=2)"] loguru = ["loguru (>=0.5)"] opentelemetry = ["opentelemetry-distro (>=0.35b0)"] opentelemetry-experimental = ["opentelemetry-distro (>=0.40b0,<1.0)", "opentelemetry-instrumentation-aiohttp-client (>=0.40b0,<1.0)", "opentelemetry-instrumentation-django (>=0.40b0,<1.0)", "opentelemetry-instrumentation-fastapi (>=0.40b0,<1.0)", "opentelemetry-instrumentation-flask (>=0.40b0,<1.0)", "opentelemetry-instrumentation-requests (>=0.40b0,<1.0)", "opentelemetry-instrumentation-sqlite3 (>=0.40b0,<1.0)", "opentelemetry-instrumentation-urllib (>=0.40b0,<1.0)"] -pure-eval = ["asttokens", "executing", "pure-eval"] +pure-eval = ["asttokens", "executing", "pure_eval"] pymongo = ["pymongo (>=3.1)"] pyspark = ["pyspark (>=2.4.4)"] quart = ["blinker (>=1.1)", "quart (>=0.16.1)"] @@ -3432,4 +3432,4 @@ validation = ["fastjsonschema"] [metadata] lock-version = "2.0" python-versions = ">=3.8,<4.0.0" -content-hash = "ff8aabf1c8d72613c8b43d6f85bd1127acfbc2f4d6ad7cc352e4f87ebaf6222a" +content-hash = "56925f0c4bf816e507650c4f78ea418822a2afb8a9f6d6d7828127811a9a8223" diff --git a/pyproject.toml b/pyproject.toml index 0df5354c4f3..ac571066035 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -52,7 +52,7 @@ jsonpath-ng = { version = "^1.6.0", optional = true } [tool.poetry.dev-dependencies] coverage = { extras = ["toml"], version = "^7.4" } pytest = "^8.0.1" -black = "^24.1" +black = "^24.2" boto3 = "^1.26.164" isort = "^5.13.2" pytest-cov = "^4.1.0" diff --git a/tests/functional/event_handler/test_api_gateway.py b/tests/functional/event_handler/test_api_gateway.py index 9c98faff062..fa166bac77e 100644 --- a/tests/functional/event_handler/test_api_gateway.py +++ b/tests/functional/event_handler/test_api_gateway.py @@ -716,16 +716,13 @@ def test_cors_preflight(): app = ApiGatewayResolver(cors=CORSConfig()) @app.get("/foo") - def foo_cors(): - ... + def foo_cors(): ... @app.route(method="delete", rule="/foo") - def foo_delete_cors(): - ... + def foo_delete_cors(): ... @app.post("/foo", cors=False) - def post_no_cors(): - ... + def post_no_cors(): ... # WHEN calling the handler result = app({"path": "/foo", "httpMethod": "OPTIONS", "headers": {"Origin": "http://example.org"}}, None) @@ -756,8 +753,7 @@ def custom_preflight(): ) @app.route(method="CUSTOM", rule="/some-call", cors=True) - def custom_method(): - ... + def custom_method(): ... # AND the request includes an origin headers = {"Origin": "https://example.org"} @@ -1094,8 +1090,7 @@ def pay_foo(): raise ValueError("should not be matching") @app.get("/foo") - def foo(): - ... + def foo(): ... # WHEN calling handler response = app({"httpMethod": "GET", "path": path}, None) @@ -1116,8 +1111,7 @@ def test_remove_prefix_by_regex(path: str): app = ApiGatewayResolver(strip_prefixes=[re.compile(r"/(dev|stg)")]) @app.get("/foo") - def foo(): - ... + def foo(): ... response = app({"httpMethod": "GET", "path": path}, None) @@ -1128,8 +1122,7 @@ def test_empty_path_when_using_regexes(): app = ApiGatewayResolver(strip_prefixes=[re.compile(r"/(dev|stg)")]) @app.get("/") - def foo(): - ... + def foo(): ... response = app({"httpMethod": "GET", "path": "/dev"}, None) @@ -1151,8 +1144,7 @@ def test_ignore_invalid(prefix): app = ApiGatewayResolver(strip_prefixes=prefix) @app.get("/foo/status") - def foo(): - ... + def foo(): ... # WHEN calling handler response = app({"httpMethod": "GET", "path": "/foo/status"}, None) @@ -1473,8 +1465,7 @@ def handle_validation_error(ex: RequestValidationError): ) @app.get("/my/path") - def get_lambda(param: int): - ... + def get_lambda(param: int): ... # WHEN calling the event handler # AND a RequestValidationError is raised @@ -1502,8 +1493,7 @@ def handle_validation_error(ex: RequestValidationError): ) @app.get("/my/path") - def get_lambda(param: int): - ... + def get_lambda(param: int): ... # WHEN calling the event handler # AND a RequestValidationError is raised @@ -1519,8 +1509,7 @@ def test_data_validation_error(): app = ApiGatewayResolver(enable_validation=True) @app.get("/my/path") - def get_lambda(param: int): - ... + def get_lambda(param: int): ... # WHEN calling the event handler # AND a RequestValidationError is raised diff --git a/tests/functional/test_logger.py b/tests/functional/test_logger.py index fc6e5b98ee8..d9e3b8d4e37 100644 --- a/tests/functional/test_logger.py +++ b/tests/functional/test_logger.py @@ -575,8 +575,7 @@ def test_logger_set_correlation_id_path_custom_functions(lambda_context, stdout, logger = Logger(service=service_name, stream=stdout) @logger.inject_lambda_context(correlation_id_path="Records[*].powertools_json(body).id") - def handler(event, context): - ... + def handler(event, context): ... # WHEN handler is called request_id = "xxx-111-222" @@ -719,8 +718,7 @@ def handler(event, context): def test_logger_custom_formatter_has_standard_and_custom_keys(stdout, service_name, lambda_context): - class CustomFormatter(LambdaPowertoolsFormatter): - ... + class CustomFormatter(LambdaPowertoolsFormatter): ... # GIVEN a Logger is initialized with a custom formatter logger = Logger(service=service_name, stream=stdout, logger_formatter=CustomFormatter(), my_key="value") diff --git a/tests/functional/test_logger_powertools_formatter.py b/tests/functional/test_logger_powertools_formatter.py index 62f3f36c025..b9113485a47 100644 --- a/tests/functional/test_logger_powertools_formatter.py +++ b/tests/functional/test_logger_powertools_formatter.py @@ -1,4 +1,5 @@ """aws_lambda_logging tests.""" + import io import json import os diff --git a/tests/functional/test_utilities_parameters.py b/tests/functional/test_utilities_parameters.py index 2ee544f87c8..5ff043f7ed3 100644 --- a/tests/functional/test_utilities_parameters.py +++ b/tests/functional/test_utilities_parameters.py @@ -655,8 +655,7 @@ class TestProvider(BaseProvider): def _get(self, name: str, **kwargs) -> str: return mock_value - def _get_multiple(self, path: str, **kwargs) -> Dict[str, str]: - ... + def _get_multiple(self, path: str, **kwargs) -> Dict[str, str]: ... monkeypatch.setitem(parameters.base.DEFAULT_PROVIDERS, "ssm", TestProvider()) monkeypatch.setitem(parameters.base.DEFAULT_PROVIDERS, "secrets", TestProvider()) diff --git a/tests/unit/parser/test_kinesis.py b/tests/unit/parser/test_kinesis.py index e8b1ae87378..730759f1230 100644 --- a/tests/unit/parser/test_kinesis.py +++ b/tests/unit/parser/test_kinesis.py @@ -94,8 +94,7 @@ def test_kinesis_stream_event_cloudwatch_logs_data_extraction(): def test_kinesis_stream_event_cloudwatch_logs_data_extraction_fails_with_custom_model(): # GIVEN a custom model replaces Kinesis Record Data bytes - class DummyModel(BaseModel): - ... + class DummyModel(BaseModel): ... raw_event = load_event("kinesisStreamCloudWatchLogsEvent.json") stream_data = KinesisDataStreamModel(**raw_event) diff --git a/tests/unit/test_tracing.py b/tests/unit/test_tracing.py index fdb7310495b..7b09bcde885 100644 --- a/tests/unit/test_tracing.py +++ b/tests/unit/test_tracing.py @@ -48,8 +48,7 @@ def in_subsegment(self, *args, **kwargs): def patch(self, *args, **kwargs): return self.patch_mock(*args, **kwargs) - def patch_all(self): - ... + def patch_all(self): ... return CustomProvider