feat: Add command cooldown

B1ue-Dev · Jul 7, 2023 · f1d0444 · f1d0444
1 parent 27a75cf
commit f1d0444
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/exts/fun/trivia.py b/exts/fun/trivia.py
@@ -137,6 +137,7 @@ async def get_question(self, category: str, difficulty: str) -> list:
             ),
         ],
     )
+    @interactions.cooldown(interactions.Buckets.USER, 1, 10)
     async def trivia(
         self,
         ctx: HybridContext,

diff --git a/exts/fun/urban.py b/exts/fun/urban.py
@@ -32,6 +32,7 @@ def __init__(self, client: interactions.Client) -> None:
             )
         ],
     )
+    @interactions.cooldown(interactions.Buckets.USER, 1, 7)
     async def urban(self, ctx: HybridContext, *, term: str) -> None:
         """Define a word on Urban Dictionary."""
 

diff --git a/exts/fun/whos_that_pokemon.py b/exts/fun/whos_that_pokemon.py
@@ -132,6 +132,7 @@ def __init__(self, client: interactions.Client) -> None:
             ),
         ],
     )
+    @interactions.cooldown(interactions.Buckets.USER, 3, 10)
     async def whos_that_pokemon(
         self,
         ctx: HybridContext,
-Original file line number
+Diff line change
@@ Expand Up @@
                 ),
             ],
         )
+        @interactions.cooldown(interactions.Buckets.USER, 1, 10)
         async def trivia(
             self,
             ctx: HybridContext,
@@ Expand Down @@