Added cooldowns for commands that utilise PIL

pull/9/head
sgoudham 4 years ago
parent ea52659491
commit 01892abb27

@ -27,7 +27,7 @@ from PIL.ImageOps import invert
from discord import Embed, Role, File from discord import Embed, Role, File
from discord import Member, TextChannel from discord import Member, TextChannel
from discord import __version__ as discord_version from discord import __version__ as discord_version
from discord.ext.commands import bot_has_permissions, guild_only, Cog, group from discord.ext.commands import bot_has_permissions, guild_only, Cog, group, cooldown, BucketType
from discord.ext.commands import command from discord.ext.commands import command
from psutil import Process, virtual_memory from psutil import Process, virtual_memory
@ -465,6 +465,7 @@ class Info(Cog):
@get_user_avatar.command(name="greyscale", aliases=["gs"]) @get_user_avatar.command(name="greyscale", aliases=["gs"])
@bot_has_permissions(embed_links=True) @bot_has_permissions(embed_links=True)
@cooldown(1, 2, BucketType.user)
async def greyscale_user_avatar(self, ctx, *, member: Optional[Member] = None): async def greyscale_user_avatar(self, ctx, *, member: Optional[Member] = None):
"""Get the greyscale avatar of the member""" """Get the greyscale avatar of the member"""
@ -491,6 +492,7 @@ class Info(Cog):
@get_user_avatar.command(name="invert", aliases=["negative"]) @get_user_avatar.command(name="invert", aliases=["negative"])
@bot_has_permissions(embed_links=True) @bot_has_permissions(embed_links=True)
@cooldown(1, 2, BucketType.user)
async def greyscale_user_avatar(self, ctx, *, member: Optional[Member] = None): async def greyscale_user_avatar(self, ctx, *, member: Optional[Member] = None):
"""Get the inverted avatar of the member""" """Get the inverted avatar of the member"""

Loading…
Cancel
Save