chore: Use ParamSpec from PEP 612 (#2891)

Author: aahung

samtranslator/internal/deprecation_control.py

@@ -13,22 +13,25 @@
 from functools import wraps
 from typing import Callable, Optional, TypeVar
 
+from typing_extensions import ParamSpec
+
+PT = ParamSpec("PT")  # parameters
 RT = TypeVar("RT")  # return type
 
 
 def _make_message(message: str, replacement: Optional[str]) -> str:
     return f"{message}, please use {replacement}" if replacement else message
 
 
-def deprecated(replacement: Optional[str]) -> Callable[[Callable[..., RT]], Callable[..., RT]]:
+def deprecated(replacement: Optional[str]) -> Callable[[Callable[PT, RT]], Callable[PT, RT]]:
     """
     Mark a function/method as deprecated.
 
     The warning is shown by default when triggered directly
     by code in __main__.
     """
 
-    def decorator(func: Callable[..., RT]) -> Callable[..., RT]:
+    def decorator(func: Callable[PT, RT]) -> Callable[PT, RT]:
         @wraps(func)
         def wrapper(*args, **kwargs) -> RT:  # type: ignore
             warning_message = _make_message(

samtranslator/metrics/method_decorator.py

@@ -6,11 +6,14 @@
 from datetime import datetime
 from typing import Callable, Optional, TypeVar, Union, overload
 
+from typing_extensions import ParamSpec
+
 from samtranslator.metrics.metrics import DummyMetricsPublisher, Metrics
 from samtranslator.model import Resource
 
 LOG = logging.getLogger(__name__)
 
+_PT = ParamSpec("_PT")  # parameters
 _RT = TypeVar("_RT")  # return value
 
 
@@ -81,18 +84,18 @@ def _send_cw_metric(prefix, name, execution_time_ms, func, args):  # type: ignor
 @overload
 def cw_timer(
     *, name: Optional[str] = None, prefix: Optional[str] = None
-) -> Callable[[Callable[..., _RT]], Callable[..., _RT]]:
+) -> Callable[[Callable[_PT, _RT]], Callable[_PT, _RT]]:
     ...
 
 
 @overload
-def cw_timer(_func: Callable[..., _RT], name: Optional[str] = None, prefix: Optional[str] = None) -> Callable[..., _RT]:
+def cw_timer(_func: Callable[_PT, _RT], name: Optional[str] = None, prefix: Optional[str] = None) -> Callable[_PT, _RT]:
     ...
 
 
 def cw_timer(
-    _func: Optional[Callable[..., _RT]] = None, name: Optional[str] = None, prefix: Optional[str] = None
-) -> Union[Callable[..., _RT], Callable[[Callable[..., _RT]], Callable[..., _RT]]]:
+    _func: Optional[Callable[_PT, _RT]] = None, name: Optional[str] = None, prefix: Optional[str] = None
+) -> Union[Callable[_PT, _RT], Callable[[Callable[_PT, _RT]], Callable[_PT, _RT]]]:
     """
     A method decorator, that will calculate execution time of the decorated method, and store this information as a
     metric in CloudWatch by calling the metrics singleton instance.
@@ -105,7 +108,7 @@ def cw_timer(
     If prefix is defined, it will be added in the beginning of what is been generated above
     """
 
-    def cw_timer_decorator(func: Callable[..., _RT]) -> Callable[..., _RT]:
+    def cw_timer_decorator(func: Callable[_PT, _RT]) -> Callable[_PT, _RT]:
         @functools.wraps(func)
         def wrapper_cw_timer(*args, **kwargs) -> _RT:  # type: ignore[no-untyped-def]
             start_time = datetime.now()