Skip to content

Commit b4195eb

Browse files
refactor(middleware_factory): add from __future__ import annotations (#4941)
* refactor(middleware_factory): add from __future__ import annotations and update code according to ruff rules TCH, UP006, UP007, UP037 and FA100. * Prefer absolute imports --------- Co-authored-by: Leandro Damascena <[email protected]>
1 parent a57bbfd commit b4195eb

File tree

1 file changed

+9
-7
lines changed
  • aws_lambda_powertools/middleware_factory

1 file changed

+9
-7
lines changed

aws_lambda_powertools/middleware_factory/factory.py

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,21 @@
1+
from __future__ import annotations
2+
13
import functools
24
import inspect
35
import logging
46
import os
5-
from typing import Any, Callable, Optional
7+
from typing import Any, Callable
68

7-
from ..shared import constants
8-
from ..shared.functions import resolve_truthy_env_var_choice
9-
from ..tracing import Tracer
10-
from .exceptions import MiddlewareInvalidArgumentError
9+
from aws_lambda_powertools.middleware_factory.exceptions import MiddlewareInvalidArgumentError
10+
from aws_lambda_powertools.shared import constants
11+
from aws_lambda_powertools.shared.functions import resolve_truthy_env_var_choice
12+
from aws_lambda_powertools.tracing import Tracer
1113

1214
logger = logging.getLogger(__name__)
1315

1416

1517
# Maintenance: we can't yet provide an accurate return type without ParamSpec etc. see #1066
16-
def lambda_handler_decorator(decorator: Optional[Callable] = None, trace_execution: Optional[bool] = None) -> Callable:
18+
def lambda_handler_decorator(decorator: Callable | None = None, trace_execution: bool | None = None) -> Callable:
1719
"""Decorator factory for decorating Lambda handlers.
1820
1921
You can use lambda_handler_decorator to create your own middlewares,
@@ -112,7 +114,7 @@ def lambda_handler(event, context):
112114
)
113115

114116
@functools.wraps(decorator)
115-
def final_decorator(func: Optional[Callable] = None, **kwargs: Any):
117+
def final_decorator(func: Callable | None = None, **kwargs: Any):
116118
# If called with kwargs return new func with kwargs
117119
if func is None:
118120
return functools.partial(final_decorator, **kwargs)

0 commit comments

Comments
 (0)