Skip to content

Commit d6374e9

Browse files
refactor(middleware_factory): use standard collections for types (#6485)
Using generics types
1 parent 5c2f815 commit d6374e9

File tree

3 files changed

+11
-3
lines changed

3 files changed

+11
-3
lines changed

aws_lambda_powertools/middleware_factory/__init__.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,6 @@
33
[`Middleware Factory`](../utilities/middleware_factory.md)
44
"""
55

6-
from .factory import lambda_handler_decorator
6+
from aws_lambda_powertools.middleware_factory.factory import lambda_handler_decorator
77

88
__all__ = ["lambda_handler_decorator"]

aws_lambda_powertools/middleware_factory/factory.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
import inspect
55
import logging
66
import os
7-
from typing import Any, Callable
7+
from typing import TYPE_CHECKING, Any
88

99
from aws_lambda_powertools.middleware_factory.exceptions import MiddlewareInvalidArgumentError
1010
from aws_lambda_powertools.shared import constants
@@ -13,6 +13,9 @@
1313

1414
logger = logging.getLogger(__name__)
1515

16+
if TYPE_CHECKING:
17+
from collections.abc import Callable
18+
1619

1720
# Maintenance: we can't yet provide an accurate return type without ParamSpec etc. see #1066
1821
def lambda_handler_decorator(decorator: Callable | None = None, trace_execution: bool | None = None) -> Callable:

tests/functional/middleware_factory/required_dependencies/test_middleware_factory.py

+6-1
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,7 @@
1+
from __future__ import annotations
2+
13
import json
2-
from typing import Callable
4+
from typing import TYPE_CHECKING
35

46
import pytest
57

@@ -8,6 +10,9 @@
810
MiddlewareInvalidArgumentError,
911
)
1012

13+
if TYPE_CHECKING:
14+
from collections.abc import Callable
15+
1116

1217
@pytest.fixture
1318
def say_hi_middleware() -> Callable:

0 commit comments

Comments
 (0)