feat: allow getting a cooldown with just an id

interactions/models/internal/cooldowns.py

@@ -1,5 +1,6 @@
 import asyncio
 import time
+import typing
 from enum import IntEnum
 from typing import TYPE_CHECKING, Any, Dict
 
@@ -64,6 +65,12 @@ def __init__(self, cooldown_bucket: Buckets, rate: int, interval: float) -> None
         self.interval: float = interval
 
     async def get_cooldown(self, context: "BaseContext") -> "CooldownSystem":
+        """
+        Get the cooldown system for the command.
+
+        Args:
+            context: The context of the command
+        """
         key = await self.bucket(context)
 
         if key not in self.cooldown_repositories:
@@ -72,6 +79,23 @@ async def get_cooldown(self, context: "BaseContext") -> "CooldownSystem":
             return cooldown
         return self.cooldown_repositories.get(await self.bucket(context))
 
+    def get_cooldown_with_key(self, key: Any, *, create: bool = False) -> typing.Optional["CooldownSystem"]:
+        """
+        Get the cooldown system for the command.
+
+        Note:
+            The preferred way to get the cooldown system is to use `get_cooldown` as it will use the context to get the correct key.
+
+        Args:
+            key: The key to get the cooldown system for
+            create: Whether to create a new cooldown system if one does not exist
+        """
+        if key not in self.cooldown_repositories and create:
+            cooldown = CooldownSystem(self.rate, self.interval)
+            self.cooldown_repositories[key] = cooldown
+            return cooldown
+        return self.cooldown_repositories.get(key)
+
     async def acquire_token(self, context: "BaseContext") -> bool:
         """
         Attempt to acquire a token for a command to run. Uses the context of the command to use the correct CooldownSystem.