Discord机器人实现涩图功能

上篇文章我们构建了一个最基础的机器人，接下来就得给他加点功能了

作为一个聊天机器人最重要的功能，那当然是涩图！

我们的需求是调用一个指令 /涩图 就可以获取一张涩图，而且还可以带点可选的参数

调用API

这里使用lolicon API（十分感谢API作者）

API可以传入几个参数，我这里就给用户提供三个参数: r18 tags exclude_AI ，使用POST方式请求

先写一个最基础的请求类

def RequestLolicon(tags: list[str], r18: bool, excludeAI: bool) -> dict:

这段代码写了一个基础的请求图片数据的函数，传入三个参数，返回结果。

下载图片

我想让图片直接下载到本地，然后上传到discord的CDN上，所以需要一个图片下载功能

先写一个简单的下载图片函数

def Download(pid: str, url: str) -> str:
    return f'{pid}.{ext}'

这个函数传入两个参数 pid url

其中 url 是我们通过上一个调用API之后的结果获得的，因为这里只是个工具函数，所以不需要去关心API返回的数据。

pid 也是返回的数据，用于文件名

这个函数会把图片下载到 ./cache/setu/{pid}.{ext} 这个地方，其中 ext 根据下载 url 的后缀定

整合

接下来就是整合这两个函数，并且和Discord交互

async def HandleRequest(ctx: SlashContext, args, r18, excludeAI):
        os.remove(f'./cache/setu/{filename}')

这段代码接受一个 SlashContext 参数，是python库 discord-py-interactions 提供给我们的，就是指令的上下文

这四个参数是我们在注册指令的地方传入的

这个函数的大意就是整合请求，下载图片，上传图片，发送消息，删除本地图片为一体

注册Discord Slash指令

现在开始注册Discord的Slash指令

scopes = [XXX]

这里有一些东西需要说明：

• scopes : 指令的作用域，一个数组，里面存的是你要启用指令的服务器ID。

  如果你不设置这个的话，discord默认你在所有服务器里启用，但是指令生效需要1个小时左右，如果你设置了瞬间就会注册好

  如果你设置正确的服务器ID，但是还是没有注册成功，那就有可能是你权限没开

  服务器ID的获取方式是 打开开发者模式 -> 右键服务器名 -> 复制服务器ID，具体方法如下：

  

• 指令注册Slash Command

  @interactions.slash_command(
            )
  

  注册一个基本指令的注解

• 指令参数Slash Option

  @interactions.slash_option(
      )
  

  name代表这个参数的名字，description就是简介

  required 是指定这个参数是不是必要的，如果不必要，那么在函数的参数声明里必须给它一个默认值，比如下面代码的 tags = ""

  opt_type 就是参数的类型，这个类型在 interactions.OptionType 这个枚举里面

• 函数内部:

   await ctx.defer()
   await Setu.HandleRequest(ctx, tags, r18, exclude_ai)
  

  await ctx.defer() 这个代码是让机器人显示一个 正在响应 的文字，通常一些耗时操作需要用到这个指令，正好我们下载图片挺耗时的，就加上这个

测试

以上步骤没问题的话，这个功能应该就能正常实现了，成果图如下: