使用fpdf2/pygal/cairosvg生成PDF图表时遇BytesIO无rfind属性错误

错误信息 #

'_io.BytesIO' object has no attribute 'rfind'  

Traceback (most recent call last):
File ... in generate_pdf
pdf.image(img_bytesio, x=x, y=y, w=w, h=h)
File "/usr/local/lib/python3.8/dist-packages/fpdf/fpdf.py", line 150, in wrapper
font_size_pt: Optional[int] = None,
File "/usr/local/lib/python3.8/dist-packages/fpdf/fpdf.py", line 963, in image
AttributeError: '_io.BytesIO' object has no attribute 'rfind'

涉事代码段 #

svg_img = bar_chart.render()
img_bytesio = BytesIO()
cairosvg.svg2png(svg_img, write_to=img_bytesio, dpi=96)
pdf.add_page()
pdf.image(img_bytesio, x=x, y=y, w=w, h=h)

已确认的库版本 #

Name: fpdf2
Version: 2.7.3
Name: pygal
Version: 3.0.0
Name: CairoSVG
Version: 2.7.0

解决方法 #

错误根源有两点：

1. cairosvg向BytesIO写入图片后，文件指针停留在流的末尾，fpdf2读取时无法获取内容；
2. fpdf2尝试从传入对象的“文件名”推断图片格式，但BytesIO无文件名属性，误将其当作字符串调用rfind方法。

以下两种方法均可解决：

方法一：重置指针并指定图片格式 #

svg_img = bar_chart.render()
img_bytesio = BytesIO()
cairosvg.svg2png(svg_img, write_to=img_bytesio, dpi=96)
img_bytesio.seek(0)  # 将流指针移到起始位置
pdf.add_page()
pdf.image(img_bytesio, x=x, y=y, w=w, h=h, type="PNG")  # 明确指定图片类型

方法二：直接传入字节数据 #

svg_img = bar_chart.render()
img_bytesio = BytesIO()
cairosvg.svg2png(svg_img, write_to=img_bytesio, dpi=96)
img_data = img_bytesio.getvalue()  # 获取流中的字节数据
pdf.add_page()
pdf.image(img_data, x=x, y=y, w=w, h=h, type="PNG")

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