feat: improve patch type hint

HXSecurity · Sep 21, 2023 · 846538c · 846538c
1 parent 61399c1
commit 846538c
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/dongtai_conf/patch/__init__.py b/dongtai_conf/patch/__init__.py
@@ -92,6 +92,14 @@ def wrapper(*args: Any, **kwargs: Any):
 
 
 def patch(patch_func: Callable[..., Any], patch_id: int = 0):
+    @overload
+    def wrapper(func: Callable[[Unpack[tuple[T]]], T]):
+        ...
+
+    @overload
+    def wrapper(func: Callable[[Unpack[Ts]], tuple[Unpack[Ts]]]):
+        ...
+
     def wrapper(func: Callable[..., Any]):
         to_patch_func = getattr(patch_func, "to_patch_func", None)
         if to_patch_func is None: