如何利用Foundry Transform API保存ReportLab生成的多表格PDF？

核心思路 #

在Foundry Transform中，将ReportLab生成的PDF二进制内容写入Foundry存储系统，通过transforms.api提供的文件操作接口完成保存。

具体步骤 #

1. 配置环境依赖 #

在Transform的环境配置中添加reportlab依赖，确保代码能正常导入使用。

2. 编写完整处理逻辑 #

读取多个Foundry数据集，处理后用ReportLab生成表格并组装成PDF，再通过Transform API将二进制内容写入指定路径。

示例代码片段：

from transforms.api import transform, Output, Input
from reportlab.platypus import SimpleDocTemplate, Table, TableStyle
from reportlab.lib import colors
import io

@transform(
    output=Output("/your/foundry/path/to/save/pdf"),
    inputs=[Input("/your/foundry/path/dataset1"), Input("/your/foundry/path/dataset2")]
)
def generate_and_save_pdf(ctx, dataset1, dataset2):
    # 读取并处理数据集
    df1 = dataset1.dataframe()
    df2 = dataset2.dataframe()
    # 自定义数据处理逻辑（示例：转为列表格式）
    table_data1 = [df1.columns.tolist()] + df1.to_numpy().tolist()
    table_data2 = [df2.columns.tolist()] + df2.to_numpy().tolist()

    # 使用ReportLab生成PDF
    buffer = io.BytesIO()
    doc = SimpleDocTemplate(buffer)
    elements = []

    # 生成第一个表格并添加样式
    table1 = Table(table_data1)
    table1.setStyle(TableStyle([
        ('BACKGROUND', (0,0), (-1,0), colors.grey),
        ('TEXTCOLOR', (0,0), (-1,0), colors.whitesmoke),
        ('ALIGN', (0,0), (-1,-1), 'CENTER'),
        ('GRID', (0,0), (-1,-1), 1, colors.black)
    ]))
    elements.append(table1)

    # 生成第二个表格（可扩展更多表格）
    table2 = Table(table_data2)
    table2.setStyle(TableStyle([
        ('BACKGROUND', (0,0), (-1,0), colors.grey),
        ('TEXTCOLOR', (0,0), (-1,0), colors.whitesmoke),
        ('ALIGN', (0,0), (-1,-1), 'CENTER'),
        ('GRID', (0,0), (-1,-1), 1, colors.black)
    ]))
    elements.append(table2)

    # 构建PDF并获取二进制内容
    doc.build(elements)
    buffer.seek(0)
    pdf_bytes = buffer.getvalue()

    # 通过Transform API保存PDF
    ctx.output.set_content(pdf_bytes, content_type="application/pdf")

3. 关键注意事项 #

• 路径权限：确保指定的Output路径有写入权限，输入数据集路径有读取权限。
• 内容类型：必须设置content_type="application/pdf"，让Foundry正确识别文件类型。
• 无状态处理：用io.BytesIO缓存PDF内容，避免写入本地文件，适配Transform的运行环境。

4. 验证结果 #

提交Transform作业并运行完成后，在指定的Output路径下即可找到生成的PDF文件，打开检查表格是否符合预期。

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