如何无需临时中间文件将Tkinter控件追加至已有PDF？

可以通过io.BytesIO在内存中生成图像对应的PDF字节流，直接传递给PyPDF2完成追加，完全避免临时文件的使用，同时保留PyPDF2的灵活性。

修改后的完整代码 #

import tkinter as tk
from PIL import ImageGrab
import io
import PyPDF2

def process_pdf(widget, pdf_filepath):
    """Append Tkinter widget to pdf without temporary files"""

    # 捕获控件图像
    img = capture_widget(widget)

    # 创建PDF合并器
    merger = PyPDF2.PdfMerger()

    # 添加原PDF文件
    with open(pdf_filepath, 'rb') as pdf_org:
        merger.append(pdf_org)
    
    # 内存中生成图像对应的PDF字节流
    img_pdf_bytes = io.BytesIO()
    img.save(img_pdf_bytes, format='PDF')
    img_pdf_bytes.seek(0)  # 将文件指针移到开头，以便PyPDF2读取

    # 追加内存中的PDF流
    merger.append(img_pdf_bytes)

    # 写入输出文件
    with open(pdf_filepath, "wb") as outputfile:
        merger.write(outputfile)


def capture_widget(widget):
    """Take screenshot of the passed widget"""

    x0 = widget.winfo_rootx()
    y0 = widget.winfo_rooty()
    x1 = x0 + widget.winfo_width()
    y1 = y0 + widget.winfo_height()

    img = ImageGrab.grab((x0, y0, x1, y1))
    return img

关键改进说明 #

• 使用io.BytesIO替代磁盘临时文件，所有操作在内存中完成，减少IO开销，代码更简洁。
• 采用with语句管理文件对象，自动完成资源关闭，避免手动关闭文件的冗余代码。
• 调用img.save()时指定format='PDF'，确保图像正确转换为PDF格式的字节流，之后通过seek(0)重置指针位置，让PyPDF2能正确读取流内容。

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