Python - 开源库 ReportLab 库合并 CVS 和图像生成 PDF 文档

欢迎关注我的CSDN：https://spike.blog.csdn.net/
本文地址：https://spike.blog.csdn.net/article/details/140281680

免责声明：本文来源于个人知识与公开资料，仅用于学术交流，欢迎讨论，不支持转载。

ReportLab：https://docs.reportlab.com/

ReportLab 是一个非常流行的 Python 库，专门用于生成 PDF 文档，提供了丰富的功能，可以创建复杂的文档，包括文本、图像、表格、图表等。即：

• PDF 支持：ReportLab 专注于生成 PDF 文件，支持 PDF 的各种特性，如字体、颜色、图层等。
• 可扩展性：可以通过编写 Python 代码来扩展 ReportLab 的功能，满足更复杂的需求。
• 图形和图像：支持在 PDF 中嵌入图像，包括 JPEG、PNG 等格式。同时，ReportLab 也提供了绘制图形的功能。
• 表格：ReportLab 提供了强大的表格支持，可以创建复杂的表格布局，并支持表格的格式化和样式设置。
• 文本处理：支持多种字体和文本样式，包括粗体、斜体、下划线等。还可以调整文本的对齐方式和行间距。
• 图表：可以生成各种图表，如条形图、饼图、折线图等，支持图表的自定义样式和数据源。
• 自动化文档生成：可以自动化生成文档，如报告、发票、证书等，适合批量生成文档的场景。
• 跨平台：ReportLab 可以在多种操作系统上运行，包括 Windows、Linux 和 macOS。
• 开源：ReportLab 是一个开源项目，可以在遵守其许可证的情况下自由使用和修改。

示例函数 create_pdf 用于 构建 PDF 文件，输入 csv 文件列表和 image 文件列表：

• csv_files 是 csv 文件路径列表；
• image_files 是 image 文件路径列表；
• headline 是 pdf 文件标题；
• output_pdf 是 输出 的 PDF 路径；

源码如下：

import csv
import os.pathfrom reportlab.lib import colors
from reportlab.lib.pagesizes import letter
from reportlab.lib.styles import getSampleStyleSheet
from reportlab.platypus import SimpleDocTemplate, Table, TableStyle, Image, Spacer, Paragraphfrom root_dir import DATA_DIR
from utils.project_utils import traverse_dir_filesdef create_pdf(csv_files, image_files, headline, output_pdf):"""Merge the data from each CSV file into a single PDF file."""# 创建 PDF 文档doc = SimpleDocTemplate(output_pdf, pagesize=letter)elements = []line1 = headlinestyles = getSampleStyleSheet()style_normal = styles['Heading1']elements.append(Paragraph(line1, style_normal))# 读取并添加每个 CSV 文件的数据for csv_file in csv_files:data = []with open(csv_file, 'r', newline='') as file:reader = csv.reader(file)headers = next(reader)  # 读取标题行data += [headers]data += [row for row in reader]  # 读取数据行# 创建表格table = Table(data)table.setStyle(TableStyle([('BACKGROUND', (0, 0), (-1, 0), colors.white),('TEXTCOLOR', (0, 0), (-1, 0), colors.black),('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)elements.append(Spacer(1, 20))for image_file in image_files:img = Image(image_file)# img.hAlign = 'CENTER'# img.vAlign = 'TOP'ratio = img.imageWidth / 400img.drawHeight = img.imageHeight / ratioimg.drawWidth = img.imageWidth / ratio# img.scaleToFit(200, 200)  # 调整图像大小以适应页面elements.append(img)# 构建文档doc.build(elements)def main():input_csv_path = os.path.join(DATA_DIR, "abag-summary.csv")input_img_path = os.path.join(DATA_DIR, "plots")path_list = traverse_dir_files(input_img_path, ext="png")output_path = os.path.join(DATA_DIR, "output.pdf")create_pdf([input_csv_path], path_list, output_path)if __name__ == '__main__':main()