|
|
from docx.shared import Cm
|
|
|
from pptx.enum.shapes import MSO_SHAPE_TYPE
|
|
|
import hashlib
|
|
|
from docx import Document
|
|
|
from pptx import Presentation
|
|
|
|
|
|
|
|
|
def WordReplaceText(doc, tag, pv):
|
|
|
for paragraph in doc.paragraphs:
|
|
|
if tag not in paragraph.text:
|
|
|
continue
|
|
|
tmp = ''
|
|
|
runs = paragraph.runs
|
|
|
for i, run in enumerate(runs):
|
|
|
tmp += run.text # 合并run字符串
|
|
|
if tag in tmp:
|
|
|
# 如果存在匹配得字符串，那么将当前得run替换成合并后得字符串
|
|
|
run.text = run.text.replace(run.text, tmp)
|
|
|
run.text = run.text.replace(tag, pv)
|
|
|
tmp = ''
|
|
|
else:
|
|
|
# 如果没匹配到目标字符串则把当前run置空
|
|
|
run.text = run.text.replace(run.text, '')
|
|
|
if i == len(runs) - 1:
|
|
|
# 如果是当前段落一直没有符合规则得字符串直接将当前run替换为tmp
|
|
|
run.add_text(tmp)
|
|
|
|
|
|
|
|
|
def WordReplacePic(doc, tag, pic, width=14.63):
|
|
|
for paragraph in doc.paragraphs:
|
|
|
if tag in paragraph.text:
|
|
|
has_replaced = False
|
|
|
for run in paragraph.runs:
|
|
|
run.clear()
|
|
|
if not has_replaced:
|
|
|
run.add_picture(pic, width=Cm(width))
|
|
|
has_replaced = True
|
|
|
|
|
|
|
|
|
def WordReplaceTextInTable(doc, tag, pv):
|
|
|
for table in doc.tables:
|
|
|
for row in table.rows:
|
|
|
for cell in row.cells:
|
|
|
# 如果只是为了内容，直接替换cell.text,但是为了保存原有格式，需要将每个单元格的文本当作一段看待，以此提取出run来不修改原格式
|
|
|
for paragraph in cell.paragraphs:
|
|
|
if tag in paragraph.text:
|
|
|
has_replaced = False
|
|
|
for run in paragraph.runs:
|
|
|
run.clear()
|
|
|
if not has_replaced:
|
|
|
run.add_text(pv)
|
|
|
has_replaced = True
|
|
|
|
|
|
|
|
|
def PptReplaceText(prs, search_str, repl_str):
|
|
|
for x in range(len(prs.slides)):
|
|
|
for shape in prs.slides[x].shapes:
|
|
|
if hasattr(shape, "text"):
|
|
|
if (shape.text.find(search_str)) != -1:
|
|
|
text_frame = shape.text_frame
|
|
|
cur_texts = text_frame.paragraphs[0].runs
|
|
|
for index in range(len(cur_texts)):
|
|
|
if (cur_texts[index].text.find(search_str)) != -1:
|
|
|
cur_text = text_frame.paragraphs[0].runs[index].text
|
|
|
new_text = cur_text.replace(str(search_str), str(repl_str))
|
|
|
text_frame.paragraphs[0].runs[index].text = new_text
|
|
|
|
|
|
group_shapes = [shp for shp in prs.slides[x].shapes
|
|
|
if shp.shape_type == MSO_SHAPE_TYPE.GROUP]
|
|
|
for group_shape in group_shapes:
|
|
|
for shape in group_shape.shapes:
|
|
|
if shape.has_text_frame:
|
|
|
if (shape.text.find(search_str)) != -1:
|
|
|
text_frame = shape.text_frame
|
|
|
for index in range(len(text_frame.paragraphs)):
|
|
|
cur_text = text_frame.paragraphs[index].text
|
|
|
if (cur_text.find(search_str)) != -1:
|
|
|
new_text = cur_text.replace(str(search_str), str(repl_str))
|
|
|
text_frame.paragraphs[index].text = new_text
|
|
|
|
|
|
|
|
|
def PptReplacePic(prs, newpic, oldpic):
|
|
|
# 把旧样本图片Logo,获取指纹
|
|
|
imageFile = open(oldpic, "rb")
|
|
|
imgBlob = imageFile.read()
|
|
|
md5finger = hashlib.md5(imgBlob).hexdigest()
|
|
|
for slide in list(prs.slides)[0:]:
|
|
|
for shape in list(slide.shapes):
|
|
|
ispicture = False
|
|
|
try:
|
|
|
md5img = hashlib.md5(shape.image.blob).hexdigest()
|
|
|
ispicture = True
|
|
|
except:
|
|
|
pass
|
|
|
e = shape.element
|
|
|
if ispicture:
|
|
|
if md5img == md5finger:
|
|
|
slide.shapes.add_picture(newpic, shape.left, shape.top, shape.width, shape.height)
|
|
|
e.getparent().remove(e)
|
|
|
pass
|
|
|
|
|
|
|
|
|
def ReplaceTxtInTable(ppt, oldStr, newStr):
|
|
|
# 所有幻灯片
|
|
|
for i in range(0, len(ppt.slides)):
|
|
|
placeholder = ppt.slides[i].shapes
|
|
|
for j in range(len(placeholder)):
|
|
|
if placeholder[j].has_table: # 所有表格
|
|
|
for row in range(len(placeholder[j].table.rows)):
|
|
|
for col in range(len(placeholder[j].table.columns)):
|
|
|
if placeholder[j].table.cell(row, col).text == oldStr:
|
|
|
placeholder[j].table.cell(row, col).text = newStr
|
