用Python将word文件转换成html
序
最近公司一个客户大大购买了一堆医疗健康方面的科普文章,希望能放到我们正在开发的健康档案管理软件上。客户大大说,要智能推送!要掌握节奏!要深度学习!要让用户留恋网站无法自拔!
话说符合以上特点的我也只能联想到某榴了。
当然,万里长征的第一步是把文章导入我们的数据库。项目使用的是AWS的dynamoDB,是非关系型数据库,所有内容都是以json的形式储存的。而客户大大购买来的文章,一共600多篇,要么是word要么是Adobe indesign的indd。找了一圈,没有发现合适的应用可以把word或indd转化成干净的html。所以我只能自己造轮子啦~听说python很擅长文本处理,所以就是你了,python!这是我第一次用python写项目,不符合规范的地方欢迎大神提点。
太长不看
用逆天的python 模块mammoth和docx 处理你的word文件;把indd批量转化成pdf然后用layout_scanner转化成html。
word批量转化为html
1、 建立文件结构并批量读取文件
在根目录下创建几个文件夹,用来放不同格式的文件,我把所有要处理的word文件放在docfiles 这个子目录里。word.py里写转化程序。
-
├── docfiles -
├── imgs -
├── inddfiles -
├── output -
└── pdfs -
└── word.py
2、引入模块和申明文件路径
-
import mammoth -
import mammoth.transforms -
import os -
from docx import Document -
from bson import json_util -
import zipfile -
import json -
import unidecode -
import requests -
guidUrl = "https://my.phrplus.com/REST/guid" -
inputPath = '/Users/admin/cwell/parser/docfiles/' -
imgPath = "/Users/admin/cwell/parser/imgs/" -
outputFile = '/Users/admin/cwell/parser/output/output.json'
mammoth: 核心组件,用来做转化工作
docx: 另一个做转化工作的模块,用来补充mammoth
os: 用来在系统中读取写入文件
zipfile: 用来解压word文档以提取图片
json: 用来把数据转化成json
bson: 用来配置写入json文件
unicode:用来处理字符
requests:用来调用api
3、转换单个文件
-
styleMap = """ -
p[style-name='Title'] => h1.hide -
p[style-name='Subhead 1'] => h3 -
p[style-name='List Bullet'] => ul.first > li:fresh -
p[style-name='List Bullet 2'] => ul.second > li:fresh -
p[style-name='Hyperlink']=>a.link -
""" -
def convert_image(image): -
return { -
"src":"" -
} -
def parseFile(f): -
document = Document(inputPath+f) -
article = {"Title":document.core_properties.title,"Content":""} -
with open(inputPath+f,"rb") as docFile: -
html = mammoth.convert_to_html(docFile,style_map=styleMap,convert_image=mammoth.images.img_element(convert_image)) -
decoded = unidecode.unidecode(html.value) -
if not article["Title"]: -
for para in document.paragraphs: -
if para.style.name == 'Title': -
if para.text: -
article["Title"] = para.text -
article["Content"]=decoded -
return article
parseFile就是核心功能了。传递进来的参数f是文件名,和文件路径合在一起能够帮我们准确定位要转化的文件。首先用docx找到文档的标题,并创建一个dictionary,里面包含标题和内容。然后用mammoth转化整个文件。注意命令中要用到stylemap和convertimage。前者用来规定转化规则:'style-name'是word里的式样名称,用word打开文档,点击任意一个元素可以查看其式样名称;这里规定标题转化为h1,副标题转化为h2等等。关于列表的转化规则这里就不详细叙述了,具体可以参考下面的文章:
参考链接
-
Converting docx to clean HTML: handling the XML structure mismatch
'convert_image' 是用来规定图片的转化方式的,由于我准备之后批处理所有文档中的图片,在这里就告诉程序不储存任何图片信息。但是于此同时保留图片的img tag以便标注图片在文档中的位置。如果不规定任何转化方式,生成的html里面会包含一大长串base64的图片信息。
mammoth转化出来的html是含有unicode的,不知道为什么python里跑一直报错,就用unicode解码了一下。
这之后,如果前面的程序没有抓取到文档标题,用docx换个姿势再抓取一下。
最后返回article这个dictionary。
4、抓取图片
-
def extractImage(f): -
ziped = zipfile.ZipFile(inputPath+f) -
allFiles = ziped.namelist() -
imgs = filter(lambda x: x.startswith('word/media/'), allFiles) -
imgNameArr = [] -
for img in imgs: -
res = requests.post(guidUrl) -
if res.status_code is 200: -
guid = res.text -
data = ziped.read(img,imgPath) -
idxStr = os.path.basename(img).split(".")[0][-1:] -
imgDict = {} -
imgDict["index"] = int(idxStr)-1 -
imgDict["fileName"] = guid+".jpg" -
imgNameArr.append(imgDict) -
targetPath = os.path.join(imgPath,guid+".jpg") -
target = open(targetPath,"wb") -
target.write(data) -
target.close() -
ziped.close() -
return imgNameArr
没想到word文档其实是一个压缩文件吧?如果直接把word文档的后缀名改成zip然后再用解压软件查看,会看到一个media文件夹,里面就包含所有插入的图片。
用ziped读取文档,然后找到存放图片的media文件夹,每一个图片重新用guid命名,生成一个dictionary,里面包含的信息有“此图片在文档中出现的顺序”和文件名。话说media中的图片都被按照顺序重新命名为image1.png, image2.png,刚好为我们抓取顺序信息提供了方便。
(python也有生成guid的模块,我在这里调用api有点多此一举,但是为了和项目中其他图片需要用到的uuidv4保持一致还是用了)
之后就是把图片存在‘imgs’这个文件夹下。
5、生成json
-
def processDocs(path): -
result = [] -
for f in os.listdir(path): -
if not f.startswith('.'): -
imgNameArr = extractImage(f) -
article = parseFile(f) -
fileName = os.path.basename(f) -
contentArr = article["Content"].split("<img") -
for idx, section in enumerate(contentArr): -
for info in imgNameArr: -
if idx is info["index"]: -
contentArr[idx] = section+"<img alt='"+info["fileName"]+"' data-fileName='"+info["fileName"]+"'" -
article["Content"] = ''.join(contentArr) -
result.append(article) -
with open(outputFile,'w+') as f: -
json.dump(result,f,default=json_util.default)
最后要用到的一个function就是写个循环挨个处理docfiles文件夹底下的文件了。针对每一个文件,记得把之前生成的图片信息的数组map到html里,然后在写入到json文件里就大功告成了!
indd转化为html
话说,到现在为止,我还没有找到一个完美的解决方案。我使用了相同的思路,把indd先批量生成为pdf(有一个indesign 脚本就是专门批量转化pdf的),然后用了一个叫做layout_scanner的github项目抓取pdf信息并转化为html。最后生成的html包含了文字和图片,但是图标和排版就保存不下来了。客户大大表示不满意。我也很惆怅啊!机智的小伙伴们如果有更好的思路请务必告诉我!如果需要我详细说明一下这一块内容,我会更新在这篇文章中。