Discord.py 中的异步 REST API

Question

我正在寻找一种使用重写分支将 REST API 集成到我的 Discord.py 中的方法。我想使用 aiohttp 来处理请求，但我不确定应该采用哪种方法。例如，objective 向 API 发出 GET 请求，该请求将 return 机器人所在的公会列表。或者作为另一个示例，POST 请求会要求机器人将给定消息写入特定频道。总的来说，它是关于从网页向机器人发出指令。

我尝试将 aiohttp 应用路由器和运行ner 放入我的 Discord.py 客户端 class。 Web 服务器确实是运行ning，我为机器人所在的 return 行会创建了一个异步函数，但看起来该函数不会接受我在转到 [ 时传递给它的请求参数=15=]。从而导致 missing 1 required positional argument 错误。

import discord
import asyncio
from aiohttp import web

class MyClient(discord.Client):
    async def on_ready(self):
        print('Logged on as {0}!'.format(self.user))

    async def get_guilds(self, request):
        response_obj = self.guilds
        return web.json_response(response_obj,status=200,content_type='application/json')

    app = web.Application()
    app.router.add_get('/guilds', get_guilds)
    web.run_app(app, port=80)

client = MyClient()
client.run(TOKEN)

此外，aiohttp 服务器不会运行异步。我期望 on_ready(self) 到运行，但它从来没有。我做错了什么？

Answer 1

我认为下面的内容应该运行这两个任务一起工作。但这不是很好的设计，只会变得更加复杂。将机器人和服务器分开并使用比共享内存更结构化的方式在它们之间进行通信可能会更好地为您服务。

import discord
import asyncio
from aiohttp import web

client = discord.Client()

async def on_ready():
    print('Logged on as {0}!'.format(client.user))

async def get_guilds(request):
    response_obj = client.guilds
    return web.json_response(response_obj,status=200,content_type='application/json')


try:
    bot_task = client.loop.create_task(client.start("token"))
    app = web.Application()
    app.router.add_get('/guilds', get_guilds)
    web.run_app(app, port=80)
except:
    client.loop.run_until_complete(client.logout())

Answer 2

嗯，我找到了一个方法。

bot.py

from asyncio import gather, get_event_loop
from logging import basicConfig, INFO
from discord.ext.commands import Bot
from aiohttp.web import AppRunner, Application, TCPSite
from sys import argv

from api import routes

basicConfig(level=INFO)

async def run_bot():

    app = Application()
    app.add_routes(routes)

    runner = AppRunner(app)
    await runner.setup()
    site = TCPSite(runner, '0.0.0.0', 8080)
    await site.start()

    bot = Bot(command_prefix="$")
    app['bot'] = bot

    try:
        await bot.start(TOKEN)

    except:
        bot.close(),
        raise

    finally:
        await runner.cleanup()

if __name__ == '__main__':
    loop = get_event_loop()
    loop.run_until_complete(run_bot())

api.py

routes = RouteTableDef()

@routes.get('/guilds')
async def get_guilds(request):
    client = request.app['bot']
    guilds = []
    for guild in client.guilds:
        guilds.append(guild.id)

    response = Handler.success(guilds)
    return json_response(response, status=200, content_type='application/json')

Discord.py 中的异步 REST API

Asynchronous REST API inside Discord.py

python

python-asyncio

aiohttp

discord.py

discord.py-rewrite

bot.py

api.py