Adding shared

Author: leandrodamascena

aws_lambda_powertools/shared/dynamodb_deserializer.py

@@ -1,7 +1,10 @@
 from __future__ import annotations
 
 from decimal import Clamped, Context, Decimal, Inexact, Overflow, Rounded, Underflow
-from typing import Any, Callable, Sequence
+from typing import TYPE_CHECKING, Any
+
+if TYPE_CHECKING:
+    from collections.abc import Callable, Sequence
 
 # NOTE: DynamoDB supports up to 38 digits precision
 # Therefore, this ensures our Decimal follows what's stored in the table

aws_lambda_powertools/shared/functions.py

@@ -8,10 +8,13 @@
 import warnings
 from binascii import Error as BinAsciiError
 from pathlib import Path
-from typing import Any, Generator, overload
+from typing import TYPE_CHECKING, Any, overload
 
 from aws_lambda_powertools.shared import constants
 
+if TYPE_CHECKING:
+    from collections.abc import Generator
+
 logger = logging.getLogger(__name__)
 
 

aws_lambda_powertools/shared/types.py

@@ -1,3 +1,4 @@
-from typing import Any, Callable, TypeVar
+from collections.abc import Callable
+from typing import Any, TypeVar
 
-AnyCallableT = TypeVar("AnyCallableT", bound=Callable[..., Any])  # noqa: VNE001
+AnyCallableT = TypeVar("AnyCallableT", bound=Callable[..., Any])

aws_lambda_powertools/utilities/serialization.py

@@ -2,7 +2,8 @@
 
 import base64
 import json
-from typing import Any, Callable
+from collections.abc import Callable
+from typing import Any
 
 
 def base64_encode(data: str) -> str:

tests/unit/shared/test_dynamodb_deserializer.py

@@ -1,4 +1,6 @@
-from typing import Any, Dict, Optional
+from __future__ import annotations
+
+from typing import Any
 
 import pytest
 
@@ -10,14 +12,14 @@ def __init__(self, data: dict):
         self._data = data
         self._deserializer = TypeDeserializer()
 
-    def _deserialize_dynamodb_dict(self) -> Optional[Dict[str, Any]]:
+    def _deserialize_dynamodb_dict(self) -> dict[str, Any] | None:
         if self._data is None:
             return None
 
         return {k: self._deserializer.deserialize(v) for k, v in self._data.items()}
 
     @property
-    def data(self) -> Optional[Dict[str, Any]]:
+    def data(self) -> dict[str, Any] | None:
         """The primary key attribute(s) for the DynamoDB item that was modified."""
         return self._deserialize_dynamodb_dict()
 

tests/unit/test_cookie_class.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from datetime import datetime
 
 from aws_lambda_powertools.shared.cookies import Cookie, SameSite

tests/unit/test_data_classes.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import base64
 import datetime
 import json

tests/unit/test_json_encoder.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import decimal
 import json
 from dataclasses import dataclass

tests/unit/test_lru_cache.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import random
 import sys
 

tests/unit/test_shared_functions.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import os
 import warnings
 from dataclasses import dataclass