python如何生成pdf文件

Python生成PDF文件的方法有很多，主要包括使用ReportLab库、FPDF库、WeasyPrint库、以及PyPDF2库。这些库各有特点，适用于不同的应用场景。最常用和功能强大的要数ReportLab，它能够创建复杂的PDF文档。

一、REPORTLAB库

ReportLab是Python生成PDF文件的最强大、最灵活的库之一。它允许用户创建复杂的PDF文档，支持绘图、文本、表格、图像等多种元素。以下是使用ReportLab生成PDF文件的详细步骤：

安装ReportLab

首先，您需要安装ReportLab库，可以使用pip命令：

pip install reportlab

生成基本PDF文件

接下来，您可以使用以下代码生成一个简单的PDF文件：

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

在这个示例中，我们创建了一个名为“output.pdf”的文件，并在页面上绘制了“Hello, World!”文本。

添加更多内容

您可以使用ReportLab添加更多复杂的内容，包括图像、表格、绘图等。下面是一个示例，展示如何添加图像和表格：

from reportlab.lib import colors
from reportlab.lib.pagesizes import letter
from reportlab.platypus import SimpleDocTemplate, Table, TableStyle, Image
def create_pdf(filename):
    doc = SimpleDocTemplate(filename, pagesize=letter)
    elements = []
    # 添加图像
    img = Image('path/to/image.jpg', 2 * inch, 2 * inch)
    elements.append(img)
    # 添加表格
    data = [['Item', 'Description', 'Price'],
            ['1', 'Product 1', '$10'],
            ['2', 'Product 2', '$20'],
            ['3', 'Product 3', '$30']]
    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'),
                               ('FONTSIZE', (0, 0), (-1, 0), 14),
                               ('BOTTOMPADDING', (0, 0), (-1, 0), 12),
                               ('BACKGROUND', (0, 1), (-1, -1), colors.beige),
                               ('GRID', (0, 0), (-1, -1), 1, colors.black)]))
    elements.append(table)
    doc.build(elements)
create_pdf("output.pdf")

在这个示例中，我们在PDF中添加了一张图像和一个简单的表格。

二、FPDF库

FPDF是另一个常用的Python库，用于生成PDF文件。它比ReportLab更简单，但功能相对较少。适用于需要快速生成简单PDF文件的场景。

安装FPDF

首先，您需要安装FPDF库：

pip install fpdf

生成基本PDF文件

以下是使用FPDF生成一个简单PDF文件的示例：

from fpdf import FPDF
class PDF(FPDF):
    def header(self):
        self.set_font('Arial', 'B', 12)
        self.cell(0, 10, 'Sample PDF', 0, 1, 'C')
    def footer(self):
        self.set_y(-15)
        self.set_font('Arial', 'I', 8)
        self.cell(0, 10, f'Page {self.page_no()}', 0, 0, 'C')
    def chapter_title(self, title):
        self.set_font('Arial', 'B', 12)
        self.cell(0, 10, title, 0, 1, 'L')
        self.ln(10)
    def chapter_body(self, body):
        self.set_font('Arial', '', 12)
        self.multi_cell(0, 10, body)
        self.ln()
def create_pdf(filename):
    pdf = PDF()
    pdf.add_page()
    pdf.chapter_title('Introduction')
    pdf.chapter_body('This is a sample PDF generated using FPDF library.')
    pdf.output(filename)
create_pdf('output.pdf')

在这个示例中，我们创建了一个PDF文件，并在页面上添加了标题和正文。

三、WEASYPRINT库

WeasyPrint是一个现代的HTML/CSS到PDF转换库，它允许用户使用HTML和CSS来生成PDF文件，非常适合需要生成复杂布局的PDF文件的场景。

安装WeasyPrint

首先，您需要安装WeasyPrint库：

pip install weasyprint

生成基本PDF文件

以下是使用WeasyPrint生成一个简单PDF文件的示例：

from weasyprint import HTML
def create_pdf(filename):
    html = """
    <html>
    <head>
        <style>
            body { font-family: Arial, sans-serif; }
            h1 { color: #333; }
        </style>
    </head>
    <body>
        <h1>Hello, World!</h1>
        <p>This is a sample PDF generated using WeasyPrint library.</p>
    </body>
    </html>
    """
    HTML(string=html).write_pdf(filename)
create_pdf('output.pdf')

在这个示例中，我们使用HTML和CSS定义了PDF的内容和样式，并生成了一个PDF文件。

四、PYPDF2库

PyPDF2是一个用于操作PDF文件的库，虽然它不能直接生成PDF文件，但可以用于合并、拆分、加密和解密PDF文件等操作。

安装PyPDF2

首先，您需要安装PyPDF2库：

pip install PyPDF2

合并PDF文件

以下是使用PyPDF2合并两个PDF文件的示例：

import PyPDF2
def merge_pdfs(input_files, output_file):
    pdf_writer = PyPDF2.PdfFileWriter()
    for file in input_files:
        pdf_reader = PyPDF2.PdfFileReader(file)
        for page_num in range(pdf_reader.getNumPages()):
            page = pdf_reader.getPage(page_num)
            pdf_writer.addPage(page)
    with open(output_file, 'wb') as out:
        pdf_writer.write(out)
input_files = ['file1.pdf', 'file2.pdf']
merge_pdfs(input_files, 'merged.pdf')