🔍 Add ParamSpec and TypeVar for type hints in Client class in squarecloud/client.py

Author: robert-nogueira

squarecloud/client.py

@@ -3,7 +3,7 @@
 
 from functools import wraps
 from io import BytesIO
-from typing import Any, Callable, Literal, TextIO
+from typing import Any, Callable, Literal, ParamSpec, TextIO, TypeVar
 
 from typing_extensions import deprecated
 
@@ -27,6 +27,9 @@
 from .listeners.request_listener import RequestListenerManager
 from .logging import logger
 
+P = ParamSpec('P')
+R = TypeVar('R')
+
 
 @deprecated(
     'create_config_file is deprecated, '
@@ -180,10 +183,12 @@ def _notify_listener(endpoint: Endpoint):
         :return: a callable
         """
 
-        def wrapper(func: Callable):
+        def wrapper(func: Callable[P, R]) -> Callable[P, R]:
             @wraps(func)
-            async def decorator(self: Client, *args, **kwargs) -> Response:
-                result: Any
+            async def decorator(
+                self: Client, *args: P.args, **kwargs: P.kwargs
+            ) -> R:
+                # result: Any
                 response: Response
                 result = await func(self, *args, **kwargs)
                 response = self._http.last_response