python之reportlab生成PDF文件
2024-08-31 19:20:06
项目需要，需要自动生成PDF测试报告。经过对比之后，选择使用了reportlab模块。
项目背景：开发一个测试平台，供测试维护测试用例，执行测试用例，并且生成测试报告（包含PDF和excel），将生成的测试报告以邮件的形式发送相关人。
reportlab生成PDF文件的代码如下：
from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Image, PageBreak, Table, TableStyle
from reportlab.lib.styles import ParagraphStyle, getSampleStyleSheet
from reportlab.lib.units import mm
from reportlab.lib import colors
from reportlab.lib.enums import TA_CENTER, TA_LEFT
from reportlab.pdfbase import pdfmetrics
from reportlab.pdfbase.ttfonts import TTFont
pdfmetrics.registerFont(TTFont('pingbold', 'PingBold.ttf'))
pdfmetrics.registerFont(TTFont('ping', 'ping.ttf'))
pdfmetrics.registerFont(TTFont('hv', 'Helvetica.ttf')) # 生成PDF文件
class PDFGenerator:
def __init__(self, filename):
self.filename = filename
self.file_path = '/xxx/xxx/xxx/xxx/'
self.title_style = ParagraphStyle(name="TitleStyle", fontName="pingbold", fontSize=48, alignment=TA_LEFT,)
self.sub_title_style = ParagraphStyle(name="SubTitleStyle", fontName="hv", fontSize=32,
textColor=colors.HexColor(0x666666), alignment=TA_LEFT, )
self.content_style = ParagraphStyle(name="ContentStyle", fontName="ping", fontSize=18, leading=25, spaceAfter=20,
underlineWidth=1, alignment=TA_LEFT, )
self.foot_style = ParagraphStyle(name="FootStyle", fontName="ping", fontSize=14, textColor=colors.HexColor(0xB4B4B4),
leading=25, spaceAfter=20, alignment=TA_CENTER, )
self.table_title_style = ParagraphStyle(name="TableTitleStyle", fontName="pingbold", fontSize=20, leading=25,
spaceAfter=10, alignment=TA_LEFT, )
self.sub_table_style = ParagraphStyle(name="SubTableTitleStyle", fontName="ping", fontSize=16, leading=25,
spaceAfter=10, alignment=TA_LEFT, )
self.basic_style = TableStyle([('FONTNAME', (0, 0), (-1, -1), 'ping'),
('FONTSIZE', (0, 0), (-1, -1), 12),
('ALIGN', (0, 0), (-1, -1), 'LEFT'),
('VALIGN', (0, 0), (-1, -1), 'MIDDLE'),
('BOTTOMPADDING', (0, 0), (-1, -1), 6),
# 'SPAN' (列,行)坐标
('SPAN', (1, 0), (3, 0)),
('SPAN', (1, 1), (3, 1)),
('SPAN', (1, 2), (3, 2)),
('SPAN', (1, 5), (3, 5)),
('SPAN', (1, 6), (3, 6)),
('SPAN', (1, 7), (3, 7)),
('GRID', (0, 0), (-1, -1), 0.5, colors.black),
])
self.common_style = TableStyle([('FONTNAME', (0, 0), (-1, -1), 'ping'),
('FONTSIZE', (0, 0), (-1, -1), 12),
('ALIGN', (0, 0), (-1, -1), 'LEFT'),
('VALIGN', (0, 0), (-1, -1), 'MIDDLE'),
('BOTTOMPADDING', (0, 0), (-1, -1), 6),
('GRID', (0, 0), (-1, -1), 0.5, colors.black),
]) def genTaskPDF(self, home_data, task_data, basic_data, case_set_data, fail_case_data, p0_case_data):
story = [] # 首页内容
story.append(Spacer(1, 20 * mm))
img = Image('/xxx/xxx.png')
img.drawHeight = 20 * mm
img.drawWidth = 40 * mm
img.hAlign = TA_LEFT
story.append(img)
story.append(Spacer(1, 10 * mm))
story.append(Paragraph("测试报告", self.title_style))
story.append(Spacer(1, 20 * mm))
story.append(Paragraph("Test Report of XXX", self.sub_title_style))
story.append(Spacer(1, 45 * mm))
story.append(Paragraph("报告编号：" + home_data['report_code'], self.content_style))
story.append(Paragraph("计划名称：" + home_data['task_name'], self.content_style))
story.append(Paragraph("报告日期：" + home_data['report_date'], self.content_style))
story.append(Paragraph(" 负责人：" + home_data['report_creator'], self.content_style))
story.append(Spacer(1, 55 * mm))
story.append(Paragraph("内部文档，请勿外传", self.foot_style))
story.append(PageBreak()) # 表格允许单元格内容自动换行格式设置
stylesheet = getSampleStyleSheet()
body_style = stylesheet["BodyText"]
body_style.wordWrap = 'CJK'
body_style.fontName = 'ping'
body_style.fontSize = 12 # 测试计划
story.append(Paragraph("测试计划", self.table_title_style))
story.append(Spacer(1, 3 * mm))
task_table = Table(task_data, colWidths=[25 * mm, 141 * mm], rowHeights=12 * mm, style=self.common_style)
story.append(task_table) story.append(Spacer(1, 10 * mm)) # 基础参数
story.append(Paragraph("基础参数", self.sub_table_style))
basic_table = Table(basic_data, colWidths=[25*mm, 61*mm, 25*mm, 55*mm], rowHeights=12 * mm, style=self.basic_style)
story.append(basic_table) story.append(Spacer(1, 10 * mm)) # 测试用例集
story.append(Paragraph("用例集参数", self.sub_table_style))
case_set_table = Table(case_set_data, colWidths=[25 * mm, 141 * mm], rowHeights=12 * mm, style=self.common_style)
story.append(case_set_table) # story.append(PageBreak())
story.append(Spacer(1, 15 * mm)) # 失败用例--使用可以自动换行的方式需要data里都是str类型的才OK
story.append(Paragraph("失败用例", self.table_title_style))
story.append(Spacer(1, 3 * mm))
para_fail_case_data = [[Paragraph(cell, body_style) for cell in row] for row in fail_case_data]
fail_case_table = Table(para_fail_case_data, colWidths=[20 * mm, 35 * mm, 91 * mm, 20 * mm])
fail_case_table.setStyle(self.common_style)
story.append(fail_case_table) story.append(Spacer(1, 15 * mm)) # 基础用例（P0）
story.append(Paragraph("基础用例（P0）", self.table_title_style))
story.append(Spacer(1, 3 * mm))
para_p0_case_data = [[Paragraph(cell, body_style) for cell in row] for row in p0_case_data]
p0_case_table = Table(para_p0_case_data, colWidths=[20 * mm, 35 * mm, 91 * mm, 20 * mm])
p0_case_table.setStyle(self.common_style)
story.append(p0_case_table) doc = SimpleDocTemplate(self.file_path + self.filename + ".pdf",
leftMargin=20 * mm, rightMargin=20 * mm, topMargin=20 * mm, bottomMargin=20 * mm) doc.build(story)
生成样式：
需要说明的：
1.项目后台使用的是python2.7.5；
2.如果需要使用到特定的字体，需要将字体下载下来放到reportlab模块安装位置的font文件夹下，如：/xxx/python2.7/site-packages/reportlab/fonts/；
3.有官方的使用文档可以参考，如有不懂，可以下载官方说明文档查看；
4.支持表格中的中文依文字长度自动换行
代码中使用到的字体下载链接：
链接: https://pan.baidu.com/s/10p7YxZ0Z32d5z85O8puabg 提取码: sk8q
最新文章
- nginx源码分析之模块初始化
- Xcode 8 支持 iOS 7 真机解决过程记录
- .NET重构—单元测试的代码重构
- 二分查找C++
- visual studio 2005 编fortran程序，运行后dos窗口显示问题
- jQuery 显示加载更多(节流) 实现预加载
- smarty对网页性能的影响--开启opcache
- video详解 HTML5中的视频：
- [置顶] java 枚举
- Hibernate一对一主键映射
- java学习笔记 --- 面向对象
- javaWeb中request请求转发和response重定向
- 说一下Dubbo 的工作原理？注册中心挂了可以继续通信吗？
- 二叉查找树（BST)、平衡二叉树(AVL树)（只有插入说明）
- Matplotlib 基本用法
- cocos2d JS 错误异常抛出捕获和崩溃拦截
- DNS隧道 iodns
- PHPCMS增加SEO字段调用
- mysql的check约束问题
- RabbitMQ学习（一）：RabbitMQ要点简介