Discord.py(重写)如何使用on_message事件获得冷却时间?

时间:2020-06-24 14:50:30

标签: python python-3.x discord discord.py-rewrite

我一直在尝试将命令转换为on_message事件,因为在这种情况下,它节省了空间并且看起来更干净。但是我似乎不再使用@cooldown()了,因为我不得不使用命令。Cog.listener()

还有其他方法可以使冷却时间发挥作用吗?我的代码在下面列出

# Cog on_message for waifus and husbandos
    @commands.Cog.listener()
    # Cooldown NOT WORKING
    @cooldown(1, 1, BucketType.user)
    async def on_message(self, message):

        # Defining the channel and global variables
        global waifu_split_msg
        global husbando_split_msg
        channel = message.channel

        # Defining the message content in lowercase
        user_msg = message.content.lower()

        # Defining array for the list of waifus/husbando's available
        waifu_array = ["toga", "yumeko"]
        husbando_array = ["husk", "kakashi", "tamaki"]

        # If the channel that the command has been sent is in the list of accepted channels
        if str(channel) in settings.channels:

            # Surround with try/except to catch any exceptions that may occur
            try:

                # Makes sure that the user wants a random image of a waifu
                if 'w random' in user_msg:

                    # Get embed from randomWaifu() and send it to the channel
                    embed = randomWaifu(message, waifu_array)
                    await channel.send(embed=embed)

                # Makes sure that the user wants a specific image of a waifu
                elif user_msg.startswith('~w'):

                    # Define who the waifu is using string splitting
                    waifu_split_msg = user_msg.split("w ", 1)
                    w_array = str(waifu_split_msg[-1]).lower()

                    # Retrieve the image of the waifu that the user has specified
                    with open(f'images/AnimeImages/Waifus/{w_array}.txt') as file:
                        images_array = file.readlines()

                    # Get the full name of the waifu
                    full_name = Abbrev(w_array)

                    # Get the embed from a displayAnimeImage() and send it to the channel
                    embed = displayAnimeImage(images_array, message, full_name)
                    await channel.send(embed=embed)

            except FileNotFoundError as e:
                print(e)

                # Throw error message saying no waifu's could be found
                await channel.send(f"Sorry! That Waifu doesn't exist!! Try the Waifu's listed below!")

                # Send list of suitable waifu's to the channel
                nice = string.capwords(', '.join(map(str, waifu_array)))
                await channel.send(nice)

            # Surround with try/except to catch any exceptions that may occur
            try:

                # Makes sure that the user wants a random image of a husbando
                if 'h random' in user_msg:

                    # Get embed from randomHusbando() and send it to the channel
                    embed = randomHusbando(message, husbando_array)
                    await channel.send(embed=embed)

                # Makes sure that the user wants a specific image of a husbando
                elif user_msg.startswith('~h'):

                    # Define who the husbando is using string splitting
                    husbando_split_msg = user_msg.split("h ", 1)
                    h_array = str(husbando_split_msg[-1]).lower()

                    # Retrieve the image of the Husbando that the user has specified
                    with open(f'images/AnimeImages/Husbandos/{h_array}.txt') as file:
                        images_array = file.readlines()

                    # Get the full name of the husbando
                    full_name = Abbrev(h_array)

                    # Get the embed from a displayAnimeImage() and send it to the channel
                    embed = displayAnimeImage(images_array, message, full_name)
                    await channel.send(embed=embed)

            except FileNotFoundError as e:
                print(e)

                # Throw error message saying no husbando's could be found
                await channel.send(f"Sorry! That Husbando doesn't exist!! Try the Husbando's listed below!")

                # Send list of suitable Husbando's to the channel
                nice = string.capwords(', '.join(map(str, husbando_array)))
                await channel.send(nice)

        # if the message is outwith the enso-chan-commands
        else:
            # Makes sure that the user only typed ~w or ~h
            if user_msg.endswith('~w') or user_msg.endswith('~h'):
                # Send error message
                message = await channel.send(error_function())

                # Let the user read the message for 2.5 seconds
                await asyncio.sleep(2.5)
                # Delete the message
                await message.delete()

除了顶部的cooldown()对事件没有任何影响之外,该代码还可以完美地工作

有人可以帮助我提出另一种解决方案吗?

1 个答案:

答案 0 :(得分:0)

您可以使用时间参数或计数参数来限制事件的使用次数。您将无法轻松地为每个用户做到这一点。如果您希望每个用户冷静一下,我强烈建议您切换回命令方法。 这可能会有所帮助。 How can I limit the on_message replies (Discord Python bot)