python-3.x - 使用 Subprocess 来避免长时间运行的任务断开 discord.py bot？

Question

我为我的 Discord 服务器创建了一个机器人，它转到给定 subreddit 的 Reddit API，并根据您输入的 subreddit 在 Discord 聊天中发布当天的前 10 个结果。它不理会自己的帖子，实际上只发布图片和 GIF。Discord 消息命令看起来像这样：=get funny awww news programming，发布每个 subreddit 的结果，因为它从 Reddit API (PRAW) 获取它们。这没有问题。我知道该机器人能够访问 API 并发布到不和谐的作品。

我添加了另一个命令=getshuffled，它将来自 subreddits 的所有结果放在一个大列表中，然后在发布之前将它们打乱。这对于最多约 50 个子版块的请求非常有效。

这是我需要帮助的：

因为它可能是一个很大的结果列表，100 多个结果来自 100 多个 subreddit，所以机器人在非常大的请求上崩溃。根据我昨天从我的问题中得到的帮助，我明白出了什么问题。机器人正在启动，它正在与我的 Discord 服务器对话，当我向它传递一个长请求时，它在 Reddit API 调用完成时停止与服务器对话太久，并且 Discord 连接失败。

所以，我认为我需要做的是有一个代码的子进程去 Reddit API 并提取结果，（我认为这会让不和谐的连接保持运行），然后将这些结果传回给机器人当它完成时......

或者...这是 Asyncio 可以自己处理的事情...

正如我知道的那样，我在子流程调用方面遇到了困难。

基本上，我要么需要这个子流程技巧的帮助，要么需要知道我是否是个白痴，而 Asyncio 可以为我处理所有这些。我认为这只是那些“我不知道我不知道什么”的例子之一。

回顾一下：机器人在少量的子目录被洗牌的情况下工作得很好。它通过发送的参数（这是 subreddits），获取每个帖子的信息，然后在发布不和谐的链接之前进行洗牌。问题是当它是一组更大的约 50 多个子版块时。为了让它与更大的数量一起工作，我需要让 Reddit 调用不阻止主要的不和谐连接，这就是我试图创建一个子进程的原因。

Python 版本是 3.6，Discord.py 版本是 0.16.12 这个机器人在 PythonAnywhere 上托管和运行

代码：

from redditBot_auth import reddit

import discord
import asyncio
from discord.ext.commands import Bot
#from discord.ext import commands
import platform
import subprocess
import ast

client = Bot(description="Pulls posts from Reddit", command_prefix="=", pm_help = False)

@client.event
async def on_ready():
    return await client.change_presence(game=discord.Game(name='Getting The Dank Memes')) 

def is_number(s):
    try:
        int(s)
        return True
    except:
        pass

def show_title(s):
    try:
        if s == 'TITLES':
            return True
    except:
        pass

async def main_loop(*args, shuffled=False):
    print(type(args))

    q=10

    #This takes a integer value argument from the input string.
    #It sets the number variable,
    #Then deletes the number from the arguments list.
    title = False
    for item in args:
        if is_number(item):
            q = item
            q = int(q)
            if q > 15:
                q=15
            args = [x for x in args if not is_number(x)]

        if show_title(item):
            title = True
            args = [x for x in args if not show_title(x)]

    number_of_posts = q * len(args)
    results=[]

    TESTING = False #If this is turned to True, the subreddit of each post will be posted. Will use defined list of results


    if shuffled == False: #If they don't want it shuffled

        for item in args:
            #get subreddit results
            #post links into Discord as it gets them
            #The code for this works

    else: #if they do want it shuffled
        output = subprocess.run(["python3.6", "get_reddit.py", "*args"])
        results = ast.literal_eval(output.decode("ascii"))
        # ^^ this is me trying to get the results back from the other process.

. 这是我的 get_reddit.py 文件：

#THIS CODE WORKS, JUST NEED TO CALL THE FUNCTION AND RETURN RESULTS
#TO THE MAIN_LOOP FUNCTION

from redditBot_auth import reddit
import random

def is_number(s):
    try:
        int(s)
        return True
    except:
        pass

def show_title(s):
    try:
        if s == 'TITLES':
            return True
    except:
        pass

async def get_results(*args, shuffled=False):

    q=10

    #This takes a integer value argument from the input string.
    #It sets the number variable,
    #Then deletes the number from the arguments list.
    title = False
    for item in args:
        if is_number(item):
            q = item
            q = int(q)
            if q > 15:
                q=15
            args = [x for x in args if not is_number(x)]

        if show_title(item):
            title = True
            args = [x for x in args if not show_title(x)]

    results=[]

    TESTING = False #If this is turned to True, the subreddit of each post will be posted. Will use defined list of results.
    NoGrabResults = False

    #This pulls the data and creates a list of links for the bot to post

    if NoGrabResults == False:
        for item in args:
            try:
                #get the posts
                #put them in results list    

            except Exception as e:
                #handle error
                pass

        try:
            #print('____SHUFFLED___')
            random.shuffle(results)
            random.shuffle(results)
            random.shuffle(results)

        except:
            #error stuff

        print(results)
#I should be able to read that print statement for the results, 
#and then use that in the main bot function to post the results.

.

@client.command()
async def get(*args, brief="say '=get' followed by a list of subreddits", description="To get the 10 Top posts from a subreddit, say '=get' followed by a list of subreddits:\n'=get funny news pubg'\n would get the top 10 posts for today for each subreddit and post to the chat."):
    #sr = '+'.join(args)
    await main_loop(*args)

#THIS POSTS THE POSTS RANDOMLY   
@client.command()
async def getshuffled(*args, brief="say '=getshuffled' followed by a list of subreddits", description="Does the same thing as =get, but grabs ALL of the posts and shuffles them, before posting."):

    await main_loop(*args, shuffled=True)


client.run('my ID')

更新：按照建议，我通过 ThreadPoolExecutor 传递了命令，如下所示：

async def main(*args, shuffled):

    if shuffled==True:

        with concurrent.futures.ThreadPoolExecutor() as pool:
            results = await asyncio.AbstractEventLoop().run_in_executor(
                executor=pool, func=await main_loop(*args, shuffled=True))
            print('custom thread pool', results)

但是当脚本尝试与 Discord 对话时，这仍然会导致错误：

ERROR:asyncio:Task was destroyed but it is pending!
task: <Task pending coro=<Client._run_event() running at /home/GageBrk/.local/lib/python3.6/site-packages/discord/client.py:307> wait_for=<Future pending cb=[<TaskWakeupMethWrapper object at 0x7f28acd8db28>()]>>
Event loop is closed
Destination must be Channel, PrivateChannel, User, or Object. Received NoneType
Destination must be Channel, PrivateChannel, User, or Object. Received NoneType
Destination must be Channel, PrivateChannel, User, or Object. Received NoneType
...

它正在正确发送结果，但不和谐仍然失去连接。

标签： python-3.xsubprocesspython-asynciodiscord.py

Answer 1

praw依赖于requests库，这是同步的，这意味着代码是阻塞的。如果阻塞代码执行时间过长，这可能会导致您的机器人冻结。

为了解决这个问题，可以创建一个单独的线程来处理阻塞代码。下面是一个例子。注意blocking_function将如何使用time.sleep阻止 10 分钟（600 秒）。这应该足以冻结并最终使机器人崩溃。但是，由于该函数在它自己的线程中 using run_in_executor，因此机器人继续正常运行。

新版本

import time
import asyncio
from discord.ext import commands
from concurrent.futures import ThreadPoolExecutor

def blocking_function():
    print('entering blocking function')
    time.sleep(600)
    print('sleep has been completed')
    return 'Pong'

client = commands.Bot(command_prefix='!')

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

@client.command()
async def ping(ctx):
    loop = asyncio.get_event_loop()
    block_return = await loop.run_in_executor(ThreadPoolExecutor(), blocking_function)
    await ctx.send(block_return)

client.run('token')

旧async版本

import time
import asyncio
from discord.ext import commands
from concurrent.futures import ThreadPoolExecutor

def blocking_function():
    print('entering blocking function')
    time.sleep(600)
    print('sleep has been completed')
    return 'Pong'

client = commands.Bot(command_prefix='!')

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

@client.command()
async def ping():
    loop = asyncio.get_event_loop()
    block_return = await loop.run_in_executor(ThreadPoolExecutor(), blocking_function)
    await client.say(block_return)

client.run('token')