python异步协程发起请求并执行异步下载的代码
时间: 2023-12-30 18:06:35 浏览: 237
下面是一个使用异步协程发起请求并执行异步下载的Python示例代码:
```python
import asyncio
import aiohttp
async def download_file(url):
async with aiohttp.ClientSession() as session:
async with session.get(url) as response:
filename = url.split("/")[-1]
with open(filename, "wb") as f:
while True:
chunk = await response.content.read(1024)
if not chunk:
break
f.write(chunk)
print(f"{filename} has been downloaded.")
async def main():
urls = [
"https://example.com/file1.pdf",
"https://example.com/file2.pdf",
"https://example.com/file3.pdf",
"https://example.com/file4.pdf"
]
tasks = [asyncio.create_task(download_file(url)) for url in urls]
await asyncio.gather(*tasks)
if __name__ == "__main__":
asyncio.run(main())
```
在这个示例代码中,我们使用了 `aiohttp` 库来发起异步请求。我们定义了 `download_file` 函数来执行异步下载任务,其中使用了 `async with` 语句来创建一个 `ClientSession` 对象,并发起异步请求。我们使用了 `response.content.read` 方法来逐块读取响应内容,并将其写入磁盘文件中。最后,我们使用 `asyncio.gather` 方法来并行执行多个下载任务,并等待它们全部完成。在 `main` 函数中,我们定义了需要下载的文件列表,并使用 `asyncio.create_task` 方法来创建下载任务。最后,我们通过 `asyncio.run` 方法来运行整个异步协程程序。
阅读全文