当我清楚地表明我只想完成第一个任务时，为什么所有任务都在asyncio.wait（）中完成？

Question

我正在尝试使用asyncio从OpenSubtitles中获取一些数据，然后下载该数据中包含信息的文件。我想使用asyncio同时获取该数据并下载该文件。

问题是我想等待列表tasks中的1个任务完成，然后开始列表中的其余任务或download_tasks。原因是在self._perform_query()我写信息到文件和self._download_and_save_file()我正在从该文件中读取相同的信息。换句话说，download_tasks需要等待tasks中的至少一项任务在开始之前完成。

我发现我可以用asyncio.wait(return_when=FIRST_COMPLETED)做到这一点，但由于某种原因它不能正常工作：

payloads = [create_payloads(entry) for entry in retreive(table_in_database)] 
tasks = [asyncio.create_task(self._perform_query(payload, proxy)) for payload in payloads]
download_tasks = [asyncio.create_task(self._download_and_save_file(url, proxy) for url in url_list]

done, pending = await asyncio.wait(tasks, return_when=asyncio.FIRST_COMPLETED)
print(done)
print(len(done))
print(pending)
print(len(pending))
await asyncio.wait(download_tasks)

输出与预期完全不同。看来tasks列表中的3个任务中有3个正在完成，尽管我通过了asyncio.FIRST_COMPLETED。为什么会这样？

{<Task finished coro=<SubtitleDownloader._perform_query() done, defined at C:\Users\...\subtitles.py:71> result=None>, <Task finished coro=<SubtitleDownloader._perform_query() done, defined at C:\Users\...\subtitles.py:71> result=None>, <Task finished coro=<SubtitleDownloader._perform_query() done, defined at C:\Users\...\subtitles.py:71> result=None>}
3
set()
0
Exiting

据我所知，self._perform_query()中的代码不应该影响这个问题。无论如何，这只是为了确保：

async def _perform_query(self, payload, proxy):
    try:
        query_result = proxy.SearchSubtitles(self.opensubs_token, [payload], {"limit": 25})
    except Fault as e:
        raise "A fault has occurred:\n{}".format(e)
    except ProtocolError as e:
        raise "A ProtocolError has occurred:\n{}".format(e)
    else:
        if query_result["status"] == "200 OK":
            with open("dl_links.json", "w") as dl_links_json:
                result = query_result["data"][0]
                subtitle_name = result["SubFileName"]
                download_link = result["SubDownloadLink"]
                download_data = {"download link": download_link,
                                 "file name": subtitle_name}
                json.dump(download_data, dl_links_json)
        else:
            print("Wrong status code: {}".format(query_result["status"]))

现在，我一直在测试这个没有运行download_tasks，但我已经将它包含在这里用于上下文。也许我会以完全错误的方式解决这个问题。如果是这样，我将非常感谢您的投入！

编辑：

问题非常简单，如下所述。 _perform_query不是一个等待的功能，而是同步运行。我通过编辑_perform_query的文件写入与aiofiles异步来改变它：

def _perform_query(self, payload, proxy):
    query_result = proxy.SearchSubtitles(self.opensubs_token, [payload], {"limit": 25})
    if query_result["status"] == "200 OK":
        async with aiofiles.open("dl_links.json", mode="w") as dl_links_json:
            result = query_result["data"][0]
            download_link = result["SubDownloadLink"]
            await dl_links_json.write(download_link)