使用Django与ReportLab生成PDF时外键字段引发500错误咨询

问题场景 #

使用Django结合ReportLab生成PDF时，添加外键关联的province3字段会触发服务器500错误，移除该字段后PDF生成恢复正常；但该字段可正常导出为CSV文件。相关模型中province3是允许为null的外键字段，关联到Province模型。

问题原因 #

1. ReportLab的textLine()方法仅接受字符串类型参数，无法直接处理Django模型实例或None值
2. 当customer.province3为null时，该值是None，直接传入textLine()会引发类型错误
3. 即使province3有值，直接传递Province模型实例时，ReportLab不会自动调用其__str__方法转换为字符串（CSV导出能正常工作是因为CSV处理逻辑会自动触发str()转换）

解决方法 #

显式将province3字段转换为字符串，并处理null的情况，确保传入textLine()的始终是字符串类型。

修改后的views.py代码 #

def report_pdf(request):
    # Create a Bytestream buffer
    buf = io.BytesIO()
    # Create a canvas
    c = canvas.Canvas(buf, pagesize=letter, bottomup=0)
    # Create a text object
    textob = c.beginText()
    textob.setTextOrigin(inch, inch)
    textob.setFont("Helvetica", 14)

    # 优化查询，避免N+1数据库查询
    customers = Customer.objects.all().select_related('province3')

    lines =[]

    for customer in customers:
        # 处理普通字段的null情况
        lines.append(customer.address or "")
        lines.append(customer.address2 or "")
        lines.append(customer.city or "")
        # 处理外键字段，转换为字符串并处理null
        lines.append(str(customer.province3) if customer.province3 else "")
        
    # 写入所有行
    for line in lines:
        textob.textLine(line)
    
    # 完成PDF生成
    c.drawText(textob)
    c.showPage()
    c.save()
    buf.seek(0)

    return FileResponse(buf, as_attachment=True, filename='customer_list.pdf')

额外优化 #

如果不想显示"None"而是空字符串，也可以直接取关联模型的字段值：

lines.append(customer.province3.province if customer.province3 else "")

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