|
83 | 83 | TypeModelOrEnum, |
84 | 84 | ) |
85 | 85 | from aws_lambda_powertools.shared.cookies import Cookie |
| 86 | + from aws_lambda_powertools.shared.types import AnyCallableT |
86 | 87 | from aws_lambda_powertools.utilities.typing import LambdaContext |
87 | 88 |
|
88 | 89 |
|
@@ -924,7 +925,7 @@ def route( |
924 | 925 | security: list[dict[str, list[str]]] | None = None, |
925 | 926 | openapi_extensions: dict[str, Any] | None = None, |
926 | 927 | middlewares: list[Callable[..., Any]] | None = None, |
927 | | - ): |
| 928 | + ) -> Callable[[AnyCallableT], AnyCallableT]: |
928 | 929 | raise NotImplementedError() |
929 | 930 |
|
930 | 931 | def use(self, middlewares: list[Callable[..., Response]]) -> None: |
@@ -984,7 +985,7 @@ def get( |
984 | 985 | security: list[dict[str, list[str]]] | None = None, |
985 | 986 | openapi_extensions: dict[str, Any] | None = None, |
986 | 987 | middlewares: list[Callable[..., Any]] | None = None, |
987 | | - ): |
| 988 | + ) -> Callable[[AnyCallableT], AnyCallableT]: |
988 | 989 | """Get route decorator with GET `method` |
989 | 990 |
|
990 | 991 | Examples |
@@ -1041,7 +1042,7 @@ def post( |
1041 | 1042 | security: list[dict[str, list[str]]] | None = None, |
1042 | 1043 | openapi_extensions: dict[str, Any] | None = None, |
1043 | 1044 | middlewares: list[Callable[..., Any]] | None = None, |
1044 | | - ): |
| 1045 | + ) -> Callable[[AnyCallableT], AnyCallableT]: |
1045 | 1046 | """Post route decorator with POST `method` |
1046 | 1047 |
|
1047 | 1048 | Examples |
@@ -1099,7 +1100,7 @@ def put( |
1099 | 1100 | security: list[dict[str, list[str]]] | None = None, |
1100 | 1101 | openapi_extensions: dict[str, Any] | None = None, |
1101 | 1102 | middlewares: list[Callable[..., Any]] | None = None, |
1102 | | - ): |
| 1103 | + ) -> Callable[[AnyCallableT], AnyCallableT]: |
1103 | 1104 | """Put route decorator with PUT `method` |
1104 | 1105 |
|
1105 | 1106 | Examples |
@@ -1157,7 +1158,7 @@ def delete( |
1157 | 1158 | security: list[dict[str, list[str]]] | None = None, |
1158 | 1159 | openapi_extensions: dict[str, Any] | None = None, |
1159 | 1160 | middlewares: list[Callable[..., Any]] | None = None, |
1160 | | - ): |
| 1161 | + ) -> Callable[[AnyCallableT], AnyCallableT]: |
1161 | 1162 | """Delete route decorator with DELETE `method` |
1162 | 1163 |
|
1163 | 1164 | Examples |
@@ -1214,7 +1215,7 @@ def patch( |
1214 | 1215 | security: list[dict[str, list[str]]] | None = None, |
1215 | 1216 | openapi_extensions: dict[str, Any] | None = None, |
1216 | 1217 | middlewares: list[Callable] | None = None, |
1217 | | - ): |
| 1218 | + ) -> Callable[[AnyCallableT], AnyCallableT]: |
1218 | 1219 | """Patch route decorator with PATCH `method` |
1219 | 1220 |
|
1220 | 1221 | Examples |
@@ -1274,7 +1275,7 @@ def head( |
1274 | 1275 | security: list[dict[str, list[str]]] | None = None, |
1275 | 1276 | openapi_extensions: dict[str, Any] | None = None, |
1276 | 1277 | middlewares: list[Callable] | None = None, |
1277 | | - ): |
| 1278 | + ) -> Callable[[AnyCallableT], AnyCallableT]: |
1278 | 1279 | """Head route decorator with HEAD `method` |
1279 | 1280 |
|
1280 | 1281 | Examples |
@@ -1950,10 +1951,10 @@ def route( |
1950 | 1951 | security: list[dict[str, list[str]]] | None = None, |
1951 | 1952 | openapi_extensions: dict[str, Any] | None = None, |
1952 | 1953 | middlewares: list[Callable[..., Any]] | None = None, |
1953 | | - ): |
| 1954 | + ) -> Callable[[AnyCallableT], AnyCallableT]: |
1954 | 1955 | """Route decorator includes parameter `method`""" |
1955 | 1956 |
|
1956 | | - def register_resolver(func: Callable): |
| 1957 | + def register_resolver(func: AnyCallableT) -> AnyCallableT: |
1957 | 1958 | methods = (method,) if isinstance(method, str) else method |
1958 | 1959 | logger.debug(f"Adding route using rule {rule} and methods: {','.join(m.upper() for m in methods)}") |
1959 | 1960 |
|
@@ -2492,8 +2493,8 @@ def route( |
2492 | 2493 | security: list[dict[str, list[str]]] | None = None, |
2493 | 2494 | openapi_extensions: dict[str, Any] | None = None, |
2494 | 2495 | middlewares: list[Callable[..., Any]] | None = None, |
2495 | | - ): |
2496 | | - def register_route(func: Callable): |
| 2496 | + ) -> Callable[[AnyCallableT], AnyCallableT]: |
| 2497 | + def register_route(func: AnyCallableT) -> AnyCallableT: |
2497 | 2498 | # All dict keys needs to be hashable. So we'll need to do some conversions: |
2498 | 2499 | methods = (method,) if isinstance(method, str) else tuple(method) |
2499 | 2500 | frozen_responses = _FrozenDict(responses) if responses else None |
@@ -2598,7 +2599,7 @@ def route( |
2598 | 2599 | security: list[dict[str, list[str]]] | None = None, |
2599 | 2600 | openapi_extensions: dict[str, Any] | None = None, |
2600 | 2601 | middlewares: list[Callable[..., Any]] | None = None, |
2601 | | - ): |
| 2602 | + ) -> Callable[[AnyCallableT], AnyCallableT]: |
2602 | 2603 | # NOTE: see #1552 for more context. |
2603 | 2604 | return super().route( |
2604 | 2605 | rule.rstrip("/"), |
|
0 commit comments