python如何生成pdf

Python生成PDF的方法有多种，包括使用ReportLab库、PyFPDF库、以及结合PDF模板库如Jinja2等。在这些方法中，ReportLab库是最为强大和灵活的选择之一，而PyFPDF则更为轻量级和易用。ReportLab允许你创建复杂和专业的PDF文档，而PyFPDF更适合简单的PDF生成任务。接下来，我们将详细介绍如何使用这些工具来生成PDF文件。

一、REPORTLAB库

ReportLab是一个功能强大的Python库，用于创建PDF文档。它支持创建复杂的PDF，包括嵌入的图片、图形、表格和样式化的文本。

安装ReportLab

要使用ReportLab库，你首先需要安装它。你可以通过pip命令来安装：

pip install reportlab

创建基本PDF文档

使用ReportLab创建PDF文档的基本步骤如下：

from reportlab.lib.pagesizes import letter
from reportlab.pdfgen import canvas
def create_pdf(filename):
    c = canvas.Canvas(filename, pagesize=letter)
    c.drawString(100, 750, "Hello, ReportLab!")
    c.save()
create_pdf("example_reportlab.pdf")

在这个例子中，我们创建了一个简单的PDF文件，包含了“Hello, ReportLab!”这行文本。

添加图形和图片

ReportLab允许你在PDF中添加图形和图片。以下是一个简单的例子：

from reportlab.lib import colors
def create_pdf_with_graphics(filename):
    c = canvas.Canvas(filename, pagesize=letter)
    c.setStrokeColor(colors.blue)
    c.setFillColor(colors.lightgrey)
    c.rect(100, 600, 400, 200, fill=1)
    c.drawImage("example.jpg", 150, 620, width=200, height=150)
    c.save()
create_pdf_with_graphics("example_graphics.pdf")

在这个例子中，我们绘制了一个填充的矩形和一张图片。

添加表格和样式化文本

ReportLab还支持复杂的文本格式和表格。你可以使用Paragraph和Table类来实现：

from reportlab.platypus import SimpleDocTemplate, Table, TableStyle, Paragraph
from reportlab.lib.styles import getSampleStyleSheet
def create_pdf_with_table(filename):
    doc = SimpleDocTemplate(filename, pagesize=letter)
    elements = []
    data = [
        ['Header 1', 'Header 2', 'Header 3'],
        ['Row 1, Col 1', 'Row 1, Col 2', 'Row 1, Col 3'],
        ['Row 2, Col 1', 'Row 2, Col 2', 'Row 2, Col 3'],
    ]
    table = Table(data)
    table.setStyle(TableStyle([
        ('BACKGROUND', (0, 0), (-1, 0), colors.grey),
        ('TEXTCOLOR', (0, 0), (-1, 0), colors.whitesmoke),
        ('ALIGN', (0, 0), (-1, -1), 'CENTER'),
        ('FONTNAME', (0, 0), (-1, 0), 'Helvetica-Bold'),
        ('BOTTOMPADDING', (0, 0), (-1, 0), 12),
        ('BACKGROUND', (0, 1), (-1, -1), colors.beige),
    ]))
    elements.append(table)
    styles = getSampleStyleSheet()
    paragraph = Paragraph("This is a paragraph in ReportLab.", styles['Normal'])
    elements.append(paragraph)
    doc.build(elements)
create_pdf_with_table("example_table.pdf")

这个例子展示了如何在PDF中添加一个简单的表格和一段文本。

二、PYFPDF库

PyFPDF是一个轻量级的PDF生成库，易于使用且不依赖于外部库。它特别适合生成简单的PDF文档。

安装PyFPDF

你可以通过以下命令安装PyFPDF：

pip install fpdf

创建基本PDF文档

以下是使用PyFPDF创建PDF文档的基本示例：

from fpdf import FPDF
class PDF(FPDF):
    def header(self):
        self.set_font('Arial', 'B', 12)
        self.cell(0, 10, 'PDF Header', align='C', ln=True)
    def footer(self):
        self.set_y(-15)
        self.set_font('Arial', 'I', 8)
        self.cell(0, 10, f'Page {self.page_no()}', align='C')
pdf = PDF()
pdf.add_page()
pdf.set_font('Arial', '', 12)
pdf.cell(0, 10, 'Hello, PyFPDF!', ln=True)
pdf.output('example_pyfpdf.pdf')

在这个例子中，我们创建了一个PDF文件，并添加了页眉和页脚。

添加图片和多行文本

PyFPDF允许你添加图片和处理多行文本：

pdf = PDF()
pdf.add_page()
pdf.set_font('Arial', 'B', 16)
pdf.cell(0, 10, 'Welcome to PyFPDF', ln=True)
pdf.set_font('Arial', '', 12)
pdf.multi_cell(0, 10, 'This is an example of multi-line text in PyFPDF. '
                      'You can add more text here to see how it wraps around.')
pdf.image('example.jpg', x=10, y=60, w=100)
pdf.output('example_image_text.pdf')

这个例子展示了如何在PDF中添加图片和多行文本。

三、结合JINJA2和PDF模板

当你需要生成基于模板的PDF时，可以结合Jinja2和WeasyPrint等工具来实现。这种方法特别适合生成复杂布局的PDF，如发票和报告。

安装Jinja2和WeasyPrint

首先安装必要的库：

pip install jinja2 weasyprint

创建HTML模板

创建一个HTML模板文件template.html，其中包含Jinja2占位符：

<!DOCTYPE html>
<html>
<head>
    <title>PDF Template</title>
    <style>
        body { font-family: Arial, sans-serif; }
        .header { text-align: center; font-size: 24px; }
        .content { margin: 20px; }
    </style>
</head>
<body>
    <div class="header">{{ title }}</div>
    <div class="content">
        <p>{{ content }}</p>
    </div>
</body>
</html>

渲染模板并生成PDF

使用Python代码来渲染模板并生成PDF：

from jinja2 import Environment, FileSystemLoader
from weasyprint import HTML
def generate_pdf_from_template(output_filename, context):
    env = Environment(loader=FileSystemLoader('.'))
    template = env.get_template('template.html')
    html_out = template.render(context)
    HTML(string=html_out).write_pdf(output_filename)
context = {
    'title': 'PDF Generation with Jinja2',
    'content': 'This PDF is generated using a Jinja2 template.'
}
generate_pdf_from_template('example_template.pdf', context)