python - 限制命令

Question

我得到了这段代码，它阻止除了 LIST_OF_ID 中的用户 ID 之外的所有人对 ?hello 命令的访问，因此我通过替换 ctx.message.author 来修改此代码.id 到 ctx.message.server.id 现在它也绕过服务器。所以我添加了这两个代码，但现在它不适用于用户和服务器，只适用于任何人。如何使其适用于服务器和用户。

LIST_OF_SERVER_IDS = ['3557657657647676', '36567565756766767']
LIST_OF_USER_IDS = ['3557657657647676', '36567565756766767'] 

def is_any_user(ids):
    def predicate(ctx):
        return ctx.message.author.id in ids
    return commands.check(predicate)

def is_any_server(ids):
    def predicate(ctx):
        return ctx.message.server.id in ids
    return commands.check(predicate)

@bot.command(pass_context=True)
@is_any_user(LIST_OF_USER_IDS)
@is_any_server(LIST_OF_SERVER_IDS)
async def hello(ctx):
     await bot.say("Hello {}".format(ctx.message.author.mention))

Answer 1

通过定义两个 is_any_user 装饰器，您只需保留最近定义的装饰器，并丢失第一个装饰器。给服务器检查一个更具代表性的名称(毕竟，它不是检查User，为什么名称中有“user”)。然后我们可以保留两个id白名单。由于commands.check可以链接，我们只需用这两个检查来装饰我们的命令。

LIST_OF_SERVER_IDS = [3557657657647676, 36567565756766767, 343657687786435432]
LIST_OF_USER_IDS = [1, 2, 3]  
# Replace as needed.  Keep in mind that 0.16 ids are strings and 1.0 ids are integers


def is_any_user(ids):
    def predicate(ctx):
        return ctx.message.author.id in ids
    return commands.check(predicate)

def is_any_server(ids):
    def predicate(ctx):
        return ctx.message.server.id in ids
    return commands.check(predicate)

@bot.command(pass_context=True)
@is_any_user(LIST_OF_USER_IDS)
@is_any_server(LIST_OF_SERVER_IDS)
async def hello(ctx):
     await bot.say("Hello {}".format(ctx.message.author.mention))

编辑:

使用这些版本的检查来尝试调试正在发生的事情

def is_any_user(ids):
    def predicate(ctx):
        if ctx.message.author.id in ids:
            print("Good author")
            return True
        else:
            print("Bad author")
            return False
    return commands.check(predicate)

def is_any_server(ids):
    def predicate(ctx):
        if ctx.message.server.id in ids:
            print("Good server")
            return True
        else:
            print("Bad server")
            return False
    return commands.check(predicate)

编辑2:

您可以编写一个检查来查看某人是否在用户白名单中或是否从服务器白名单上的服务器访问命令。

def whitelists(users, servers):
    def predicate(ctx):
        return ctx.message.author.id in users or ctx.message.server.id in servers
    return commands.check(predicate)

@bot.command(pass_context=True)
@whitelists(LIST_OF_USER_IDS, LIST_OF_SERVER_IDS)
async def hello(ctx):
     await bot.say("Hello {}".format(ctx.message.author.mention))