Merge pull request #1841 from st1020/feat/improve-patch-type-hint

feat: improve patch type hint
HXSecurity · Sep 22, 2023 · 6fc7aa6 · 6fc7aa6
2 parents 61399c1 + 846538c
commit 6fc7aa6
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: