关键字： word go语言 办公软件 自动化 excel office
背景
近来因为工作需要，经常性要重复地写报告，搞安全的就是苦逼啊。每次都需要人工查这查那，还要复制黏贴。 重复地复制黏贴会比较烦人。所以弄好一份表就自动让机器代替人工黏贴到word文档，自动生成会相对舒服且快速，有什么问题直接修改excel即可。
ps：因为苦于unioffice库的问题，所以重新找了两个库来实现，发现生成后的文件小了很多，而且文档及库的使用极度方便。
步骤
我是用go语言进行开发的。
go的版本：go version go1.14.4 windows/amd64
github.com/360EntSecGroup-Skylar/excelize/v2
github.com/nguyenthenguyen/docx
命令进行安装：
go get -v github.com/xxx
ps：
这两个库相对简单，excel的手册地址：https://xuri.me/excelize/zh-hans/workbook.html#SetDocProps
docx就暂时没有手册，但是提供的函数不多。研究一下即可。
注意两个函数：ReplaceRaw和Replace ，这两个的区别是ReplaceRaw直接动docx底层文件xml；Replace只是把你想要替换的内容给替换。
确定流程
其实最好先确定一下程序的处理流程和具体人工处理的步骤，因为偷懒我这里就不提供了，因为我也是画着画着就没有弄了。
该项目的步骤：
1、先填写好一个excel表，作为程序的数据输入。
2、打开模板docx，程序处理，输出修改后的报告。
3、输出写好的数据到一个汇总的表进行统计和闭环。
编写伪代码
在vscode上编写具体的流程伪代码
编写正式的代码
按照自己的思路，确定好变量、常量、具体的处理流程等等，按照伪代码进行一步一步的代码编写。
最终的代码
package main
import (
"log"
"os"
"strconv"
"strings"
"time"
"github.com/360EntSecGroup-Skylar/excelize/v2"
"github.com/nguyenthenguyen/docx"
)
type biaozhi struct {
yeMeiTime string // 页眉的时间编号
neiRouTime string // 内容的时间格式
sjmc string // 事件名称
sjjb string // 事件级别（漏洞、风险、应急）
ipaddr string // 告警对象（应急）
gjlj []string // 告警路径（应急、风险）
tms string // 告警数量（应急）
gjxx string // 告警描述（漏洞、风险）
czjy []string // 处置建议(漏洞、风险)
sjly string // 事件类型(应急)
llmc string // 漏洞名称(漏洞)
yxbb []string // 影响版本(漏洞)
ywxt string // 业务系统（应急、风险）
ywdw string // 业务单位（漏洞、应急、风险）
fxms string // 风险验证描述（风险）
kyfx string // 可疑分析（可疑）
czjyStr string // 写入excel的处置建立
lujingStr string // 写入excel中的路径
lcw string // 另存为文件的名称
}
var (
// Baogao 要打开的报告类型
Baogao string
// SaveFile 要保存的报告类型
SaveFile string
dataList biaozhi
// qiantou这个比较重要。
/*
我不知道什么原因，这两个参数我已经第二次修改了，可能是因为docx的问题，所以建议如果你要动程序之前，最好先自己测试一下docx的这两个参数对不对。直接解压docx看就可以了。
*/
qiantou string = `<w:p><w:pPr><w:rPr><w:rFonts w:hint="eastAsia" w:ascii="仿宋_GB2312" w:hAnsi="仿宋_GB2312" w:eastAsia="仿宋_GB2312" w:cs="仿宋_GB2312"/><w:sz w:val="24"/><w:szCs w:val="24"/><w:lang w:val="en-US" w:eastAsia="zh-CN"/></w:rPr></w:pPr><w:r><w:rPr><w:rFonts w:hint="eastAsia" w:ascii="仿宋_GB2312" w:hAnsi="仿宋_GB2312" w:eastAsia="仿宋_GB2312" w:cs="仿宋_GB2312"/><w:sz w:val="24"/><w:szCs w:val="24"/><w:lang w:val="en-US" w:eastAsia="zh-CN"/></w:rPr><w:t>`
jiewei string = `</w:t></w:r><w:bookmarkStart w:id="0" w:name="_GoBack"/><w:bookmarkEnd w:id="0"/></w:p>`
)
// 1.判断输出的是什么通告
func tonggao() {
cl, err := excelize.OpenFile("素材.xlsx")
if err != nil {
log.Println("打开素材文件错误！")
return
}
SaveFile, _ = cl.GetCellValue("需要填写的资料", "B1")
switch SaveFile {
case "可疑行为":
Baogao = "安全可疑行为通告模板.docx"
case "风险预警":
Baogao = "安全风险预警通告模板.docx"
case "应急预警":
Baogao = "安全事件应急通告模板.docx"
case "漏洞预警":
Baogao = "安全漏洞预警通告模板.docx"
default:
log.Println("请检查(素材.xlsx)文件（需要填写的资料）表中报告类型是否正确！")
}
// 获取相关的数据
dataList.事件名称, _ = cl.GetCellValue("需要填写的资料", "B2") // 风险预警：获取事件名称
dataList.sjjb, _ = cl.GetCellValue("需要填写的资料", "B3") // 所有预警：获取事件级别
dataList.ipaddr, _ = cl.GetCellValue("需要填写的资料", "B4") // 应急预警：告警对象
gjlj, _ := cl.GetCellValue("需要填写的资料", "B5")
dataList.gjlj = strings.Split(gjlj, "\n") // 告警路径
dataList.tms, _ = cl.GetCellValue("需要填写的资料", "B6") // 告警数量
dataList.gjxx, _ = cl.GetCellValue("需要填写的资料", "B7") // 告警描述
dataList.czjyStr, _ = cl.GetCellValue("需要填写的资料", "B8")
dataList.czjy = strings.Split(dataList.czjyStr, "\n") // 所有预警：获取处置建议
dataList.sjly, _ = cl.GetCellValue("需要填写的资料", "B9") // 应急预警：事件类型
dataList.llmc, _ = cl.GetCellValue("需要填写的资料", "B10") // 漏洞预警：漏洞名称
yxbb, _ := cl.GetCellValue("需要填写的资料", "B11")
dataList.yxbb = strings.Split(yxbb, "\n") // 漏洞预警：影响版本
dataList.ywxt, _ = cl.GetCellValue("需要填写的资料", "B12") // 风险预警：业务系统名称
dataList.ywdw, _ = cl.GetCellValue("需要填写的资料", "B13") // 风险预警：业务单位名称
dataList.fxms, _ = cl.GetCellValue("需要填写的资料", "B14") // 风险预警：风险验证描述
dataList.kyfx, _ = cl.GetCellValue("需要填写的资料", "B15") // 可疑行为：可疑分析
}
// 2.处理并输出报告
func baogao() {
// 打开一个已有格式的文档，这个是要打开的文档路径。
filesDir := "./template/" + Baogao
// 添加页眉
r, err := docx.ReadDocxFile(filesDir)
if err != nil {
log.Printf("打开word文档错误，错误信息: %s", err)
}
log.Println("打开报告完成")
defer r.Close()
docx1 := r.Editable()
docx1.ReplaceHeader("页眉日期", dataList.yeMeiTime)
switch SaveFile {
case "应急预警":
log.Println("当前正在处理应急预警类型的报告")
docx1.Replace("事件名称：", "事件名称："+dataList.sjmc, -1)
docx1.Replace("事件类型：", "事件类型："+dataList.sjly, -1)
docx1.Replace("事件级别：", "事件级别："+dataList.sjjb, -1)
docx1.Replace("告警对象：", "告警对象："+dataList.ipaddr, -1)
docx1.Replace("所属系统：", "所属系统："+dataList.ywxt+"系统,隶属于"+dataList.ywdw, -1)
docx1.Replace("告警内容：", "告警内容："+dataList.gjxx, -1)
docx1.Replace("应急描述", dataList.neiRouTime+"，发现业务(IP："+dataList.ipaddr+")产生"+dataList.tms+"条“"+dataList.sjmc+"”日志告警信息。经云主机资产确认，该主机为“"+dataList.ywxt+"系统,隶属于"+dataList.ywdw+"。", -1)
// 告警路径切片写入
var shuZu []string
for _, t := range dataList.gjlj {
t = strings.Replace(t, "<", "<", -1)
t = strings.Replace(t, ">", ">", -1)
shuZu = append(shuZu, t+jiewei+qiantou)
}
shuZu = append(shuZu, jiewei)
st1 := strings.Join(shuZu, "")
docx1.ReplaceRaw("告警路径", st1, -1)
// 处置建议切片写入
var shuZu2 []string
for _, t := range dataList.czjy {
// 加了转换验证解决特殊符号导致的错误问题。
t = strings.Replace(t, "<", "<", -1)
t = strings.Replace(t, ">", ">", -1)
shuZu2 = append(shuZu2, t+jiewei+qiantou)
}
shuZu2 = append(shuZu, jiewei)
st2 := strings.Join(shuZu2, "") //把切片装回string类型的字符串
docx1.ReplaceRaw("应急处置建议", st2, -1)
case "漏洞预警":
log.Println("当前正在处理漏洞预警类型的报告")
docx1.Replace("漏洞名称填写", dataList.llmc, -1)
docx1.Replace("风险等级填写", dataList.sjjb, -1)
docx1.Replace("风险描述填写", dataList.gjxx, -1)
// 影响版本切片写入
var shuZu []string
for _, t := range dataList.yxbb {
t = strings.Replace(t, "<", "<", -1)
t = strings.Replace(t, ">", ">", -1)
shuZu = append(shuZu, t+jiewei+qiantou)
//shuZu = append(shuZu, t)
}
shuZu = append(shuZu, jiewei)
st1 := strings.Join(shuZu, "")
// ReplaceRaw该函数是直接修改底层的xml文件。而Replace只是单纯的替换你想替换的内容，不修改格式。
docx1.ReplaceRaw("影响版本填写", st1, 1)
// 处置建议切片写入
var shuZu2 []string
for _, t := range dataList.czjy {
t = strings.Replace(t, "<", "<", -1)
t = strings.Replace(t, ">", ">", -1)
shuZu2 = append(shuZu2, t+jiewei+qiantou)
}
shuZu2 = append(shuZu, jiewei)
st2 := strings.Join(shuZu2, "")
docx1.ReplaceRaw("处置建议填写", st2, -1)
case "可疑行为":
log.Println("当前正在处理可疑行为类型的报告")
docx1.Replace("行为概述填写", dataList.neiRouTime+",资产："+dataList.ipaddr+"产生"+dataList.tms+"条"+dataList.sjmc+"日志告警信息。经云主机资产确认，该主机为“"+dataList.ywxt+"”系统，隶属于“"+dataList.ywdw+"”。", -1)
docx1.Replace("疑点分析填写", dataList.kyfx, -1)
// 处置建议切片写入
var shuZu []string
for _, t := range dataList.czjy {
t = strings.Replace(t, "<", "<", -1)
t = strings.Replace(t, ">", ">", -1)
shuZu = append(shuZu, t+jiewei+qiantou)
}
shuZu = append(shuZu, jiewei)
st2 := strings.Join(shuZu, "")
docx1.ReplaceRaw("处置建议填写", st2, -1)
default:
log.Println("正在处理默认的表格内容")
docx1.Replace("事件名称填写", dataList.sjmc, -1)
docx1.Replace("风险等级填写", dataList.sjjb, -1)
docx1.Replace("风险描述填写", dataList.gjxx, -1)
docx1.Replace("业务所属", "经云主机资产确认，该主机为"+dataList.ywxt+"系统,隶属于"+dataList.ywdw, -1)
docx1.Replace("风险验证描述", dataList.fxms, -1)
// 风险验证路径切片写入
var shuZu []string
for _, t := range dataList.gjlj {
t = strings.Replace(t, "<", "<", -1)
t = strings.Replace(t, ">", ">", -1)
shuZu = append(shuZu, t+jiewei+qiantou)
}
shuZu = append(shuZu, jiewei)
st1 := strings.Join(shuZu, "")
docx1.ReplaceRaw("风险验证路径", st1, -1)
// 处置建议切片写入
var shuZu2 []string
for _, t := range dataList.czjy {
t = strings.Replace(t, "<", "<", -1)
t = strings.Replace(t, ">", ">", -1)
shuZu2 = append(shuZu2, t+jiewei+qiantou)
}
shuZu2 = append(shuZu, jiewei)
st2 := strings.Join(shuZu2, "")
docx1.ReplaceRaw("处置建议填写", st2, -1)
}
switch SaveFile {
case "可疑行为":
dataList.lcw = "【" + dataList.sjmc + "】" + "(" + dataList.ywdw + ")" + "安全可疑行为预警通告" + dataList.yeMeiTime + ".docx"
case "风险预警":
dataList.lcw = "【" + dataList.sjmc + "】" + "(" + dataList.ywdw + ")" + "安全风险预警通告" + dataList.yeMeiTime + ".docx"
case "应急预警":
dataList.lcw = "【" + dataList.sjmc + "】" + "(" + dataList.ywdw + ")" + "安全事件应急通告" + dataList.yeMeiTime + ".docx"
case "漏洞预警":
dataList.lcw = "【" + dataList.llmc + "】" + "(" + dataList.ywdw + ")" + "安全漏洞预警通告" + dataList.yeMeiTime + ".docx"
}
sp := "./report/" + dataList.lcw
err = docx1.WriteToFile(sp)
if err != nil {
os.Mkdir("./report", os.ModePerm)
log.Println("当前目录中没有report报告文件夹，现在创建文件夹成功！！！")
docx1.WriteToFile(sp)
log.Println("保存docx文件成功！！！")
}
log.Println("保存docx文件成功！！！")
}
func exc() {
f, err := excelize.OpenFile("预警通报汇总表.xlsx")
if err != nil {
log.Println(err.Error())
}
//告警函编号、业务单位、业务系统、告警状态、漏洞名称、告警类型、告警级别、告警对象、告警描述、应急处置方法、发生时间、结束时间、告警根治处置方法建议、安全处置联系人、支撑人员、修复方法描述、修复时间、修复验证描述、修复验证人、修复验证时间
// 非应急预警的格式
exStr := []string{dataList.yeMeiTime, dataList.ywdw, dataList.ywxt, "未修复", dataList.llmc, dataList.事件名称, dataList.sjjb, dataList.ipaddr, dataList.gjxx, "", dataList.neiRouTime, "", dataList.czjyStr, "", "xxx", "", "", "", dataList.ywdw, SaveFile}
// 检查A列一千行，判断是否有值，如果有值就不写。没有值就整行填写记录
for n1 := 1; n1 < 1000; n1++ {
s1 := strconv.Itoa(n1)
lie := "A" + s1
// 获取当前表，固定表格的值
cell, _ := f.GetCellValue("Sheet1", lie)
if cell == "" { // 判断是否为空
// 为空就提交单元格编号并跳出循环
f.SetSheetRow("Sheet1", lie, &exStr)
// celSave = lie
break
}
}
// 保存文件
f.Save()
log.Println("通告内容已经写入“预警通报汇总表.xlsx”表中！！！")
}
func main() {
// 获取符合我们格式的时间。
dataList.neiRouTime = time.Now().Format("2006年01月02日15时04分")
dataList.yeMeiTime = time.Now().Format("200601021504")
// 判断是什么报告类型，提取素材exels里面的内容
tonggao()
// 将获取到的数据写入word并保存文件
baogao()
// 将获取到的数据写入到汇总excel里面
exc()
// 等待时间查看日志
time.Sleep(time.Minute * 999)
}
以前的久问题
如果使用了unioffice库会出现以下问题：
1.页眉其实有排版和字体问题，可惜时间太短用了一天时间写了这个程序，因为是新手所以大神勿喷。
2.word读取的时候存在问题，需要docx清除所有格式才能如你所愿地识别出来字段，举个有问题的例子：
以下是word文档的两个段：
恶意告警
攻击事件经过分析还存在其他问题，1：端口继续对外开放；2：漏洞未被修复。
如果这个文档有自己的格式如：仿宋字体；加粗；四号等等。。。
那么识别就会出现问题，识别结果如下：
恶意
告警
攻击事件
经过
分析还存在其它问题，
1
：端口继续对外开放；2：漏洞未被修复。
那么识别出来的字段就会有问题，不能按照我们关键字的方式进行识别并替换。
如我想识别出恶意告警，作为关键字，再替换为我想要的写的内容。但是这样的情况下就无法做了。
3.excel读出数据之后，可以读取成切片，变成数组，通过unioffice这个库的AddBreak()函数进行换行，继续写入内容。示例代码如下：（目前已经不用该库了。直接动xml文件）
for _, t := range dataList.chuzhijianyi { //重复执行
r.AddText(t) //写入内容
r.AddBreak() //换行
}
新的注意点
针对以上的旧问题，已经不再是问题了，通过新的库已经解决。
注意的点
模版的问题
这里要说一下，docx是基于xml格式的文本文档；使用压缩软件打开docx文件：
这个就是docx的内容，看到下面还有页眉页尾的文件。打开后的内容，关注的是关键字的连续，如图：
硬件性能这个关键字，我们可以拿来作为替换；要注意的是，有可能这个关键字不一定是连续的。不连续的情况下，要重新删除内容，重新写，如图：
再用压缩软件检查，是否已经连续。不然后面会无法替换原文档内容。
插入的内容
因为文档是docx，所以要使用doc库里面的ReplaceRaw函数，该函数插入的内容是不经过转换的。
- 所以需要添加xml的前缀内容：<w:p><w:pPr><w:rPr><w:rFonts w:hint=“eastAsia” w:ascii=“仿宋_GB2312” w:hAnsi=“仿宋_GB2312” w:eastAsia=“仿宋_GB2312” w:cs=“仿宋_GB2312”/><w:sz w:val=“24”/><w:szCs w:val=“24”/><w:lang w:val=“en-US” w:eastAsia=“zh-CN”/></w:rPr></w:pPr><w:r><w:rPr><w:rFonts w:hint=“eastAsia” w:ascii=“仿宋_GB2312” w:hAnsi=“仿宋_GB2312” w:eastAsia=“仿宋_GB2312” w:cs=“仿宋_GB2312”/><w:sz w:val=“24”/><w:szCs w:val=“24”/><w:lang w:val=“en-US” w:eastAsia=“zh-CN”/></w:rPr><w:t>
- 后缀内容是：</w:t></w:r><w:bookmarkStart w:id=“0” w:name="_GoBack"/><w:bookmarkEnd w:id=“0”/></w:p>
中间内容就是你想替换的内容啦。详情可以看我的源码。挺简单的。
特殊符号
在实践中，发现xml并是无脑地插入文本，特别是遇到符号的时候，
如果无脑代入，会在打开docx的时候报错，实际上，就是因为符号引起。
举个例子：
原因如下：
更正一下图片中的内容：“大于号这四个字符组成的 & g t ;” 小于号是这四个字符组成的"& l t ;"
以下这个图是正常能打开的docx截图：
正常的docx的xml如下：
与什么空格并没有什么关系。
PS：这个问题，目前已经找到折衷办法，使用strings.Replace()函数，把大小于号都替换就可以了。
- 在 XML 中，有 5 个预定义的实体引用：
| 转义后的字符 | 需要被转义的字符 | 备注 |
|---|---|---|
| < | < | less than |
| > | > | greater than |
| & | & | ampersand |
| ' | ’ | apostrophe |
| " | " | quotation mark |
注释：在 XML 中，只有字符 “<” 和 “&” 确实是非法的。大于号是合法的，但是用实体引用来代替它是一个好习惯。
总结
1、检查好自己的模版文档。
2、检查好输入输出的excel文档。
3、调试好程序的输出。
4、可以使用日志告警作为程序的接入、数据库表作为资产查询的输出。看个人喜欢。
至于图片的插入，目前需要用到的是unioffice库，我还没有研究过。
最终成果：
扫一扫
1106
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
