怎么用Python+Excel+Word制作百份合同

前言

大家好，又到了Python辦公自動(dòng)化系列。

今天我們繼續(xù)分享一個(gè)真實(shí)的辦公自動(dòng)化需求：如何使Python+Excel+Word批量生成指定格式內(nèi)容的合同。

主要涉及的知識(shí)點(diǎn)有：openpyxl模塊的綜合運(yùn)用與Word文檔的兩種遍歷邏輯。

需求描述

你是乙方建筑公司，手上有一份空白合同模板的Word文件，如下圖:

另外還有一份Excel合同信息表，其中是所有甲方(發(fā)包人)在合同中需要填寫的內(nèi)容

可見一行為一個(gè)公司的全部信息，現(xiàn)在需要把Excel中每一個(gè)公司的信息填入空白Word合同模板中，生成各公司的合同，最終結(jié)果如下

步驟分析

原本我們需要將Excel匯總表中每一行的信息填進(jìn)word模板中，生成相應(yīng)的合同。

現(xiàn)在我們需要交給Python來實(shí)現(xiàn)，就引出了一個(gè)問題：程序如何知道要將某個(gè)信息填到哪個(gè)下劃線？ 為了解決這個(gè)問題，我們需要對(duì)模板進(jìn)行修改。

即將下劃線改成某種標(biāo)識(shí)，讓程序可以看到標(biāo)識(shí)就明白此處應(yīng)該放什么信息，這里采取的策略是：將需要填寫的下劃線改成匯總表中的列名，即下圖所示

這樣程序就可以識(shí)別需要填寫什么內(nèi)容了。所謂的識(shí)別在這里可以換一個(gè)特別簡單的詞，即文本替換。只要檢索到#xxxx#(excel中的列名)，把這個(gè)替換成具體的信息就可以了。

出于這種策略，列名就需要用#xxxx#的格式，否則正常的無關(guān)文本中的信息也會(huì)被替換，就破壞了原有的需求，最后模板被修改成如下：

通過Excel表我們可以看到，一行為一個(gè)公司的信息，而每一列的列名就存在于模板中，用各個(gè)公司的實(shí)際信息替換到模板中的列名(程序識(shí)別和文本替換的依據(jù))

用這樣的方法就可以完成這個(gè)需求。整個(gè)大需求的實(shí)現(xiàn)可以按照下面的步驟：

分析后的步驟：

1. 將 空白合同 調(diào)整成 合同模板，需要填寫的下劃線改成專屬的列名
2. 打開Excel表，按行循環(huán)，然后按單元格逐個(gè)循環(huán)各個(gè)信息，每個(gè)信息都找到模板中存在的對(duì)應(yīng)列名并將其替換（如果不理解下文還有解釋）
3. 每次循環(huán)完一行的全部單元格后保存合同，生存各個(gè)公司單獨(dú)的合同

分析清楚后邏輯就非常簡單了，但有一個(gè)隱含的知識(shí)點(diǎn)沒有提到，讓我們邊寫代碼邊說！

代碼實(shí)現(xiàn)

首先導(dǎo)入模塊，設(shè)置路徑，建立文件夾，本例中涉及Excel表的打開和Word的創(chuàng)建，因此需要從openpyxl導(dǎo)入load_workbook，而Word無論打開還是創(chuàng)建，用docx模塊的Document均可

from docx import Document
from openpyxl import load_workbook
# 利用os模塊建立文件夾，用于存放生成的合同
import os

# 給定合同模板和匯總表所在的文件夾路徑，方便復(fù)用
path = r'C:\Users\chenx\Desktop\合同'

# 結(jié)合路徑判斷生成文件夾，規(guī)避程序報(bào)錯(cuò)而終止的風(fēng)險(xiǎn)
if not os.path.exists(path + '/' + '全部合同'):
    os.mkdir(path + '/' + '全部合同')

接著打開Excel文件

workbook = load_workbook(path + '/' + '合同信息表.xlsx')
sheet = workbook.active

現(xiàn)在遍歷Excel，生成合同。前面也反復(fù)提到，Excel的每一行是一份特定合同的信息，因此docx針對(duì)Word文件的實(shí)例化和保存一定是在循環(huán)體里的，而不像Excel的實(shí)例化是在循環(huán)體外面

# 有效信息行是從第二行開始的，第二行是表頭，包含列名，也是文本替換的依據(jù)
for table_row in range(2, sheet.max_row + 1):
    # 每循環(huán)一行實(shí)例化一個(gè)新的word文件
    wordfile = Document(path + '/' + '合同模板.docx')
    # 單元格需要逐個(gè)遍歷，每一個(gè)都包含著有用的信息
    for table_col in range(1, sheet.max_column + 1):
        # 舊的文本也就是列名，已經(jīng)在模板里填好了，用于文本替換，將row限定在第一行后就是列名
        old_text = str(sheet.cell(row=1, column=table_col).value)
        # 新的文本就是實(shí)際的信息，table_col循環(huán)到某個(gè)數(shù)值時(shí)，實(shí)際的單元格和列名就確定了
        new_text = str(sheet.cell(row=table_row, column=table_col).value)
        # 加上這個(gè)判斷是因?yàn)槿掌谛畔⒆x進(jìn)程序是“日期 時(shí)間”格式的，如果要保留日期信息可以用字符串方法或者用time/datetime模塊處理
        if ' ' in new_text:
            new_text = new_text.split()[0]

通過下圖進(jìn)一步理解這個(gè)替換：

例如程序已經(jīng)進(jìn)入第3個(gè)循環(huán)(循環(huán)到第3個(gè)公司)，針對(duì)單元格的循環(huán)進(jìn)入第4個(gè)循環(huán)，那么此時(shí)獲取的實(shí)際值是建設(shè)C公園，對(duì)應(yīng)的列名是#工程內(nèi)容#。

此時(shí)就明確了需要被替換的內(nèi)容了，只要在模板中找到#工程內(nèi)容#把它替換為建設(shè)C公園即可！了解了這個(gè)替換后，下一步就是遍歷Word模板，找到對(duì)應(yīng)列名替換！

之前我們說過docx模塊，Word文本存在文檔Document-段落Paragraph-文字塊Run的三級(jí)結(jié)構(gòu)，需要遍歷文本可以用以下代碼：

all_paragraphs = wordfile.paragraphs
for paragraph in all_paragraphs:
    print(paragraph.text)
    for run in paragraph.runs:
        print(run.text)

針對(duì)段落和文字塊均可用.text獲取到文字信息。本需求隱含的陷阱就在這里，注意一下合同最后需要填寫的內(nèi)容：

這部分內(nèi)容如果用上述代碼是遍歷不到的。為什么？因?yàn)檫@是Word文檔中的表格！

遍歷表格需要有專門的遍歷邏輯:文檔Document-表格Table-行Row/列Column-單元格Cell，遍歷表格中文本的代碼如下：

all_tables = wordfile.tables
for table in all_tables:
    # 也可按列遍歷
    for row in table.rows:
        for cell in row.cells:
            print(cell.text)

有了這些補(bǔ)充的知識(shí)之后，本案例中最核心的代碼就可以這么寫

for table_row in range(2, sheet.max_row + 1):
    wordfile = Document(path + '/' + '合同模板.docx')
    for table_col in range(1, sheet.max_column + 1):
        old_text = str(sheet.cell(row=1, column=table_col).value)
        new_text = str(sheet.cell(row=table_row, column=table_col).value)
        if ' ' in new_text:
            new_text = new_text.split()[0]
        
        # 文檔Document - 段落Paragraph - 文字塊Run
        all_paragraphs = wordfile.paragraphs
        for paragraph in all_paragraphs:
            for run in paragraph.runs:
                run.text = run.text.replace(old_text, new_text)

        # 文檔Document - 表格Table - 行Row/列Column - 單元格Cell
        all_tables = wordfile.tables
        for table in all_tables:
            for row in table.rows:
                for cell in row.cells:
                    cell.text = cell.text.replace(old_text, new_text)

    # 獲取公司名用以生成合同的名稱
    company = str(sheet.cell(row=table_row, column=1).value)
    wordfile.save(path + '/' + f'全部合同/{company}合同.docx')

寫在最后

本次的案例具有較強(qiáng)的實(shí)用性，并且需求可以延伸成為：將一份信息匯總表Excel中的每一個(gè)單獨(dú)信息(每一行或者每一列為個(gè)人、公司或者其他的信息)填寫到指定的模板Eord中，生成單獨(dú)的文檔，不過在寫自動(dòng)化腳本之前也要先拆分任務(wù)，明確思路再進(jìn)行！