python与word交互_[Python] 自动化办公 邮件/Excel/Word交互快速生成介绍信

转载请注明：陈熹 chenx6542@ (简书号：半为花间酒)

若公众号内转载请联系公众号：早起Python

这篇文章能学到的主要内容：

imbox 读取邮件解析附件

openpyxl 和 python-docx 对文件的交互操作

一、需求描述

你在某三家医院的医务处工作，之前已经发通知让医生们申请外派 A 医院进修，表格 申请.xlsx 如下：

你需要根据他们的申请表开出相应的介绍信：

每个人会单独自己填写好的表格以 “进修申请 xxx” 的邮件标题发到你的邮箱。申请截止日期到了，你打开邮件发现有 300 多人申请，而你觉得从邮件中下载附件，打开 Excel 文件并把对应信息填写到 Word，再修改介绍信文件名为 “xxx 进修介绍信” 实在过于繁琐，你希望借助 Python 自动化高效完成上述任务

二、逻辑梳理

这次的真实需求实际上和之前的推文 批量生成多份合同：/p/3ee47f594d81 非常类似，不同之处在于需要配合邮件相关的工具完成整个需求。本需求同样绕不开一个问题：程序如何知道要将某个信息填到何处？为了解决这个问题，我们需要对模板 介绍信.docx 进行修改，即将需要填写的地方改成某种标识，让程序可以“看到标识就明白此处应该放什么信息”

采取的策略是：将需要填写的地方改成表中的列名，即：

这样程序通过文本识别就能够定位相应信息并完成替换

本需求完整的逻辑包括：

遍历所有邮件，将标题符合要求的邮件附件下载到指定文件夹中

遍历打开文件夹下的所有 Excel 文件

获取每个 Excel 表格中的信息，填写至 Word 模板中

保存文件到新文件夹中

三、代码实现

1. 解析邮件

首先完成第一部分的工作，读取全部邮件：

import keyring

from imbox import Imbox

利用 keyring 库，通过系统密钥环将密码(授权码)预先在本地存储好，后面在代码中调用 keyring 库的方法，通过账号把密码取出来作为变量，降低了密码(授权码)泄露的几率

通过 imbox 库获取附件：

password = keyring.get_password("yagmail","xxx@")

with Imbox('', 'xxx@', password) as imbox:

all_inbox_messages = imbox.messages()

for uid,message in all_inbox_messages:

print(message.attachments)

从需求中我们知道，特定的邮件是以 进修申请 四个字开头的，那么就可以以此为依据作为判断，获取特定邮件的附件：

password = keyring.get_password("yagmail","xxx@")

with Imbox('', 'xxx@', password) as imbox:

all_inbox_messages = imbox.messages()

for uid, message in all_inbox_messages:

if message.subject[:4] == '进修申请':

pass

pass 代码就可以写附件存储了。需要把 Excel 文件存储到指定文件夹中，因此需要先利用 os 库建立文件夹。邮件部分的代码如下：

import keyring

from imbox import Imbox

import os

path = r'C:\xxx'

if not os.path.exists(path + r'\申请表文件夹'):

os.mkdir(path + r'\申请表文件夹')

password = keyring.get_password("yagmail","xxx@")

with Imbox('', 'xxx@', password) as imbox:

all_inbox_messages = imbox.messages()

for uid, message in all_inbox_messages:

if message.subject[:4] == '进修申请':

if message.attachments: # 判断是否存在附件

for attachment in message.attachments:

with open(path + f'\申请表文件夹\\{attachment["filename"]}', 'wb') as file:

file.write(attachment['content'].getvalue())

2. Excel 和 Word 文件交互

接下来的操作涉及 Excel 读取和 Word 文件的写入，需要导入相应的模块。同时建立新文件夹存放最终的介绍信：

from docx import Document

from openpyxl import load_workbook

if not os.path.exists(path + r'\介绍信文件夹'):

os.mkdir(path + r'\介绍信文件夹')

现在 申请表文件夹 中存放 300 多个 Excel 文件，可以利用 glob 库进行遍历和读取：

import glob

for file in glob.glob(path + r'\申请表文件夹\*.xlsx'):

workbook = load_workbook(file)

sheet = workbook.active

有效信息在第二行，列名(文本替换的依据)在第一行。但考虑到有的申请表可能不按常规，填写了多个人的申请，因此用循环，不局限在第二行：

for file in glob.glob(path + r'\申请表文件夹\*.xlsx'):

workbook = load_workbook(file)

sheet = workbook.active

for table_row in range(2, sheet.max_row + 1): # 考虑到有的申请表可能不按常规，填写了多个人的申请，因此用循环

# 每循环一行实例化一个新的word文件

wordfile = Document(path + r'\新模板.docx')

# 单元格需要逐个遍历，每一个都包含着有用的信息

for table_col in range(1, sheet.max_column + 1):

# 旧的文本也就是列名，已经在模板里填好了，用于文本替换，将row限定在第一行后就是列名

old_text = '#' + str(sheet.cell(row=1, column=table_col).value) + '#'

# 新的文本就是实际的信息，table_col循环到某个数值时，实际的单元格和列名就确定了

new_text = str(sheet.cell(row=table_row, column=table_col).value)

获取到信息以后就可以进行 Word 模板文件的文本替换了，根据其 文档 Document - 段落 Paragraph - 文字块 Run的三级结构，在文字块层面完成替换：

# 文档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)

介绍信的落款日期是当天的日期，可以考虑借助 datetime 库获取，并在替换新旧文本时同时判断 #今天日期# 这个文本是否存在，存在就替换为真实日期：

run.text = run.text.replace(old_text, new_text)

run.text = run.text.replace('#今天日期#', datetime.date.today())

最后保存即可，文件名中的姓名即为当前循环行的第一个单元格，sheet.cell(row=table_row,column=1).value

完整代码如下：

import keyring

from imbox import Imbox

from docx import Document

from openpyxl import load_workbook

import os

import glob

import datetime

path = r'C:\xxx'

if not os.path.exists(path + r'\申请表文件夹'):

os.mkdir(path + r'\申请表文件夹')

password = keyring.get_password("yagmail", "xxx@")

with Imbox('', 'xxx@', password) as imbox:

all_inbox_messages = imbox.messages()

for uid, message in all_inbox_messages:

if message.subject[:4] == '进修申请':

if message.attachments:

for attachment in message.attachments:

with open(path + f'\申请表文件夹\\{attachment["filename"]}', 'wb') as file:

file.write(attachment['content'].getvalue())

if not os.path.exists(path + r'\介绍信文件夹'):

os.mkdir(path + r'\介绍信文件夹')

for file in glob.glob(path + r'\申请表文件夹\*.xlsx'):

workbook = load_workbook(file)

sheet = workbook.active

for table_row in range(2, sheet.max_row + 1): # 考虑到有的申请表可能不按常规，填写了多个人的申请，因此用循环

# 每循环一行实例化一个新的word文件

wordfile = Document(path + '\新模板.docx')

# 单元格需要逐个遍历，每一个都包含着有用的信息

for table_col in range(1, sheet.max_column + 1):

# 旧的文本也就是列名，已经在模板里填好了，用于文本替换，将row限定在第一行后就是列名

old_text = '#' + str(sheet.cell(row=1, column=table_col).value) + '#'

# 新的文本就是实际的信息，table_col循环到某个数值时，实际的单元格和列名就确定了

new_text = str(sheet.cell(row=table_row, column=table_col).value)

all_paragraphs = wordfile.paragraphs

for paragraph in all_paragraphs:

for run in paragraph.runs:

run.text = run.text.replace(old_text, new_text)

run.text = run.text.replace('#今天日期#', datetime.date.today())

wordfile.save(path + f'\\介绍信文件夹\\{sheet.cell(row=table_row,column=1).value} 进修介绍信.docx')

整个复杂的需求就被瓦解成多个问题而成功解决！