Improve typing for decorator

disrupted · Dec 14, 2024 · 1dcd10c · 1dcd10c
1 parent 66b6761
commit 1dcd10c
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/things_cloud/models/todo.py b/things_cloud/models/todo.py
@@ -4,7 +4,7 @@
 from collections.abc import Callable
 from datetime import datetime, time, timezone
 from enum import Enum
-from typing import Any, ParamSpec, TypeVar
+from typing import Any, Concatenate, ParamSpec, TypeVar
 
 import cattrs
 from attrs import define, field
@@ -48,8 +48,14 @@ class Note:
 _R = TypeVar("_R")
 
 
-def mod(*field_names: str):
-    def decorate(func: Callable[..., _R]):
+def mod(
+    *field_names: str,
+) -> Callable[
+    [Callable[Concatenate[TodoItem, P], _R]], Callable[Concatenate[TodoItem, P], _R]
+]:
+    def decorate(
+        func: Callable[Concatenate[TodoItem, P], _R],
+    ) -> Callable[Concatenate[TodoItem, P], _R]:
         def wrapper(self: TodoItem, *args: P.args, **kwargs: P.kwargs) -> _R:
             # first we call the wrapped function,
             # in case it throws an exception we don't want to modify