简单站长图片异步爬虫

import asyncio
import aiohttp
import os

async def download_image(url, session, directory):
    async with session.get(url) as response:
        if response.status == 200:
            file_name = os.path.basename(url)
            with open(os.path.join(directory, file_name), 'wb') as f:
                while True:
                    chunk = await response.content.read(1024)
                    if not chunk:
                        break
                    f.write(chunk)

async def main(urls, directory='images'):
    async with aiohttp.ClientSession() as session:
        tasks = [download_image(url, session, directory) for url in urls]
        await asyncio.gather(*tasks)

# 使用方法:
# urls = ['http://example.com/image1.jpg', 'http://example.com/image2.jpg']
# asyncio.run(main(urls))