improve decorator typing

Author: dbatten5

poetry.lock

@@ -20,6 +20,7 @@ python = "^3.8"
 click = ">=8.0.1"
 numpy = "^1.24.2"
 pyperclip = "^1.8.2"
+typing-extensions = "^4.5.0"
 
 [tool.poetry.group.test.dependencies]
 pytest = "^6.2.4"
@@ -79,6 +80,12 @@ show_column_numbers = true
 show_error_codes = true
 show_error_context = true
 
+[[tool.mypy.overrides]]
+module = [
+  "pyperclip.*",
+]
+ignore_missing_imports = true
+
 [tool.ruff]
 ignore = [
   'B019',

pyproject.toml

@@ -1,29 +1,29 @@
 """Utils module."""
 
 from functools import wraps
-from typing import Any
 from typing import Callable
 from typing import TypeVar
-from typing import cast
 
 import pyperclip
+from typing_extensions import ParamSpec
 
 
-F = TypeVar("F", bound=Callable[..., str])
+P = ParamSpec("P")
+T = TypeVar("T")
 
 
-def use_clipboard(func: F) -> F:
+def use_clipboard(func: Callable[P, T]) -> Callable[P, str]:
     """Augument decorated functions argument to copy the output to the clipboard."""
 
     @wraps(func)
-    def wrapper_func(*args: Any, **kwargs: Any) -> str:
+    def wrapper_func(*args: P.args, **kwargs: P.kwargs) -> str:
         """Wrapped function."""
         out = func(*args, **kwargs)
 
         if kwargs.get("to_clp"):
             pyperclip.copy(out)
             print("ArrayTeX: copied to clipboard")
 
-        return out
+        return str(out)
 
-    return cast(F, wrapper_func)
+    return wrapper_func