Python开发Discord机器人如何按用户维度限制指令运行次数

本方案完全适配你现有基于discord.py的角色权限校验逻辑，无需重构原有代码，通过叠加自定义校验层即可实现单用户差异化的指令调用次数限制。

核心逻辑说明 #

• 全局维护两类核心映射关系：一是不同用户对应的指令调用上限、计数周期配置；二是实时记录每个用户在当前计数周期内的已调用次数。
• 通过自定义discord.py校验装饰器，和原有@commands.has_any_role角色校验装饰器叠加使用，保证权限校验、频次管控两层逻辑独立运行，互不干扰。
• 计数逻辑加入异步锁，避免高并发下计数偏差导致的管控失效。

具体代码改造步骤 #

1. 初始化全局存储结构 #

在机器人实例初始化的全局代码区，新增频次管控需要的配置和计数器：

import discord
from discord.ext import commands
from collections import defaultdict
import time
import asyncio

# 差异化频次配置：key为用户Discord ID，value为(周期内允许调用次数, 计数周期长度/秒)
# 可根据需求给不同用户设置不同上限
USER_QUOTA_CONFIG = {
    123456789012345678: (5, 600),   # 对应用户每10分钟最多可调用5次
    987654321098765432: (2, 3600),  # 对应用户每小时最多可调用2次
}
# 无单独配置的用户使用默认配额
DEFAULT_QUOTA = (1, 3600)
# 用户调用计数器：key为用户ID，value为(已调用次数, 当前周期起始时间戳)
user_call_counter = defaultdict(lambda: (0, time.time()))
# 异步锁，规避并发计数异常
quota_lock = asyncio.Lock()

2. 编写频次校验装饰器 #

实现可复用的自定义校验逻辑，自动完成周期重置、超限拦截、计数累加流程：

def check_user_quota():
    async def predicate(ctx):
        # 跳过机器人自身的调用校验
        if ctx.author.bot:
            return True
        user_id = ctx.author.id
        # 读取当前用户的配额配置，无配置则用默认值
        quota_limit, quota_period = USER_QUOTA_CONFIG.get(user_id, DEFAULT_QUOTA)
        async with quota_lock:
            used_count, period_start = user_call_counter[user_id]
            # 超出当前计数周期则自动重置计数
            if time.time() - period_start > quota_period:
                used_count = 0
                period_start = time.time()
            # 超出配额时拦截指令，提示剩余冷却时间
            if used_count >= quota_limit:
                remain_cd = int(quota_period - (time.time() - period_start))
                await ctx.send(f"指令调用已达上限，请{remain_cd}秒后重试", delete_after=5)
                user_call_counter[user_id] = (used_count, period_start)
                return False
            # 未超限则累加调用次数，放行指令
            user_call_counter[user_id] = (used_count + 1, period_start)
            return True
    return commands.check(predicate)

3. 叠加装饰器到原有指令 #

直接在你现有的maintenance指令上叠加频次校验装饰器，原有业务逻辑完全不需要改动：

@bot.command(name='command', help='')
# 保留原有角色权限校验
@commands.has_any_role("bot", "non-bot")
# 新增用户频次校验
@check_user_quota()
async def maintenance(ctx):
    try:
        await ctx.message.delete()
        embed = discord.Embed(title='Title here',
                              description="Message here", color=0x6aa780)
        await ctx.send(embed=embed)
    except Exception as e:
        print("maintenance command failed!")
        print(e)

可扩展优化方向 #

• 接入本地JSON或SQLite做持久化存储，避免机器人重启后配额配置、计数数据丢失。
• 把计数器的key从单纯用户ID调整为(用户ID, 指令名)的元组，即可实现单用户针对不同指令的独立配额限制。
• 新增管理员专属配置指令，支持不修改代码、不重启机器人的前提下动态调整任意用户的配额数值。
• 把超限提示改为用户私信发送，避免在公共频道发送提示信息影响其他成员。

内容的提问来源于stack exchange，提问作者Snowman King