将冷却时间/计时器添加到 on_message [Discord.py]

2022-01-15 00:00:00 python python-3.x discord discord.py

问题描述

我最近开始用 Python 制作一个 Discord 机器人(用它测试 Python 的基础),并自己创建了一个带有多个命令的功能性机器人.为了扩大它的用途,我添加了一个级别/XP 系统,该系统目前正在运行.

I got into making a Discord bot in Python very recently (testing the grounds of Python with it) and created a functioning one with several commands myself. To widen its uses, I have added a level/XP system, which is working so far.

[...]
@bot.event
async def on_message(message):
        user_add_xp(message.author.id, 2)
        await bot.process_commands(message)

# commands go here

def user_add_xp(user_id, xp):
    if os.path.isfile('users.json'):
            try:
                    with open('users.json', 'r') as fp:
                            users = json.load(fp)
                    users[user_id]['xp'] += xp
                    with open('users.json', 'w') as fp:
                            json.dump(users, fp, sort_keys=True, indent=4)
            except KeyError:
                    with open('users.json', 'r') as fp:
                            users = json.load(fp)
                    users[user_id] = {}
                    users[user_id]['xp'] = xp
                    with open('users.json', 'w') as fp:
                            json.dump(users, fp, sort_keys=True, indent=4)
    else:
        users = {user_id: {}}
        users[user_id]['xp'] = xp
        with open('users.json', 'w') as fp:
                json.dump(users, fp, sort_keys=True, indent=4)
[...]

但为了防止用户只是泛滥/垃圾邮件某些频道并飙升至顶部,我想为 XP 奖励添加一个冷却时间/计时器.我尝试将 @commands.cooldown(1, 120, commands.BucketType.server) 添加到 @bot.eventuser_add_xp,但两者都没有得到我想要的结果.我不知道如何添加此冷却时间/计时器.

But to prevent users from just flooding/spamming some channels and rocketing to the top, I want to add a cooldown/timer on the awarding of XP. I have tried to add @commands.cooldown(1, 120, commands.BucketType.server) to both @bot.event and user_add_xp, but both do not get me the desired result. I have no other idea how to add this cooldown/timer.

最后,我希望机器人每两分钟只授予一次 XP.

In the end, I want the bot to only grant XP once every two minutes.


解决方案

不确定是否可以仅使用 discord.py,但您可以存储最后一次将消息授予用户 XP 的时间在你的字典里.

Not sure if it's possible with just discord.py, but you can store the last time a message was awarded XP to a user in your dictionary.

以下代码存储自静态开始日期(epoch)消息奖励 XP 以来的秒数.然后,它会检查发生新消息事件的时间.

The below code stores the number of seconds since a static start date (epoch) when a message awards XP. It then checks against this time when a new message event happens.

[...]
import datetime

epoch = datetime.datetime.utcfromtimestamp(0)

@bot.event
async def on_message(message):
    user_add_xp(message.author.id, 2)
    await bot.process_commands(message)

# commands go here

def user_add_xp(user_id, xp):
    if os.path.isfile('users.json'):
        try:
            with open('users.json', 'r') as fp:
                users = json.load(fp)

            time_diff = (datetime.datetime.utcnow() - epoch).total_seconds() - users[user_id]['xp_time']
            if time_diff >= 120:
                users[user_id]['xp'] += xp
                users[user_id]['xp_time'] = (datetime.datetime.utcnow() - epoch).total_seconds()
                with open('users.json', 'w') as fp:
                    json.dump(users, fp, sort_keys=True, indent=4)
        except KeyError:
            with open('users.json', 'r') as fp:
                users = json.load(fp)
            users[user_id] = {}
            users[user_id]['xp'] = xp
            users[user_id]['xp_time'] = (datetime.datetime.utcnow() - epoch).total_seconds()
            with open('users.json', 'w') as fp:
                json.dump(users, fp, sort_keys=True, indent=4)
    else:
        users = {user_id: {}}
        users[user_id]['xp'] = xp
        users[user_id]['xp_time'] = (datetime.datetime.utcnow() - epoch).total_seconds()
        with open('users.json', 'w') as fp:
            json.dump(users, fp, sort_keys=True, indent=4)
[...]

相关文章