diff --git a/cogs/hangman.py b/cogs/hangman.py index ac74d5f..764a2d2 100644 --- a/cogs/hangman.py +++ b/cogs/hangman.py @@ -114,6 +114,9 @@ class Hangman: """Makes a guess towards the server's currently running hangman game""" game = self.games.get(ctx.message.server.id) if not game: + bucket = ctx.command._buckets.get_bucket(ctx) + bucket.is_rate_limited() + bucket._token = bucket.rate await self.bot.say("There are currently no hangman games running!") return @@ -123,6 +126,9 @@ class Hangman: # And also add a message for a loss/win if len(guess) == 1: if guess in game.guessed_letters: + bucket = ctx.command._buckets.get_bucket(ctx) + bucket.is_rate_limited() + bucket._tokens = bucket.rate await self.bot.say("That letter has already been guessed!") # Return here as we don't want to count this as a failure return diff --git a/cogs/interaction.py b/cogs/interaction.py index 5cf7cb7..f805319 100644 --- a/cogs/interaction.py +++ b/cogs/interaction.py @@ -93,12 +93,21 @@ class Interaction: async def battle(self, ctx, player2: discord.Member): """Challenges the mentioned user to a battle""" if ctx.message.author.id == player2.id: + bucket = ctx.command._buckets.get_bucket(ctx) + bucket.is_rate_limited() + bucket._tokens = bucket.rate await self.bot.say("Why would you want to battle yourself? Suicide is not the answer") return if self.bot.user.id == player2.id: + bucket = ctx.command._buckets.get_bucket(ctx) + bucket.is_rate_limited() + bucket._tokens = bucket.rate await self.bot.say("I always win, don't even try it.") return if self.user_battling(ctx, player2): + bucket = ctx.command._buckets.get_bucket(ctx) + bucket.is_rate_limited() + bucket._tokens = bucket.rate await self.bot.say("You or the person you are trying to battle is already in a battle!") return @@ -167,9 +176,15 @@ class Interaction: """Boops the mentioned person""" booper = ctx.message.author if boopee.id == booper.id: + bucket = ctx.command._buckets.get_bucket(ctx) + bucket.is_rate_limited() + bucket._tokens = bucket.rate await self.bot.say("You can't boop yourself! Silly...") return if boopee.id == self.bot.user.id: + bucket = ctx.command._buckets.get_bucket(ctx) + bucket.is_rate_limited() + bucket._tokens = bucket.rate await self.bot.say("Why the heck are you booping me? Get away from me >:c") return