1.1. 前言
家喻户晓,安服工程师又叫做Word工程师,在打工或者批量SRC的时候,如果产出很多,又须要一个一个的写报告的状况下会十分的折磨人,因而查了一些相干的材料,发现应用python的docxtpl
库批量写报告成果很不错,记录一下。
1.2. 介绍
docxtpl
是一个用于生成 Microsoft Word 文档的模板引擎库,它联合了 docx 模块和 Jinja2 模板引擎,使用户可能应用 Microsoft Word 模板文件并在其中填充动态数据。它提供了一种不便的形式来生成个性化的 Word 文档,并反对条件语句、循环语句和变量等控制结构,以满足不同的文档生成需要。
官网GitHub地址:https://github.com/elapouya/python-docx-template
官网文档地址:https://docxtpl.readthedocs.io/en/latest/
简略来说:就是创立一个相似Jinja2语法的模板文档,而后往里面动静填充内容就能够了
装置:
pip3 install docxtpl
1.3. 根底应用
from docxtpl import DocxTemplatedoc = DocxTemplate("test.docx")context = {'whoami': "d4m1ts"}doc.render(context)doc.save("generated_doc.docx")
其中,test.docx内容如下:
生成后的后果如下:
1.4. 案例介绍
1.4.1. 需要假如
写一份不思考好看的漏扫报告,须要有统计后果图和破绽详情,每个破绽包含破绽名、破绽地址、破绽等级、破绽危害、复现过程、修复倡议六个局部。
1.4.2. 模板文档筹备
编写的模板文档如下,应用到了常见的if
、for
、赋值
等,保留为template.docx
,后续只须要向外面填充数据即可。
1.4.3. 数据结构剖析
传入数据须要一串json字符串,因而咱们依据模板文档梳理好json构造,而后传入即可。
梳理好的数据结构如下:
{ "饼图": "111", "柱状图": "222", "破绽简报": [ { "破绽名": "测试破绽名1", "破绽等级": "高危" } ], "破绽详情": [ { "破绽名": "测试破绽名1", "破绽地址": "http://blog.gm7.org/", "破绽等级": "高危", "破绽危害": "危害XXX", "复现过程": "先xxx,再xxx,最初xxx", "修复倡议": "更新到最新版本即可" } ]}
编写代码测试一下可行性:
from docxtpl import DocxTemplatedoc = DocxTemplate("template.docx")context = { "饼图": "111", "柱状图": "222", "破绽简报": [ { "破绽名": "测试破绽名1", "破绽等级": "高危" }, { "破绽名": "测试破绽名2", "破绽等级": "重大" }, { "破绽名": "测试破绽名2", "破绽等级": "中危" } ], "破绽详情": [ { "破绽名": "测试破绽名1", "破绽地址": "http://blog.gm7.org/", "破绽等级": "高危", "破绽危害": "危害XXX", "复现过程": "先xxx,再xxx,最初xxx", "修复倡议": "更新到最新版本即可" }, { "破绽名": "测试破绽名2", "破绽地址": "http://bblog.gm7.org/", "破绽等级": "重大", "破绽危害": "危害XXX", "复现过程": "先xxx,再xxx,最初xxx", "修复倡议": "更新到最新版本即可" }, { "破绽名": "测试破绽名3", "破绽地址": "http://cblog.gm7.org/", "破绽等级": "中危", "破绽危害": "危害XXX", "复现过程": "先xxx,再xxx,最初xxx", "修复倡议": "更新到最新版本即可" } ] }doc.render(context)doc.save("generated_doc.docx")
很好,达到了预期的成果。
1.4.4. 退出图表
在下面的过程中,内容简直是没问题了,然而图表还是没有展现进去。生成图表咱们应用plotly
这个库,并将生成内容写入ByteIO
。
相干代码如下:
import plotly.graph_objects as gofrom io import BytesIOdef generatePieChart(title: str, labels: list, values: list, colors: list): """ 生成饼图 https://juejin.cn/post/6911701157647745031#heading-3 https://juejin.cn/post/6950460207860449317#heading-5 :param title: 饼图题目 :param labels: 饼图标签 :param values: 饼图数据 :param colors: 饼图每块的色彩 :return: """ # 根底饼图 fig = go.Figure(data=[go.Pie( labels=labels, values=values, hole=.4, # 核心环大小 insidetextorientation="horizontal" )]) # 更新色彩 fig.update_traces( textposition='inside', # 文本显示地位 hoverinfo='label+percent', # 悬停信息 textinfo='label+percent', # 饼图中显示的信息 textfont_size=15, marker=dict(colors=colors) ) # 更新题目 fig.update_layout( title={ # 设置整个题目的名称和地位 "text": title, "y": 0.96, # y轴数值 "x": 0.5, # x轴数值 "xanchor": "center", # x、y轴绝对地位 "yanchor": "top" } ) image_io = BytesIO() fig.write_image(image_io, format="png") return image_iodef generateBarChart(title: str, x: list, y: list): """ 生成柱状图 https://cloud.tencent.com/developer/article/1817208 https://blog.csdn.net/qq_25443541/article/details/115999537 https://blog.csdn.net/weixin_45826022/article/details/122912484 :param title: 题目 :param x: 柱状图标签 :param y: 柱状图数据 :return: """ # x轴长度最为18 b = x x = [] for i in b: if len(i) >= 18: x.append(f"{i[:15]}...") else: x.append(i) # 根底柱状图 fig = go.Figure(data=[go.Bar( x=x, y=y, text=y, textposition="outside", marker=dict(color=["#3498DB"] * len(y)), width=0.3 )]) # 更新题目 fig.update_layout( title={ # 设置整个题目的名称和地位 "text": title, "y": 0.96, # y轴数值 "x": 0.5, # x轴数值 "xanchor": "center", # x、y轴绝对地位 "yanchor": "top" }, xaxis_tickangle=-45, # 歪斜45度 plot_bgcolor='rgba(0,0,0,0)' # 背景通明 ) fig.update_xaxes( showgrid=False ) fig.update_yaxes( zeroline=True, zerolinecolor="#17202A", zerolinewidth=1, showgrid=True, gridcolor="#17202A", showline=True ) image_io = BytesIO() fig.write_image(image_io, format="png") return image_io
1.4.5. 最终后果
要插入图片内容,代码语法如下:
myimage = InlineImage(tpl, image_descriptor='test_files/python_logo.png', width=Mm(20), height=Mm(10))
残缺代码如下:
from docxtpl import DocxTemplate, InlineImagefrom docx.shared import Mmimport plotly.graph_objects as gofrom io import BytesIOdef generatePieChart(title: str, labels: list, values: list, colors: list): """ 生成饼图 https://juejin.cn/post/6911701157647745031#heading-3 https://juejin.cn/post/6950460207860449317#heading-5 :param title: 饼图题目 :param labels: 饼图标签 :param values: 饼图数据 :param colors: 饼图每块的色彩 :return: """ # 根底饼图 fig = go.Figure(data=[go.Pie( labels=labels, values=values, hole=.4, # 核心环大小 insidetextorientation="horizontal" )]) # 更新色彩 fig.update_traces( textposition='inside', # 文本显示地位 hoverinfo='label+percent', # 悬停信息 textinfo='label+percent', # 饼图中显示的信息 textfont_size=15, marker=dict(colors=colors) ) # 更新题目 fig.update_layout( title={ # 设置整个题目的名称和地位 "text": title, "y": 0.96, # y轴数值 "x": 0.5, # x轴数值 "xanchor": "center", # x、y轴绝对地位 "yanchor": "top" } ) image_io = BytesIO() fig.write_image(image_io, format="png") return image_iodef generateBarChart(title: str, x: list, y: list): """ 生成柱状图 https://cloud.tencent.com/developer/article/1817208 https://blog.csdn.net/qq_25443541/article/details/115999537 https://blog.csdn.net/weixin_45826022/article/details/122912484 :param title: 题目 :param x: 柱状图标签 :param y: 柱状图数据 :return: """ # x轴长度最为18 b = x x = [] for i in b: if len(i) >= 18: x.append(f"{i[:15]}...") else: x.append(i) # 根底柱状图 fig = go.Figure(data=[go.Bar( x=x, y=y, text=y, textposition="outside", marker=dict(color=["#3498DB"] * len(y)), width=0.3 )]) # 更新题目 fig.update_layout( title={ # 设置整个题目的名称和地位 "text": title, "y": 0.96, # y轴数值 "x": 0.5, # x轴数值 "xanchor": "center", # x、y轴绝对地位 "yanchor": "top" }, xaxis_tickangle=-45, # 歪斜45度 plot_bgcolor='rgba(0,0,0,0)' # 背景通明 ) fig.update_xaxes( showgrid=False ) fig.update_yaxes( zeroline=True, zerolinecolor="#17202A", zerolinewidth=1, showgrid=True, gridcolor="#17202A", showline=True ) image_io = BytesIO() fig.write_image(image_io, format="png") return image_iodoc = DocxTemplate("template.docx")context = { "饼图": InlineImage(doc, image_descriptor=generatePieChart( title="破绽数量", labels=["重大", "高危", "中危", "低危"], values=[1, 1, 1, 0], colors=["#8B0000", "red", "orange", "aqua"] ), width=Mm(130)), "柱状图": InlineImage(doc, image_descriptor=generateBarChart( title="破绽类型", x=["测试破绽名1", "测试破绽名2", "测试破绽名3"], y=[1, 1, 1] ), width=Mm(130)), "破绽简报": [ { "破绽名": "测试破绽名1", "破绽等级": "高危" }, { "破绽名": "测试破绽名2", "破绽等级": "重大" }, { "破绽名": "测试破绽名2", "破绽等级": "中危" } ], "破绽详情": [ { "破绽名": "测试破绽名1", "破绽地址": "http://blog.gm7.org/", "破绽等级": "高危", "破绽危害": "危害XXX", "复现过程": "先xxx,再xxx,最初xxx", "修复倡议": "更新到最新版本即可" }, { "破绽名": "测试破绽名2", "破绽地址": "http://bblog.gm7.org/", "破绽等级": "重大", "破绽危害": "危害XXX", "复现过程": "先xxx,再xxx,最初xxx", "修复倡议": "更新到最新版本即可" }, { "破绽名": "测试破绽名3", "破绽地址": "http://cblog.gm7.org/", "破绽等级": "中危", "破绽危害": "危害XXX", "复现过程": "先xxx,再xxx,最初xxx", "修复倡议": "更新到最新版本即可" } ]}doc.render(context)doc.save("generated_doc.docx")
后果如下: