feat: cooldown upgrades (#1323)

Author: LordOfPolls

interactions/__init__.py

@@ -38,6 +38,7 @@
     T_co,
     utils,
 )
+from .client import const
 from .models import (
     ActionRow,
     ActiveVoiceState,
@@ -328,10 +329,11 @@
     WebhookMixin,
     WebhookTypes,
     WebSocketOPCode,
+    SlidingWindowSystem,
+    ExponentialBackoffSystem,
+    LeakyBucketSystem,
+    TokenBucketSystem,
 )
-from .api import events
-from . import ext
-from .client import const
 
 __all__ = (
     "__api_version__",
@@ -413,6 +415,10 @@
     "cooldown",
     "Cooldown",
     "CooldownSystem",
+    "SlidingWindowSystem",
+    "ExponentialBackoffSystem",
+    "LeakyBucketSystem",
+    "TokenBucketSystem",
     "CustomEmoji",
     "CustomEmojiConverter",
     "DateTrigger",

interactions/models/__init__.py

@@ -212,6 +212,10 @@
     cooldown,
     Cooldown,
     CooldownSystem,
+    SlidingWindowSystem,
+    ExponentialBackoffSystem,
+    LeakyBucketSystem,
+    TokenBucketSystem,
     CustomEmojiConverter,
     DateTrigger,
     dm_only,
@@ -361,6 +365,10 @@
     "cooldown",
     "Cooldown",
     "CooldownSystem",
+    "SlidingWindowSystem",
+    "ExponentialBackoffSystem",
+    "LeakyBucketSystem",
+    "TokenBucketSystem",
     "CustomEmoji",
     "CustomEmojiConverter",
     "DateTrigger",

interactions/models/internal/__init__.py

@@ -86,7 +86,16 @@
     UserConverter,
     VoiceChannelConverter,
 )
-from .cooldowns import Buckets, Cooldown, CooldownSystem, MaxConcurrency
+from .cooldowns import (
+    Buckets,
+    Cooldown,
+    CooldownSystem,
+    MaxConcurrency,
+    SlidingWindowSystem,
+    ExponentialBackoffSystem,
+    LeakyBucketSystem,
+    TokenBucketSystem,
+)
 from .listener import listen, Listener
 from .protocols import Converter
 from .extension import Extension
@@ -121,6 +130,10 @@
     "cooldown",
     "Cooldown",
     "CooldownSystem",
+    "SlidingWindowSystem",
+    "ExponentialBackoffSystem",
+    "LeakyBucketSystem",
+    "TokenBucketSystem",
     "CustomEmojiConverter",
     "DateTrigger",
     "dm_only",

interactions/models/internal/command.py

@@ -23,7 +23,7 @@
 from interactions.client.utils.misc_utils import get_parameters, get_object_name, maybe_coroutine
 from interactions.client.utils.serializer import no_export_meta
 from interactions.models.internal.callback import CallbackObject
-from interactions.models.internal.cooldowns import Cooldown, Buckets, MaxConcurrency
+from interactions.models.internal.cooldowns import Cooldown, Buckets, MaxConcurrency, CooldownSystem
 from interactions.models.internal.protocols import Converter
 
 if TYPE_CHECKING:
@@ -283,19 +283,21 @@ def wrapper(coro: CommandT) -> CommandT:
     return wrapper
 
 
-def cooldown(bucket: Buckets, rate: int, interval: float) -> Callable[[CommandT], CommandT]:
+def cooldown(
+    bucket: Buckets, rate: int, interval: float, cooldown_system: typing.Type[CooldownSystem] | None = None
+) -> Callable[[CommandT], CommandT]:
     """
     Add a cooldown to a command.
 
     Args:
         bucket: The bucket used to track cooldowns
         rate: How many commands may be ran per interval
         interval: How many seconds to wait for a cooldown
-
+        cooldown_system: The cooldown system to use
     """
 
     def wrapper(coro: CommandT) -> CommandT:
-        cooldown_obj = Cooldown(bucket, rate, interval)
+        cooldown_obj = Cooldown(bucket, rate, interval, cooldown_system=cooldown_system)
 
         coro.cooldown = cooldown_obj