python应用之Word生成

最新推荐文章于 2024-10-23 22:50:20 发布

落子无悔!

最新推荐文章于 2024-10-23 22:50:20 发布

阅读量4.3k

点赞数 1

分类专栏： # Python 文章标签： python word

本文链接：https://blog.csdn.net/qq_32460819/article/details/125343501

版权

Python 专栏收录该内容

30 篇文章 3 订阅

订阅专栏

1. 介绍

最近一个做文职的朋友提出了一个想法, 希望实现自动修改他的一个Word内容. 因为正常情况下一个工程其实只需要修改Word里面的日期, 名称, 和一些参数, 所以希望我能为他写个脚本, 完成Word的自动填充并生成新的Word.

1.1 需求

指定位置关键词填充: 比如XXXXX村饮水安全巩固提升工程
时间填充: 已知A时间, B时间=A时间+days
随机数: 允许偏差:±10mm 生成指定范围内的N个随机数

1.2 对应解决方案

编写Word模板, 里面标记需要填充的关键字
指定Excel表格填写关键字和对应的值
读取Word模板待填充关键字, 匹配Excel表格中的关键字对应的值
根据关键字格式如
1. date_1-4 在date_1的基础上+4天
2. random_1_4(1,5) 生成4个范围是1,5的随机数
写入到新的Word

2. 实现

2.1 环境

python3.8
pip install docx-mailmerge==0.5.0  # 用于对Word里面设置的域(关键字)进行解析和填充
pip install python-docx==0.8.11  # 用于读取Word的段落, 表格进行解析和增删改查
pip install pandas==1.4.2 # 用于读取Excel表格数据

2.2 创建Word模板, 增加域定位关键字

在这里插入图片描述

2.3 编写对应的Excel

在这里插入图片描述

2.4 文件存储路径

.
├── main.py # 主函数
├── readme.md
├── req.txt # 环境包request
├── result
│   └── t.docx # 生成的文件
├── sources
│   ├── doc1.docx # Word模板
│   └── input.csv # 定义的key-value
$ python main.py  # 运行

2.5 main.py代码

在这里插入图片描述

'''
1. 初始化对象
2. 获取模本对象 <- template_file_path
3. 读取输入csv文件<- input_csv_path
4. 写数据到对象
    1. 模板对象中key, 直接填csv中对应的key数据
    2. 模板对象中的date, 根据csv中的数据填充, 并依次生成date_n_long的数据进行填充
    3. 模板对象中的random, 根据key中的(a,b), 生成对应的随机数, 进行填充 
5. 对象保存 <- save_path
'''

import pandas as pd # pandas==1.4.2
from mailmerge import MailMerge # docx-mailmerge==0.5.0
import random
import re
from datetime import datetime
from dateutil.relativedelta import relativedelta


class WriteDocx():
    def __init__(self,format_date='%Y.%m.%d'):
        super().__init__()
        self.doc = None
        self.doc_key = None
        self.dict_csv = {}
        self.format_date = format_date
    
    def createTemplateDoc(self, template_file_path = "sources/doc1.docx"):
        self.doc = MailMerge(template_file_path)
        self.doc_key = self.doc.get_merge_fields()
        print("模板里面的关键字包含如下: ", self.doc_key)
        print("模板对象创建完毕, 开始向对象填充数据...")
        
    def getInputData(self, input_csv_path = "sources/input.csv"):
        csv = pd.read_csv("sources/input.csv")
        print(csv.head(2))
        for i in range(len(csv["key"].values)):
            self.dict_csv[csv.loc[i]["key"]] = csv.loc[i]["value"]
        print(self.dict_csv)

    # 解析生成随机数的方案
    def getRandom(self, data = "random_2_6(1,3)")->list:
        random_order = [eval(_) for _ in list(re.findall(r"(.*)\((.*),(.*)\)", data.split("_")[-1])[0])]
        res_list = ["%0.2f"%random.uniform(random_order[1], random_order[2]) for i in range(random_order[0])]
        return "[%s]"%",".join(res_list)

    def getDate(self, date="2017.10.23", data = 'date_2-6'):
        add_day = int(data.split("-")[-1]) # 增加的天数
        time_1 = datetime.strptime(date, "%Y.%m.%d") + relativedelta(days=add_day)
        return datetime.strftime(time_1, '%Y.%m.%d') # 这里是最终的格式
    
    # 格式化日期格式
    def format_date_(self, date):
        time_1 = datetime.strptime(date, "%Y.%m.%d")
        return datetime.strftime(time_1, self.format_date) # 这里是最终的格式
    
    def writeDoc(self, ):
        # 生成doc_key中random和date的数据
        # 对于random生成对应的数据, 对于date生成对应的延时
        for _ in self.doc_key:
            if "random" in _:
                self.dict_csv[_] = self.getRandom(_) # random_2_2(0,3) --> [1.2, 0.5]
            if "date" in _ and len(_.split("-"))==2: # '2017.10.23', 'date_2-6' --> 2017.10.29
                self.dict_csv[_] = self.getDate(self.dict_csv[_.split("-")[0]], _)
            if "date" in _: # 统一格式化日期格式  2017.10.23->
                self.dict_csv[_] = self.format_date_(self.dict_csv[_])

        print("要填充的数据是: ", self.dict_csv)
        print(type(self.dict_csv))
        self.doc.merge(** self.dict_csv)    #生成一份Word文档

        
    def save(self, save_path = "result/t.docx"):
        self.doc.write(save_path)
        print("新文件已经成功保存到: ", save_path)


# 只需要修改这里的位置即可
def main():
    object = WriteDocx(format_date='%Y年%m月%d日')
    object.createTemplateDoc("sources/doc1.docx")
    object.getInputData("sources/input.csv")
    object.writeDoc()
    object.save("result/t.docx")

if __name__=='__main__':
    main()

3. 思考

通过这次练习, 让我发现文职人员能够改进很多工作流程. 就拿我们每次转团关系的介绍信, 都是发下来自己手写里面的内容, 比如姓名等个人信息, 团关系从哪个位置转到哪个位置, 在交上去盖章. 如果根据本文的工作流程, 只需要统计每个人的上述信息保存到Excel的每一列, 做一个Word模板, 就可以一次性按列生成所有人对应的转团关系介绍信, 打印盖章即可, 完全不必要自己手写, 流水线操作总是比分开单干更加便捷顺利. 最后2202年了, 希望文职工作的同志都能拥有编程思想, 也许无法实现, 但是咱可以找编程朋友改善工作流程, 提高工作效率, 提高幸福指数!!!