控制RGB灯的Discord Bot如何在收到新指令时中断彩虹灯效循环

你当前的同步死循环会阻塞discord.py的异步事件循环，导致bot完全无法响应新指令，按以下方案修改即可实现可中断的循环效果：

实现逻辑 #

• 用全局变量存储当前正在运行的灯效异步任务
• 所有灯效指令触发时，先取消上一个正在运行的灯效任务，再启动新的循环任务
• 循环逻辑需要异步化，每次循环让出事件循环控制权，保证bot可以正常接收新指令

示例代码 #

import asyncio

# 全局变量：存储当前运行的灯效任务
current_light_task = None

async def cancel_running_light_task():
    """取消当前正在运行的灯效任务"""
    global current_light_task
    if current_light_task and not current_light_task.done():
        current_light_task.cancel()
        try:
            await current_light_task
        except asyncio.CancelledError:
            # 任务取消异常直接忽略即可
            pass
        current_light_task = None

@client.command()
async def rainbow(ctx):
    global current_light_task
    # 先终止之前的所有灯效
    await cancel_running_light_task()
    await ctx.send("It is rainbow")

    # 异步循环逻辑
    async def run_rainbow_cycle():
        while True:
            # 同步的rainbow_cycle放到线程池运行，避免阻塞事件循环
            await asyncio.to_thread(rainbow_cycle, 0.001)
            # 让出事件循环控制权，用于响应新指令
            await asyncio.sleep(0)
    
    # 启动新任务并存储
    current_light_task = asyncio.create_task(run_rainbow_cycle())

注意事项 #

• 所有新增的灯效指令都按照相同逻辑编写，先调用cancel_running_light_task()再启动新的循环即可
• 如果你使用的Python版本低于3.9、没有asyncio.to_thread，可以替换为asyncio.get_event_loop().run_in_executor(None, rainbow_cycle, 0.001)
• 不要删除循环内的await asyncio.sleep(0)，否则事件循环会被占满，还是无法响应新指令

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