Django中InMemoryUploadedFile格式PDF转图片报错及优化方案咨询

你的错误核心是**convert_from_path() 方法需要传入文件路径字符串，而不是文件字节流**，直接传request.FILES['file'].read()会导致poppler无法识别输入，进而触发页面计数失败和后续的UnboundLocalError。

方法1：临时文件处理（推荐） #

用Python的tempfile模块创建临时文件，自动处理文件写入与清理：

from pdf2image import convert_from_path
import tempfile
import os

def pdf_to_images(pdf_file):
    # 创建临时PDF文件并写入上传内容
    with tempfile.NamedTemporaryFile(delete=False, suffix='.pdf') as temp_pdf:
        temp_pdf.write(pdf_file.read())
        temp_pdf_path = temp_pdf.name
    
    try:
        # 执行PDF转图片
        pages = convert_from_path(temp_pdf_path)
        return pages
    finally:
        # 强制清理临时文件
        os.unlink(temp_pdf_path)

# 调用示例
pages = pdf_to_images(request.FILES['file'])

方法2：固定路径保存后转换 #

如果需要保留原始PDF文件，可先将上传文件保存到本地目录：

from pdf2image import convert_from_path
import os

def pdf_to_images(pdf_file, save_dir='./uploaded_pdfs'):
    # 创建保存目录（不存在则自动创建）
    os.makedirs(save_dir, exist_ok=True)
    pdf_path = os.path.join(save_dir, pdf_file.name)
    
    # 分块写入上传文件
    with open(pdf_path, 'wb') as f:
        for chunk in pdf_file.chunks():
            f.write(chunk)
    
    try:
        pages = convert_from_path(pdf_path)
        return pages
    except Exception as e:
        raise e
    # 可选：不需要保留PDF时，取消下面注释删除文件
    # os.unlink(pdf_path)

# 调用示例
pages = pdf_to_images(request.FILES['file'])

额外注意事项 #

• 确保已安装pdf2image依赖的poppler工具：
  • Windows：下载poppler二进制包，添加到系统PATH，或在convert_from_path中指定poppler_path参数。
  • Linux：通过包管理器安装，如apt-get install poppler-utils。
  • macOS：用Homebrew安装brew install poppler。
• 大文件场景下，临时文件方式更节省磁盘空间，避免冗余文件留存。

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