改为一个函数,传入三个参数,分别为现有word,输入word,以及水印
时间: 2024-12-24 07:32:18 浏览: 2
当然,我们可以将上述代码封装成一个函数,使其更通用,接受现有Word文档、新的内容以及水印作为参数。这是一个例子:
```python
def add_watermark_to_word(existing_doc_path, new_content_path, watermark_path):
from docx import Document
from PIL import Image
# 打开现有Word文档
existing_doc = Document(existing_doc_path)
# 新内容加载为字符串
with open(new_content_path, 'r', encoding='utf-8') as f:
new_text = f.read()
# 定义水印图片
watermark_image = Image.open(watermark_path)
# 其他步骤与之前相同
for section in existing_doc.sections:
header_footer = section.header
if header_footer is not None:
header_footer.paragraphs.clear()
break
watermark_image = ... # 调整图像等操作
watermark = ... # 插入图片
text_paragraph = ... # 创建并添加文本水印
existing_doc.add_paragraph(new_text, style=text_paragraph.style) # 添加新内容
# 保存修改后的文档
existing_doc.save('output_word.docx')
return 'Watermark added successfully.'
# 使用函数
add_watermark_to_word('existing_doc.docx', 'new_content.txt', 'watermark.png')
阅读全文