Add a py 3.10 overload to command decorator

Garrett098 · Jul 4, 2022 · 4cf035b · 4cf035b
1 parent 6800b57
commit 4cf035b
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/discord/ext/commands/core.py b/discord/ext/commands/core.py
@@ -1598,6 +1598,23 @@ async def reinvoke(self, ctx: Context, *, call_hooks: bool = False) -> None:
 # Decorators
 
 
+@overload  # for py 3.10
+def command(
+    name: str = ...,
+    cls: Type[Command[CogT, P, T]] = ...,
+    **attrs: Any,
+) -> Callable[
+    [
+        Union[
+            Callable[Concatenate[CogT, ContextT, P]], Coro[T],
+            Callable[Concatenate[ContextT, P]], Coro[T],
+        ]
+    ],
+    Command[CogT, P, T],
+]:
+    ...
+
+
 @overload
 def command(
     name: str = ...,