我正在制作一个 Discord 机器人,它可以执行各种操作,包括对提到该机器人的人做出反应,以及对某些用户在某个频道中所说的话做出反应。
我的代码(为了清楚起见缩短)是:
BOT = os.getenv('DISCORD_BOT_MENTION')
CHANNEL = os.getenv('DISCORD_CHANNEL')
USER1 = os.getenv('DISCORD_USER1_ID')
USER2 = os.getenv('DISCORD_USER2_ID')
@bot.event
async def on_message(message):
content = message.content # Did bot get mentioned?
channel = message.channel # Where was the bot mentioned
channel_ID = message.channel.id #
user = message.author.name # Who mentioned the bot
user_ID = message.author.id
if user == client.user:
return
elif BOT in content:
brankos_responses = [
"Hello",
"Hi",
]
weights_responses = [
15,
10,
]
response = random.choices(brankos_responses, weights_responses, k = 1)
await asyncio.sleep(0.7) # Wait 0.7s before answering
await channel.send(response[0])
elif channel_ID == int(CHANNEL):
if user_ID == int(USER_ID_1):
brankos_responses = [
"Test Message 1 for User 1 in Channel",
"Test Message 2 for User 1 in Channel",
]
response = random.choice(brankos_responses)
await channel.send(response)
if user_ID == int(USER_id_2):
brankos_responses = [
"Test Message 1 for User 2 In Channel",
"Test Message 2 for User 2 in Channel",
]
response = random.choice(brankos_responses)
await channel.send(response)
else:
return
await bot.process_commands(message)
然而我发现,如果你向机器人发送垃圾邮件,或者用户 1 和 2 在频道中交谈,它会继续给出回复,我想限制这一点。
在discord.py命令中很容易使用@commands.cooldown(1, 3600, commands.BucketType.user)
例如,但是我拥有的不是命令,因此不能在这里使用。
我认为我可以让机器人休眠(使用await asyncio),当它注意到频道中来自其中一个用户的消息时,然后当该人在5秒内没有说什么时,它将运行代码并发送1 个回复,但这只会导致所有回复堆积起来,并在几秒钟后全部发送。
所以我的问题是:如何限制机器人给出的回复数量?
现在我正在尝试这个:
user_list = [
int(USER1_ID),
int(USER2_ID),
int(USER3_ID),
int(USER4_ID),
]
messages = 0 # Define the count
if user_ID in user_list:
messages += 1 # Add one to count
print(messages)
if messages > 5: # If messages is more than 5, execute code
response = random.choice(brankos_responses)
await channel.send(response)
messages = 0 # Reset count
但因为它再次运行,它再次重置为 messages = 1,如果我将 messages = 0 放在函数之外,它就不起作用。