chore(typing): move backwards compat types to shared types (#3092)

Author: heitorlessa

aws_lambda_powertools/event_handler/middlewares/base.py

@@ -1,10 +1,9 @@
 from abc import ABC, abstractmethod
 from typing import Generic
 
-from typing_extensions import Protocol
-
 from aws_lambda_powertools.event_handler.api_gateway import Response
 from aws_lambda_powertools.event_handler.types import EventHandlerInstance
+from aws_lambda_powertools.shared.types import Protocol
 
 
 class NextMiddleware(Protocol):

aws_lambda_powertools/logging/types.py

@@ -1,24 +1,9 @@
 from __future__ import annotations
 
-import sys
-
-if sys.version_info >= (3, 11):
-    from typing import NotRequired
-else:
-    from typing_extensions import NotRequired
-
-if sys.version_info >= (3, 8):
-    from typing import TypedDict
-else:
-    from typing_extensions import TypedDict
-
-if sys.version_info >= (3, 10):
-    from typing import TypeAlias
-else:
-    from typing_extensions import TypeAlias
-
 from typing import Any, Dict, List, Union
 
+from aws_lambda_powertools.shared.types import NotRequired, TypeAlias, TypedDict
+
 LogRecord: TypeAlias = Union[Dict[str, Any], "PowertoolsLogRecord"]
 
 

aws_lambda_powertools/metrics/functions.py

@@ -1,12 +1,11 @@
 from __future__ import annotations
 
-from typing import List
-
 from aws_lambda_powertools.metrics.provider.cloudwatch_emf.exceptions import (
     MetricResolutionError,
     MetricUnitError,
 )
 from aws_lambda_powertools.metrics.provider.cloudwatch_emf.metric_properties import MetricResolution, MetricUnit
+from aws_lambda_powertools.shared.types import List
 
 
 def extract_cloudwatch_metric_resolution_value(metric_resolutions: List, resolution: int | MetricResolution) -> int:

aws_lambda_powertools/metrics/provider/cloudwatch_emf/types.py

@@ -1,6 +1,4 @@
-from typing import List
-
-from typing_extensions import NotRequired, TypedDict
+from aws_lambda_powertools.shared.types import List, NotRequired, TypedDict
 
 
 class CloudWatchEMFMetric(TypedDict):

aws_lambda_powertools/metrics/types.py

@@ -1,4 +1,4 @@
-from typing_extensions import NotRequired, TypedDict
+from aws_lambda_powertools.shared.types import NotRequired, TypedDict
 
 
 class MetricNameUnitResolution(TypedDict):

aws_lambda_powertools/shared/cookies.py

@@ -1,7 +1,9 @@
 from datetime import datetime
 from enum import Enum
 from io import StringIO
-from typing import List, Optional
+from typing import Optional
+
+from aws_lambda_powertools.shared.types import List
 
 
 class SameSite(Enum):

aws_lambda_powertools/shared/types.py

@@ -1,14 +1,25 @@
 import sys
 from typing import Any, Callable, Dict, List, TypeVar, Union
 
-AnyCallableT = TypeVar("AnyCallableT", bound=Callable[..., Any])  # noqa: VNE001
-# JSON primitives only, mypy doesn't support recursive tho
-JSONType = Union[str, int, float, bool, None, Dict[str, Any], List[Any]]
+if sys.version_info >= (3, 8):
+    from typing import Literal, Protocol, TypedDict
+else:
+    from typing_extensions import Literal, Protocol, TypedDict
 
 
-if sys.version_info >= (3, 8):
-    from typing import Protocol
+if sys.version_info >= (3, 11):
+    from typing import NotRequired
 else:
-    from typing_extensions import Protocol
+    from typing_extensions import NotRequired
+
+
+if sys.version_info >= (3, 10):
+    from typing import TypeAlias
+else:
+    from typing_extensions import TypeAlias
+
+AnyCallableT = TypeVar("AnyCallableT", bound=Callable[..., Any])  # noqa: VNE001
+# JSON primitives only, mypy doesn't support recursive tho
+JSONType = Union[str, int, float, bool, None, Dict[str, Any], List[Any]]
 
-__all__ = ["Protocol"]
+__all__ = ["Protocol", "TypedDict", "Literal", "NotRequired", "TypeAlias"]

aws_lambda_powertools/utilities/batch/types.py

@@ -1,10 +1,7 @@
-#
-# type specifics
-#
 import sys
-from typing import List, Optional, Type, Union
+from typing import Optional, Type, Union
 
-from typing_extensions import TypedDict
+from aws_lambda_powertools.shared.types import List, TypedDict
 
 has_pydantic = "pydantic" in sys.modules
 

aws_lambda_powertools/utilities/data_classes/secrets_manager_event.py

@@ -1,5 +1,4 @@
-from typing_extensions import Literal
-
+from aws_lambda_powertools.shared.types import Literal
 from aws_lambda_powertools.utilities.data_classes.common import DictWrapper
 
 

aws_lambda_powertools/utilities/parameters/ssm.py

@@ -8,14 +8,14 @@
 
 import boto3
 from botocore.config import Config
-from typing_extensions import Literal
 
 from aws_lambda_powertools.shared import constants
 from aws_lambda_powertools.shared.functions import (
     resolve_max_age,
     resolve_truthy_env_var_choice,
     slice_dictionary,
 )
+from aws_lambda_powertools.shared.types import Literal
 
 from .base import DEFAULT_MAX_AGE_SECS, DEFAULT_PROVIDERS, BaseProvider, transform_value
 from .exceptions import GetParameterError