如何捕获IPython中图片显示失败的错误并实现服务降级？

问题场景 #

我在Jupyter Notebook里通过外部服务（比如QuickChart）渲染Graphviz源文件，代码如下：

from IPython.display import Image
url = "https://quickchart.io/graphviz?graph=graph{a--b}"
Image(url=url)

但当服务出现故障（比如宕机、触发速率限制）时，会显示碎图，我需要一个通用方案捕获这类错误，自动切换到备选服务，且不依赖特定服务或图片格式。

解决方案 #

IPython的Image对象不会主动校验远程资源的可用性，所以核心思路是先主动验证图片URL是否有效，再决定渲染或切换服务，具体实现如下：

1. 编写通用的URL校验函数 #

用requests库发送HEAD请求（比GET更高效，仅校验资源可达性和类型）：

import requests

def is_image_url_valid(url, timeout=5):
    try:
        response = requests.head(url, timeout=timeout)
        # 检查状态码为200，且返回内容是图片类型
        return response.status_code == 200 and 'image' in response.headers.get('Content-Type', '')
    except (requests.exceptions.RequestException, ValueError):
        return False

2. 实现多服务自动切换逻辑 #

定义备选服务列表，依次尝试直到找到可用服务：

from IPython.display import Image, display

# 定义Graphviz渲染服务模板，{graph}用来替换Graphviz源码
render_services = [
    "https://quickchart.io/graphviz?graph={graph}",
    # 这里添加其他备选服务，格式统一即可
    "https://another-graphviz-service.com/render?dot={graph}",
    "https://third-service.org/graphviz?input={graph}"
]

# 你的Graphviz源码
graph_source = "graph{a--b}"

valid_image_url = None
# 遍历所有服务，找第一个可用的
for service_template in render_services:
    service_url = service_template.format(graph=graph_source)
    if is_image_url_valid(service_url):
        valid_image_url = service_url
        break

# 渲染或提示无可用服务
if valid_image_url:
    display(Image(url=valid_image_url))
else:
    print("所有Graphviz渲染服务都不可用，请稍后重试或检查服务配置")

3. 可选：添加本地渲染 fallback #

如果所有远程服务都故障，可以用本地graphviz库生成图片作为兜底：

import graphviz

def render_local(graph_source):
    try:
        dot = graphviz.Source(graph_source)
        # 生成png格式的二进制数据
        return dot.pipe(format='png')
    except Exception:
        return None

# 找不到远程服务时尝试本地渲染
if not valid_image_url:
    local_image_data = render_local(graph_source)
    if local_image_data:
        display(Image(data=local_image_data))
    else:
        print("远程服务和本地渲染都失败了")

关键说明 #

• 使用HEAD请求：避免下载完整图片，节省带宽且效率更高
• 通用适配：只要服务支持通过URL传入Graphviz源码，就能加入备选列表，无需修改核心逻辑
• 异常全覆盖：处理了网络超时、连接失败、无效URL等多种错误场景

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