阅读 352

Golang是如何操作excel的?

关键术语介绍

为了方便开源库的快速上手,我们先来了解 excel 中的几个关键术语,如下图所示,①为sheet,也就是表格中的页签;②为row,代表 excel 中的一行;③为cell,代表 excel 中的一个单元格。

正常情况下,创建一个表格的基本流程是打开 wps 点击新建,这时会默认创建一个 sheet,然后在该 sheet 中的第一行填写表头,接下来根据表头逐行填充内容,最后将文件另存为到硬盘的某个位置。这与 Golang 开源库创建 excel 的流程基本相同,下面演示一个极简表格的创建。

创建表格

创建表格前需要先引入 excel 库,我们以比较热门的 tealeg/xlsx 库为例。

go get github.com/tealeg/xlsx
复制代码

首先创建一个空文件,拿到文件句柄。

file := xlsx.NewFile()
复制代码

创建一个名为人员信息收集的 sheet。

sheet, err := file.AddSheet("人员信息收集")
if err != nil {
  panic(err.Error())
}
复制代码

然后为该 sheet 创建一行,这行作为我们的表头。

row := sheet.AddRow()
复制代码

在该行中创建一个单元格。

cell := row.AddCell()
复制代码

现在给单元格填充内容,因为是表头,暂且叫姓名

cell.Value = "姓名"
复制代码

如何创建第二个单元格呢?原理相同,此处 cell 变量已定义,再创建新单元格只需赋值即可。

cell = row.AddCell()
cell.Value = "性别"
复制代码

表头已经设置好了,可以开始创建第二行来填充内容了,方式与上述无差别。

row = sheet.AddRow()
cell = row.AddCell()
cell.Value = "张三"
cell = row.AddCell()
cell.Value = "男"
复制代码

表格设置完成后,将该文件保存,文件名可自定义。

err = file.Save("demo.xlsx")
if err != nil {
  panic(err.Error())
}
复制代码

跑起来后,可以发现目录中多了一个 demo.xlsx 文件,打开预览内容如下,达到了预期效果。

文件源码

package main

import "github.com/tealeg/xlsx"

func main() {
	file := xlsx.NewFile()
	sheet, err := file.AddSheet("人员信息收集")
	if err != nil {
		panic(err.Error())
	}
	row := sheet.AddRow()
	cell := row.AddCell()
	cell.Value = "姓名"
	cell = row.AddCell()
	cell.Value = "性别"

	row = sheet.AddRow()
	cell = row.AddCell()
	cell.Value = "张三"
	cell = row.AddCell()
	cell.Value = "男"

	err = file.Save("demo.xlsx")
	if err != nil {
		panic(err.Error())
	}
}
复制代码

读取表格

表格的读取比创建简单很多,依然以上文创建的文件为例。

output, err := xlsx.FileToSlice("demo.xlsx")
if err != nil {
  panic(err.Error())
}
复制代码

只需将文件路径传入上述方法,即可自动读取并返回一个三维切片,我们来读取第一个 sheet 的第二行中的第一个单元格。

log.Println(output[0][1][1]) //Output: 男
复制代码

由此一来就非常容易遍历了。

for rowIndex, row := range output[0] {
  for cellIndex, cell := range row {
    log.Println(fmt.Sprintf("第%d行,第%d个单元格:%s", rowIndex+1, cellIndex+1, cell))
  }
}
复制代码
2020/10/11 16:15:291行,第1个单元格:姓名
2020/10/11 16:15:291行,第2个单元格:性别
2020/10/11 16:15:292行,第1个单元格:张三
2020/10/11 16:15:292行,第2个单元格:男
复制代码

文件源码

package main

import (
	"fmt"
	"github.com/tealeg/xlsx"
	"log"
)

func main() {
	output, err := xlsx.FileToSlice("demo.xlsx")
	if err != nil {
		panic(err.Error())
	}
	log.Println(output[0][1][1])
	for rowIndex, row := range output[0] {
		for cellIndex, cell := range row {
			log.Println(fmt.Sprintf("第%d行,第%d个单元格:%s", rowIndex+1, cellIndex+1, cell))
		}
	}
}
复制代码

修改表格

只是读取表格内容可能在特定场景下无法满足需求,有时候需要对表格内容进行更改。

file, err := xlsx.OpenFile("demo.xlsx")
if err != nil {
  panic(err.Error())
}
复制代码

修改表格之前依然需要先读取文件,只是这次并没有直接将其转化为三维切片。拿到文件句柄后,可以直接修改某一行的内容。

file.Sheets[0].Rows[1].Cells[0].Value = "李四"
复制代码

上述代码将第二行的张三改为了李四,但这还没有结束,接下来需要将文件重新保存。

err = file.Save("demo.xlsx")
if err != nil {
  panic(err.Error())
}
复制代码

打开文件预览,可以看到已经成功将张三改为了李四。

文件源码

package main

import "github.com/tealeg/xlsx"

func main() {
	file, err := xlsx.OpenFile("demo.xlsx")
	if err != nil {
		panic(err.Error())
	}
	file.Sheets[0].Rows[1].Cells[0].Value = "李四"
	err = file.Save("demo.xlsx")
	if err != nil {
		panic(err.Error())
	}
}
复制代码

样式设置

该开源库不仅支持内容的编辑,还支持表格的样式设置,样式统一由结构体 Style 来负责。

type Style struct {
	Border          Border
	Fill            Fill
	Font            Font
	ApplyBorder     bool
	ApplyFill       bool
	ApplyFont       bool
	ApplyAlignment  bool
	Alignment       Alignment
	NamedStyleIndex *int
}
复制代码

拿上述生成的文件为例,假如我要将姓名所在单元格居中,首先要实例化样式对象。

style := xlsx.NewStyle()
复制代码

赋值居中属性。

style.Alignment = xlsx.Alignment{
  Horizontal:   "center",
  Vertical:     "center",
}
复制代码

给第一行第一个单元格设置样式。

file.Sheets[0].Rows[0].Cells[0].SetStyle(style)
复制代码

与修改表格处理逻辑相同,最后保存文件。

err = file.Save("demo.xlsx")
if err != nil {
  panic(err.Error())
}
复制代码

打开预览,可以看到文字已经上下左右居中。

同理,可以修改文字颜色和背景,同样通过 style 的属性来设置。

style.Font.Color = xlsx.RGB_Dark_Red
style.Fill.BgColor = xlsx.RGB_Dark_Green
复制代码

其他还有很多属性可以设置,比如合并单元格、字体、大小等等,大家可以自行测试。

文件源码

package main

import "github.com/tealeg/xlsx"

func main() {
	file, err := xlsx.OpenFile("demo.xlsx")
	if err != nil {
		panic(err.Error())
	}
	style := xlsx.NewStyle()
	style.Font.Color = xlsx.RGB_Dark_Red
	style.Fill.BgColor = xlsx.RGB_Dark_Green
	style.Alignment = xlsx.Alignment{
		Horizontal:   "center",
		Vertical:     "center",
	}
	file.Sheets[0].Rows[0].Cells[0].SetStyle(style)
	err = file.Save("demo.xlsx")
	if err != nil {
		panic(err.Error())
	}
}
复制代码

我是平也,这有一个专注Gopher技术成长的开源项目「go home」


感谢大家的观看,如果觉得文章对你有所帮助,欢迎关注公众号「平也」,聚焦Go语言与技术原理。 关注我