python使用asyncio 异步完成成批量下载图片

一个python使用异步IO asyncio批量下载图片的代码

# ---encoding:utf-8---
# @Time    : 2024/8/9 12:03
# @Author  : 公众号：右恩
# @File    : 2、基于协程下载图片.py
# @Project : pipProject
# @Software: PyCharm
# love 
# If I can't be a god, I'll be a beast

# 导入异步编程模块asyncio和异步HTTP客户端aiohttp
import asyncio
import aiohttp

# 定义异步函数fetch，用于发送HTTP请求并下载文件
async def fetch(session, url):
    # 打印发送请求的信息
    print("发送请求",url)
    # 使用with语句发起GET请求，verify_ssl=False表示不验证SSL证书
    async with session.get(url, verify_ssl=False) as response:
        # 读取响应的内容
        connet = await response.content.read()
        # 从URL中提取文件名
        file_name = url.split("/")[-1]
        # 以二进制写模式打开文件，将内容写入文件
        with open(file_name, "wb") as f:
            f.write(connet)
        # 打印下载完成的信息
        print("下载完成，保存文件",file_name)

# 定义异步函数main，作为程序的入口
async def main():
    # 使用with语句创建一个异步HTTP会话
    async with aiohttp.ClientSession() as session:
        # 定义待下载的URL列表
        urls = [
            "http://aaa.jpg",
            "http://aaa1.jpg",
        ]
    # 创建异步任务列表
    tasks = [ asyncio.create_task(fetch(session, url) for url in urls)]
    # 等待所有异步任务完成
    await asyncio.wait(tasks)

# 程序的主入口
if __name__ == '__main__':
    # 运行异步函数main
    asyncio.run(main())
    # 打印下载完成的信息
    print("下载完成")