Как вызвать асинхронный код из кода синхронизации в другом потоке? - PullRequest
Новая
       37

Как вызвать асинхронный код из кода синхронизации в другом потоке?

0 голосов
/ 08 июня 2019

Я создаю бот Discord, который отправляет PM при получении хука Github.

Он использует Discord.py и BottlePy, последний из которых запускается в выделенном потоке.Поскольку обе платформы имеют блокирующий главный цикл.

При обратном вызове BottlePy я вызываю некоторый асинхронный код Discord.py.

Я не знал, что такое асинхронность Python, это кажется сложным, когдасмешанный с синхронным кодом ...

Вот полный исходный код: 

import discord
import bottle
import threading
import asyncio

client = discord.Client()
server = bottle.Bottle()

async def dm_on_github_async(userid,request):
    print("Fire Discord dm to "+str(userid))
    global client
    user = client.get_user(userid)
    if (user==None):
        abort(500, "User lookup failed");

    dm_channel = user.dm_channel
    if (dm_channel==None):
        dm_channel = await user.create_dm()
    if (dm_channel==None):
        abort(500, "Fail to create DM channel");
    print("DM channel is "+str(asyncio.wait(dm_channel.id)))
    await dm_channel.send("There's a Github shot !")
    await dm_channel.send(str(request.body))
    return
@server.post("/dm_on_github/<userid:int>")
def dm_on_github(userid):
    return asyncio.run(dm_on_github_async(userid,bottle.request))
@client.event
async def on_ready():
    print('We have logged in as {0.user} '.format(client))

#@client.event
#async def on_message(message):
#    if message.author == client.user:
#        return
#
#    if message.content.startswith('$hello'):
#        await message.channel.send('Hello!')
#    # This sample was working very well

class HTTPThread(threading.Thread):
    def run(self):
        global server
        server.run(port=8080)
server_thread = HTTPThread()
print("Starting HTTP server")
server_thread.start()
print("Starting Discord client")
client.run('super secret key')
print("Client terminated")
server.close()
print("Asked server to terminate")
server_thread.join()
print("Server thread successful join")

Я хочу, чтобы мой бот Python отправлял тело HTTP-запроса в виде PM.

Я получаю RuntimeError: Timeout context manager should be used inside a task в return asyncio.run(dm_on_github_async(userid,bottle.request)).

Я думаю, что я не делаю этот микс правильно ...

1 Ответ

0 голосов
/ 09 июня 2019

После ночи я нашел способ.

Чтобы вызвать асинхронный код из кода синхронизации в другом потоке, мы просим loop (здесь этот из Discord.py) выполнить обратный вызов с asyncio.run_coroutine_threadsafe(), это возвращает Task(), и мы ждем его результата с result().

Обратный вызов будет выполняться в потоке цикла , в моем случае мне нужно copy() Запрос на бутылку.

Вот рабочая программа (если вы не против ее остановить ...): 

import discord
import bottle
import threading
import asyncio

client = discord.Client()
server = bottle.Bottle()
class HTTPThread(threading.Thread):
    def run(self):
        global server
        server.run(port=8080)

async def dm_on_github_async(userid,request):
    user = client.get_user(userid)
    if (user==None):
        abort(500, "User lookup failed");
    dm_channel = user.dm_channel
    if (dm_channel==None):
        dm_channel = await user.create_dm()
    if (dm_channel==None):
        abort(500, "Fail to create DM channel");
    # Handle the request
    event = request.get_header("X-GitHub-Event")
    await dm_channel.send("Got event "+str(event))
    #await dm_channel.send(str(request.body)) # Doesn't work well...
    return

@server.post("/dm_on_github/<userid:int>")
def dm_on_github(userid):
    request = bottle.request
    asyncio.run_coroutine_threadsafe(dm_on_github_async(userid,request.copy()),client.loop).result()


@client.event
async def on_ready():
    print('We have logged in as {0.user} '.format(client))
    # Wait for the old HTTP server
    if hasattr(client,"server_thread"):
        server.close()
        client.server_thread.join()
    client.server_thread = HTTPThread()
    client.server_thread.start()

#@client.event
#async def on_message(message):
#    if message.author == client.user:
#        return
#
#    if message.content.startswith('$hello'):
#        await message.channel.send('Hello!')

print("Starting Discord client")
client.run('super secret key')
print("Client terminated")
server.close()
print("Asked server to terminate")
server_thread.join()
print("Server thread successful join")
...