Skip to content

Commit cc5e72a

Browse files
refactor(idempotency): use standard collections for types (#6487)
Using generics types
1 parent 5c2bc72 commit cc5e72a

File tree

5 files changed

+14
-3
lines changed

5 files changed

+14
-3
lines changed

aws_lambda_powertools/utilities/idempotency/base.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
import datetime
1010
import logging
1111
from copy import deepcopy
12-
from typing import TYPE_CHECKING, Any, Callable
12+
from typing import TYPE_CHECKING, Any
1313

1414
from aws_lambda_powertools.utilities.idempotency.exceptions import (
1515
IdempotencyAlreadyInProgressError,
@@ -29,6 +29,8 @@
2929
)
3030

3131
if TYPE_CHECKING:
32+
from collections.abc import Callable
33+
3234
from aws_lambda_powertools.utilities.idempotency.config import (
3335
IdempotencyConfig,
3436
)

aws_lambda_powertools/utilities/idempotency/idempotency.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
import os
1010
import warnings
1111
from inspect import isclass
12-
from typing import TYPE_CHECKING, Any, Callable, cast
12+
from typing import TYPE_CHECKING, Any, cast
1313

1414
from aws_lambda_powertools.middleware_factory import lambda_handler_decorator
1515
from aws_lambda_powertools.shared import constants
@@ -23,6 +23,8 @@
2323
)
2424

2525
if TYPE_CHECKING:
26+
from collections.abc import Callable
27+
2628
from aws_lambda_powertools.utilities.idempotency.persistence.base import (
2729
BasePersistenceLayer,
2830
)

aws_lambda_powertools/utilities/idempotency/serialization/custom_dict.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,12 @@
11
from __future__ import annotations
22

3-
from typing import Any, Callable
3+
from typing import TYPE_CHECKING, Any
44

55
from aws_lambda_powertools.utilities.idempotency.serialization.base import BaseIdempotencySerializer
66

7+
if TYPE_CHECKING:
8+
from collections.abc import Callable
9+
710

811
class CustomDictSerializer(BaseIdempotencySerializer):
912
def __init__(self, to_dict: Callable[[Any], dict], from_dict: Callable[[dict], Any]):

tests/functional/idempotency/_boto3/conftest.py

+2
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
from __future__ import annotations
2+
13
import datetime
24
import json
35
from decimal import Decimal

tests/unit/idempotency/test_dynamodb_persistence.py

+2
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
from __future__ import annotations
2+
13
from dataclasses import dataclass
24

35
from aws_lambda_powertools.utilities.idempotency import DynamoDBPersistenceLayer

0 commit comments

Comments
 (0)