使用Python asyncio获取图片尺寸无结果，求代码问题排查

我来帮你拆解代码里的问题，以及对应的修复方案：

1. 没有收集并返回异步任务的结果 #

你的main函数里，虽然通过asyncio.gather拿到了所有线程池任务的结果，但只是写了个空循环for response in ...: pass，完全没把结果保存或返回出去。外部调用loop.run_until_complete(main(url_list))时根本拿不到这些结果，自然看不到有效输出。

2. 过于宽泛的异常捕获掩盖了真实问题 #

get_image_shape里用except:捕获所有异常，这会把网络请求失败、URL无效、图片格式错误等各种问题都掩盖，统一返回(0,0,0)，你根本没法排查到底是哪里出了问题。应该针对性捕获具体异常，或者至少打印异常信息方便定位。

3. 用同步请求库适配异步场景（可选优化） #

你用requests.get这种同步HTTP请求在线程池里跑，虽然能勉强工作，但效率不如用异步HTTP客户端（比如aiohttp）更贴合asyncio的异步模型，线程池还会带来额外的开销。

修复后的完整代码（推荐异步HTTP版本） #

下面是修正后的代码，解决了所有问题，并且用更高效的异步方式实现：

from PIL import Image
from io import BytesIO
import numpy as np
import asyncio
import aiohttp  # 先安装：pip install aiohttp

async def get_image_shape(session, url):
    try:
        # 用异步HTTP请求获取图片资源
        async with session.get(url) as resp:
            if resp.status != 200:
                print(f"请求失败 - URL: {url}，状态码: {resp.status}")
                return (0,0,0)
            img_data = await resp.read()
            img = Image.open(BytesIO(img_data))
            arr = np.array(img, dtype=np.uint8)
            return arr.shape
    except aiohttp.ClientError as e:
        print(f"网络错误 - URL: {url}，错误信息: {str(e)}")
        return (0,0,0)
    except Exception as e:
        print(f"图片处理错误 - URL: {url}，错误信息: {str(e)}")
        return (0,0,0)

async def main(url_list):
    # 创建异步HTTP会话，复用连接提升效率
    async with aiohttp.ClientSession() as session:
        tasks = [get_image_shape(session, url) for url in url_list]
        # 收集所有异步任务的结果
        results = await asyncio.gather(*tasks)
        return results

# 替换成你的实际图片URL列表
url_list = [
    "https://example.com/test1.jpg",
    "https://example.com/test2.png",
    # ... 更多URL
]

if __name__ == "__main__":
    # 执行异步任务并获取结果
    results = asyncio.run(main(url_list))
    print("图片尺寸结果：", results)

关键修复点说明 #

• 返回并收集结果：main函数现在会返回asyncio.gather的结果，外部通过asyncio.run获取并打印，你能直接看到每个URL的处理结果。
• 精准异常处理：分类型捕获网络错误和图片处理错误，并打印具体的URL和错误信息，方便你快速定位哪个链接出了问题。
• 改用异步HTTP客户端：用aiohttp替代同步的requests，避免线程池的额外开销，更符合asyncio的异步设计理念。
• 现代asyncio写法：用asyncio.run替代旧的get_event_loop().run_until_complete，这是Python 3.7+官方推荐的简洁写法。

如果你想保留线程池+requests的写法 #

如果你暂时不想切换到aiohttp，也可以修正原有代码的问题：

from PIL import Image
from io import BytesIO
import numpy as np
import asyncio
import concurrent.futures
import requests

def get_image_shape(url):
    try:
        resp = requests.get(url)
        resp.raise_for_status()  # 主动抛出HTTP请求错误
        img = Image.open(BytesIO(resp.content))
        arr = np.array(img, dtype=np.uint8)
        return arr.shape
    except requests.exceptions.RequestException as e:
        print(f"网络请求错误 - URL: {url}，错误信息: {str(e)}")
        return (0,0,0)
    except Exception as e:
        print(f"图片处理错误 - URL: {url}，错误信息: {str(e)}")
        return (0,0,0)

async def main(url_list):
    loop = asyncio.get_event_loop()
    # 显式创建线程池，更可控
    with concurrent.futures.ThreadPoolExecutor() as executor:
        futures = [
            loop.run_in_executor(executor, get_image_shape, url)
            for url in url_list
        ]
        results = await asyncio.gather(*futures)
        return results

url_list = [
    "https://example.com/test1.jpg",
    # ... 你的URL
]

if __name__ == "__main__":
    results = asyncio.run(main(url_list))
    print("图片尺寸结果：", results)

这样修改后，你就能看到每个URL的处理结果，以及对应的错误信息，再也不会全是(0,0,0)的无效结果了。

内容的提问来源于stack exchange，提问作者Mike