使用python根据模板批量生成docx文档

Posted 2021-05-17 2016bits

一、需求说明

能够根据模板批量生成docx文档。具体而言，读取excel中的数据，然后使用python批量生成docx文档。

二、实验准备

准备excel数据：

这里是关于学生语数英成绩的统计表，文件名为score.xls

准备模板：

这是给学生家长的成绩通知书，文件名为template.doc

另外，在使用python进行实验之前，需要先安装第三方库docxtpl和xlrd，直接pip install就行：

pip install docxtpl
pip install xlrd

然后将xls和doc和python文件放在同一个目录下

三、代码实现

首先打开xls，读取数据：

workbook = xlrd.open_workbook(sheet_path)

然后从文件中获取第一个表格：

sheet = workbook.sheet_by_index(0)

然后遍历表格的每一行，将数据存入字典列表：

tables = []
for num in range(1, sheet.nrows):
    stu = {}
    stu['name'] = sheet.cell_value(num, 0)
    stu['class'] = sheet.cell_value(num, 1)
    stu['language'] = sheet.cell_value(num, 2)
    stu['math'] = sheet.cell_value(num, 3)
    stu['English'] = sheet.cell_value(num, 4)
    tables.append(stu)

接下来将列表中的数据写入docx文档，其实这个过程可以在读数据时同时进行，即读完一行数据，然后生成一个文档。

首先在指定路径生成一个docx文档：

document = Document(word_path)

然后逐行进行正则表达式的替换：

paragraphs = document.paragraphs
    text = re.sub('name', stu['name'], paragraphs[1].text)
    paragraphs[1].text = text
    text = re.sub('name', stu['name'], paragraphs[2].text)
    text = re.sub('class', stu['class'], text)
    text = re.sub('language', str(stu['language']), text)
    text = re.sub('math', str(stu['math']), text)
    text = re.sub('English', str(stu['English']), text)
    paragraphs[2].text = text

其实不关心格式问题的，到现在为止就已经结束了。但是这样替换后docx中被替换的文字格式也被更改为系统默认的正文格式，所以接下来是将这些改成自己想要的格式：

遍历需要更改格式的段落，然后更改字体大小和字体格式：

for run in paragraph.runs:
            run.font.size = Pt(16)
            run.font.name = "宋体"
            r = run._element.rPr.rFonts
            r.set(qn("w:eastAsia"), "宋体")

最后保存文件：

document.save(path + "\\\\" + r"{}的成绩通知单.docx".format(stu['name']))

完整代码：

from docxtpl import DocxTemplate
import pandas as pd
import os
import xlrd
path = os.getcwd()
# 读表格
sheet_path = path + "\\score.xls"
workbook = xlrd.open_workbook(sheet_path)
sheet = workbook.sheet_by_index(0)
tables = []
for num in range(1, sheet.nrows):
    stu = {}
    stu['name'] = sheet.cell_value(num, 0)
    stu['class'] = sheet.cell_value(num, 1)
    stu['language'] = sheet.cell_value(num, 2)
    stu['math'] = sheet.cell_value(num, 3)
    stu['English'] = sheet.cell_value(num, 4)
    tables.append(stu)
print(tables)

# 写文档
from docx import Document
import re
from docx.oxml.ns import qn
from docx.shared import Cm,Pt
for stu in tables:
    word_path = path + "\\\\template.doc"
    document = Document(word_path)
    paragraphs = document.paragraphs
    text = re.sub('name', stu['name'], paragraphs[1].text)
    paragraphs[1].text = text
    text = re.sub('name', stu['name'], paragraphs[2].text)
    text = re.sub('class', stu['class'], text)
    text = re.sub('language', str(stu['language']), text)
    text = re.sub('math', str(stu['math']), text)
    text = re.sub('English', str(stu['English']), text)
    paragraphs[2].text = text
    for paragraph in paragraphs[1:]:
        for run in paragraph.runs:
            run.font.size = Pt(16)
            run.font.name = "宋体"
            r = run._element.rPr.rFonts
            r.set(qn("w:eastAsia"), "宋体")
    document.save(path + "\\\\" + r"{}的成绩通知单.docx".format(stu['name']))

四、实验结果

文件中的文件：

生成的文件样例：