From 846538c952b3d950758b33bfc8606747cb80bbf5 Mon Sep 17 00:00:00 2001
From: st1020 <gaoyishi@huoxian.cn>
Date: Thu, 21 Sep 2023 11:01:49 +0800
Subject: [PATCH] feat: improve patch type hint

---
 dongtai_conf/patch/__init__.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/dongtai_conf/patch/__init__.py b/dongtai_conf/patch/__init__.py
index 6cd08408..ea215ea4 100644
--- 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: