python – 为多个机器人加载cog

使用discord.py,我可以从一段代码运行多个机器人,但我正在寻找一种方法将cog或扩展加载到多个机器人中.对于一个测试用例,我有bot.py,它处理加载cog并启动bot,而cog.py是一个简单的cog,逐渐增加1到一个计数器

bot.py

from discord.ext import commands
import asyncio

client1 = commands.Bot(command_prefix='!')
client2 = commands.Bot(command_prefix='~')

client1.load_extension('cog')
client2.load_extension('cog')

@client1.event
async def on_ready():
    print('client1 ready')

@client1.command()
async def ping():
    await client1.say('Pong')

@client2.event
async def on_ready():
    print('client2 ready')

@client2.command()
async def ping():
    await client2.say('Pong')

loop = asyncio.get_event_loop()
loop.create_task(client1.start('TOKEN1'))
loop.create_task(client2.start('TOKEN2'))
loop.run_forever()

cog.py

from discord.ext import commands

class TestCog:

    def __init__(self, bot):
        self.bot = bot
        self.counter = 0

    @commands.command()
    async def add(self):
        self.counter += 1
        await self.bot.say('Counter is now %d' % self.counter)


def setup(bot):
    bot.add_cog(TestCog(bot))

使用!ping将使client1响应Pong,而使用~ping将使client2响应Pong,这是预期的行为.

但是,只有一个机器人会同时响应!add和~add,计数器会随着任一命令而增加.这似乎取决于哪个机器人最后加载cog.

有没有办法让正确的机器人响应正确的命令,同时使用任一命令增加计数器?我知道我可以把它分成两个齿轮,并将结果保存到文档中,但是可以在不将计数器保存到磁盘的情况下进行吗?

这是因为@ commands.command()只加载一次.因此,两个机器人共享同一个Command实例.您需要的是在实例级别添加命令,而不是由@ commands.command()装饰器添加.

class TestCog:
    counter = 0

    def __init__(self, bot):
        self.bot = bot
        self.bot.add_command(commands.Command('add', self.add))

    async def add(self):
        TestCog.counter += 1
        await self.bot.say('Counter is now %d' % TestCog.counter)

要么:

class TestCog:
    counter = 0

    def __init__(self, bot):
        self.bot = bot
        self.bot.command()(self.add)

    async def add(self):
        TestCog.counter += 1
        await self.bot.say('Counter is now %d' % TestCog.counter)

为了使两个机器人共享相同的属性.你想要class属性,而不是实例.

翻译自:https://stackoverflow.com/questions/49302170/load-cog-for-multiple-bots