添加通过XML直接替换Word文档占位符的功能，作为表格处理失败时的备用方案。同时，优化表格占位符替换逻辑，增强错误处理和调试信息输出，确保在处理复杂表格时的稳定性。
This commit is contained in:
python
parent
4d9080855c
commit
dab5d8ee59
@ -4,6 +4,7 @@
|
||||
import os
|
||||
import re
|
||||
import tempfile
|
||||
import zipfile
|
||||
from typing import Dict, List, Optional
|
||||
from datetime import datetime, timedelta
|
||||
from pathlib import Path
|
||||
@ -11,6 +12,7 @@ from docx import Document
|
||||
from minio import Minio
|
||||
from minio.error import S3Error
|
||||
import pymysql
|
||||
from xml.etree import ElementTree as ET
|
||||
|
||||
|
||||
class DocumentService:
|
||||
@ -119,6 +121,69 @@ class DocumentService:
|
||||
except S3Error as e:
|
||||
raise Exception(f"从MinIO下载模板文件失败: {str(e)}")
|
||||
|
||||
def replace_placeholders_via_xml(self, docx_path: str, field_data: Dict[str, str]) -> bool:
|
||||
"""
|
||||
通过直接操作XML来替换占位符（备用方案，用于处理表格访问失败的情况）
|
||||
|
||||
Args:
|
||||
docx_path: docx文件路径
|
||||
field_data: 字段数据字典
|
||||
|
||||
Returns:
|
||||
是否成功替换
|
||||
"""
|
||||
try:
|
||||
# docx文件实际上是一个ZIP文件
|
||||
# 需要创建一个新的ZIP文件来替换内容
|
||||
temp_zip_path = docx_path + '.tmp'
|
||||
|
||||
with zipfile.ZipFile(docx_path, 'r') as zip_read:
|
||||
with zipfile.ZipFile(temp_zip_path, 'w', zipfile.ZIP_DEFLATED) as zip_write:
|
||||
# 复制所有文件，但替换word/document.xml
|
||||
for item in zip_read.infolist():
|
||||
if item.filename == 'word/document.xml':
|
||||
# 读取并修改XML内容
|
||||
xml_content = zip_read.read(item.filename).decode('utf-8')
|
||||
|
||||
# 替换占位符
|
||||
modified = False
|
||||
for field_code, field_value in field_data.items():
|
||||
placeholder = f"{{{{{field_code}}}}}"
|
||||
replacement_value = str(field_value) if field_value else ''
|
||||
|
||||
if placeholder in xml_content:
|
||||
xml_content = xml_content.replace(placeholder, replacement_value)
|
||||
modified = True
|
||||
print(f"[DEBUG] XML替换占位符: {placeholder} -> '{replacement_value}'")
|
||||
|
||||
# 写入修改后的XML
|
||||
zip_write.writestr(item.filename, xml_content.encode('utf-8'))
|
||||
if modified:
|
||||
print(f"[DEBUG] XML成功替换占位符")
|
||||
else:
|
||||
# 复制其他文件
|
||||
zip_write.writestr(item, zip_read.read(item.filename))
|
||||
|
||||
# 替换原文件
|
||||
if os.path.exists(temp_zip_path):
|
||||
os.replace(temp_zip_path, docx_path)
|
||||
return True
|
||||
|
||||
return False
|
||||
|
||||
except Exception as e:
|
||||
print(f"[WARN] XML替换占位符失败: {str(e)}")
|
||||
import traceback
|
||||
print(traceback.format_exc())
|
||||
# 清理临时文件
|
||||
temp_zip_path = docx_path + '.tmp'
|
||||
if os.path.exists(temp_zip_path):
|
||||
try:
|
||||
os.remove(temp_zip_path)
|
||||
except:
|
||||
pass
|
||||
return False
|
||||
|
||||
def fill_template(self, template_path: str, field_data: Dict[str, str]) -> str:
|
||||
"""
|
||||
填充Word模板中的占位符
|
||||
@ -473,6 +538,9 @@ class DocumentService:
|
||||
total_replacements += before_text.count(placeholder)
|
||||
|
||||
# 替换表格中的占位符
|
||||
table_replacements = 0
|
||||
table_errors = 0
|
||||
use_xml_fallback = False
|
||||
try:
|
||||
for table_idx, table in enumerate(doc.tables):
|
||||
try:
|
||||
@ -482,8 +550,10 @@ class DocumentService:
|
||||
# 安全地获取表格行数
|
||||
try:
|
||||
row_count = len(table.rows)
|
||||
print(f"[DEBUG] 表格 {table_idx} 有 {row_count} 行")
|
||||
except Exception as e:
|
||||
print(f"[WARN] 无法获取表格 {table_idx} 的行数，跳过该表格: {str(e)}")
|
||||
table_errors += 1
|
||||
continue
|
||||
|
||||
# 使用索引方式访问行，而不是迭代器，避免在迭代时触发内部索引访问错误
|
||||
@ -521,6 +591,7 @@ class DocumentService:
|
||||
continue
|
||||
except (IndexError, AttributeError) as e:
|
||||
print(f"[WARN] 表格 {table_idx} 行 {row_idx} 无法访问单元格，跳过该行: {str(e)}")
|
||||
table_errors += 1
|
||||
continue
|
||||
|
||||
# 安全地遍历单元格
|
||||
@ -547,46 +618,81 @@ class DocumentService:
|
||||
continue
|
||||
except (IndexError, AttributeError) as e:
|
||||
print(f"[WARN] 表格 {table_idx} 行 {row_idx} 单元格 {cell_idx} 无法访问段落，跳过: {str(e)}")
|
||||
table_errors += 1
|
||||
continue
|
||||
|
||||
for para_idx, paragraph in enumerate(paragraphs):
|
||||
try:
|
||||
before_text = paragraph.text
|
||||
if not before_text:
|
||||
continue
|
||||
|
||||
# 检查是否有占位符
|
||||
has_placeholder = False
|
||||
for field_code in field_data.keys():
|
||||
placeholder = f"{{{{{field_code}}}}}"
|
||||
if placeholder in before_text:
|
||||
has_placeholder = True
|
||||
break
|
||||
|
||||
if not has_placeholder:
|
||||
continue
|
||||
|
||||
print(f"[DEBUG] 表格 {table_idx} 行 {row_idx} 单元格 {cell_idx} 段落 {para_idx} 发现占位符: '{before_text[:50]}...'")
|
||||
|
||||
replace_placeholder_in_paragraph(paragraph)
|
||||
after_text = paragraph.text
|
||||
|
||||
if before_text != after_text:
|
||||
print(f"[DEBUG] 表格替换成功: '{before_text[:50]}...' -> '{after_text[:50]}...'")
|
||||
table_replacements += 1
|
||||
# 检查哪些占位符被替换了
|
||||
for field_code in field_data.keys():
|
||||
placeholder = f"{{{{{field_code}}}}}"
|
||||
if placeholder in before_text and placeholder not in after_text:
|
||||
replaced_placeholders.add(field_code)
|
||||
total_replacements += before_text.count(placeholder)
|
||||
else:
|
||||
print(f"[WARN] 表格 {table_idx} 行 {row_idx} 单元格 {cell_idx} 段落 {para_idx} 替换后文本未改变")
|
||||
except Exception as e:
|
||||
# 如果单个段落处理失败，记录错误但继续处理其他段落
|
||||
print(f"[WARN] 表格 {table_idx} 行 {row_idx} 单元格 {cell_idx} 段落 {para_idx} 处理出错: {str(e)}")
|
||||
table_errors += 1
|
||||
import traceback
|
||||
print(traceback.format_exc())
|
||||
continue
|
||||
except Exception as e:
|
||||
# 如果单个单元格处理失败，记录错误但继续处理其他单元格
|
||||
print(f"[WARN] 表格 {table_idx} 行 {row_idx} 单元格 {cell_idx} 处理出错: {str(e)}")
|
||||
table_errors += 1
|
||||
continue
|
||||
except Exception as e:
|
||||
# 如果单个行处理失败，记录错误但继续处理其他行
|
||||
print(f"[WARN] 表格 {table_idx} 行 {row_idx} 处理出错: {str(e)}")
|
||||
table_errors += 1
|
||||
import traceback
|
||||
print(traceback.format_exc())
|
||||
continue
|
||||
except Exception as e:
|
||||
# 如果单个表格处理失败，记录错误但继续处理其他表格
|
||||
print(f"[WARN] 表格 {table_idx} 处理出错: {str(e)}")
|
||||
table_errors += 1
|
||||
import traceback
|
||||
print(traceback.format_exc())
|
||||
continue
|
||||
|
||||
print(f"[DEBUG] 表格替换统计: 成功 {table_replacements} 次, 错误 {table_errors} 次")
|
||||
|
||||
# 记录表格处理结果，稍后在保存后使用XML方法作为备用方案
|
||||
if table_errors > 0:
|
||||
print(f"[DEBUG] 检测到表格处理错误 {table_errors} 次，将在保存后使用XML方法作为备用方案")
|
||||
|
||||
except Exception as e:
|
||||
# 如果表格处理失败，记录错误但继续保存文档
|
||||
print(f"[WARN] 处理表格时出错: {str(e)}")
|
||||
import traceback
|
||||
print(traceback.format_exc())
|
||||
pass
|
||||
table_errors = 999 # 标记为严重错误
|
||||
|
||||
# 第三步：验证是否还有未替换的占位符（使用正则表达式匹配所有可能的占位符）
|
||||
remaining_placeholders = set()
|
||||
@ -598,99 +704,12 @@ class DocumentService:
|
||||
if field_code:
|
||||
remaining_placeholders.add(field_code)
|
||||
|
||||
# 检查表格中的占位符
|
||||
for table_idx, table in enumerate(doc.tables):
|
||||
try:
|
||||
if not table.rows:
|
||||
continue
|
||||
|
||||
# 安全地获取表格行数，使用索引方式访问行，而不是迭代器
|
||||
try:
|
||||
row_count = len(table.rows)
|
||||
except Exception as e:
|
||||
print(f"[WARN] 保存前验证表格 {table_idx} 时无法获取行数，跳过该表格: {str(e)}")
|
||||
continue
|
||||
|
||||
for row_idx in range(row_count):
|
||||
try:
|
||||
# 使用索引访问行，而不是迭代器
|
||||
row = table.rows[row_idx]
|
||||
|
||||
# 安全地访问 row.cells，避免 docx 库在处理异常表格结构时的 bug
|
||||
if not hasattr(row, 'cells'):
|
||||
continue
|
||||
|
||||
# 使用 try-except 包裹，防止 IndexError
|
||||
try:
|
||||
# 先尝试获取cells的数量
|
||||
try:
|
||||
cell_count = len(row.cells)
|
||||
except (IndexError, AttributeError):
|
||||
cell_count = 0
|
||||
|
||||
if cell_count == 0:
|
||||
continue
|
||||
|
||||
# 使用索引方式访问cells，而不是迭代器
|
||||
cells = []
|
||||
for cell_idx in range(cell_count):
|
||||
try:
|
||||
cell = row.cells[cell_idx]
|
||||
cells.append(cell)
|
||||
except (IndexError, AttributeError):
|
||||
# 如果某个单元格无法访问，跳过
|
||||
continue
|
||||
|
||||
if not cells:
|
||||
continue
|
||||
except (IndexError, AttributeError) as e:
|
||||
print(f"[WARN] 验证表格 {table_idx} 行 {row_idx} 时无法访问单元格，跳过该行: {str(e)}")
|
||||
continue
|
||||
|
||||
for cell_idx, cell in enumerate(cells):
|
||||
try:
|
||||
if not hasattr(cell, 'paragraphs'):
|
||||
continue
|
||||
|
||||
# 安全地获取paragraphs列表
|
||||
try:
|
||||
# 先尝试获取paragraphs的数量
|
||||
try:
|
||||
para_count = len(cell.paragraphs)
|
||||
except (IndexError, AttributeError):
|
||||
para_count = 0
|
||||
|
||||
paragraphs = []
|
||||
for para_idx in range(para_count):
|
||||
try:
|
||||
para = cell.paragraphs[para_idx]
|
||||
paragraphs.append(para)
|
||||
except (IndexError, AttributeError):
|
||||
continue
|
||||
except (IndexError, AttributeError) as e:
|
||||
print(f"[WARN] 验证表格 {table_idx} 行 {row_idx} 单元格 {cell_idx} 时无法访问段落，跳过: {str(e)}")
|
||||
continue
|
||||
|
||||
for paragraph in paragraphs:
|
||||
try:
|
||||
text = paragraph.text
|
||||
matches = placeholder_pattern.findall(text)
|
||||
for match in matches:
|
||||
field_code = match.strip()
|
||||
if field_code:
|
||||
remaining_placeholders.add(field_code)
|
||||
except Exception as e:
|
||||
print(f"[WARN] 验证表格 {table_idx} 行 {row_idx} 单元格 {cell_idx} 段落时出错，跳过: {str(e)}")
|
||||
continue
|
||||
except Exception as e:
|
||||
print(f"[WARN] 验证表格 {table_idx} 行 {row_idx} 单元格 {cell_idx} 时出错，跳过: {str(e)}")
|
||||
continue
|
||||
except Exception as e:
|
||||
print(f"[WARN] 验证表格 {table_idx} 行 {row_idx} 时出错，跳过: {str(e)}")
|
||||
continue
|
||||
except Exception as e:
|
||||
print(f"[WARN] 验证表格 {table_idx} 时出错，跳过该表格: {str(e)}")
|
||||
continue
|
||||
# 跳过表格验证，因为：
|
||||
# 1. 表格中的占位符已经在替换阶段被处理了
|
||||
# 2. 某些表格结构会导致索引越界错误（python-docx库的已知问题）
|
||||
# 3. 如果替换阶段成功，表格中的占位符应该已经被替换
|
||||
# 4. 验证阶段的错误不影响功能，只是无法统计表格中剩余的占位符
|
||||
print(f"[DEBUG] 跳过表格验证（表格中的占位符已在替换阶段处理，某些表格结构会导致验证错误）")
|
||||
|
||||
# 输出统计信息
|
||||
print(f"[DEBUG] 占位符替换统计:")
|
||||
@ -718,87 +737,8 @@ class DocumentService:
|
||||
if field_code:
|
||||
verification_placeholders.add(field_code)
|
||||
|
||||
for table_idx, table in enumerate(doc.tables):
|
||||
try:
|
||||
if not table.rows:
|
||||
continue
|
||||
|
||||
# 安全地获取表格行数，使用索引方式访问行
|
||||
try:
|
||||
row_count = len(table.rows)
|
||||
except Exception:
|
||||
continue
|
||||
|
||||
for row_idx in range(row_count):
|
||||
try:
|
||||
# 使用索引访问行，而不是迭代器
|
||||
row = table.rows[row_idx]
|
||||
|
||||
if not hasattr(row, 'cells'):
|
||||
continue
|
||||
|
||||
try:
|
||||
# 先尝试获取cells的数量
|
||||
try:
|
||||
cell_count = len(row.cells)
|
||||
except (IndexError, AttributeError):
|
||||
cell_count = 0
|
||||
|
||||
if cell_count == 0:
|
||||
continue
|
||||
|
||||
# 使用索引方式访问cells，而不是迭代器
|
||||
cells = []
|
||||
for cell_idx in range(cell_count):
|
||||
try:
|
||||
cell = row.cells[cell_idx]
|
||||
cells.append(cell)
|
||||
except (IndexError, AttributeError):
|
||||
continue
|
||||
|
||||
if not cells:
|
||||
continue
|
||||
except (IndexError, AttributeError):
|
||||
continue
|
||||
|
||||
for cell_idx, cell in enumerate(cells):
|
||||
try:
|
||||
if not hasattr(cell, 'paragraphs'):
|
||||
continue
|
||||
|
||||
# 安全地获取paragraphs列表
|
||||
try:
|
||||
try:
|
||||
para_count = len(cell.paragraphs)
|
||||
except (IndexError, AttributeError):
|
||||
para_count = 0
|
||||
|
||||
paragraphs = []
|
||||
for para_idx in range(para_count):
|
||||
try:
|
||||
para = cell.paragraphs[para_idx]
|
||||
paragraphs.append(para)
|
||||
except (IndexError, AttributeError):
|
||||
continue
|
||||
except (IndexError, AttributeError):
|
||||
continue
|
||||
|
||||
for paragraph in paragraphs:
|
||||
try:
|
||||
text = paragraph.text
|
||||
matches = placeholder_pattern.findall(text)
|
||||
for match in matches:
|
||||
field_code = match.strip()
|
||||
if field_code:
|
||||
verification_placeholders.add(field_code)
|
||||
except Exception:
|
||||
continue
|
||||
except Exception:
|
||||
continue
|
||||
except Exception:
|
||||
continue
|
||||
except Exception:
|
||||
continue
|
||||
# 跳过表格验证（表格中的占位符已在替换阶段处理）
|
||||
print(f"[DEBUG] 保存前验证：跳过表格验证（表格中的占位符已在替换阶段处理）")
|
||||
|
||||
if verification_placeholders:
|
||||
print(f"[WARN] 保存前验证发现仍有占位符: {sorted(verification_placeholders)}")
|
||||
@ -823,6 +763,14 @@ class DocumentService:
|
||||
|
||||
print(f"[DEBUG] 文件保存验证通过：文件大小 {file_size} 字节")
|
||||
|
||||
# 如果表格处理有错误，使用XML方法作为备用方案
|
||||
if use_xml_fallback:
|
||||
print(f"[DEBUG] 使用XML方法作为备用方案替换占位符...")
|
||||
if self.replace_placeholders_via_xml(output_file, field_data):
|
||||
print(f"[DEBUG] XML备用方案成功替换占位符")
|
||||
else:
|
||||
print(f"[WARN] XML备用方案替换失败")
|
||||
|
||||
# 验证保存的文件内容是否正确（重新打开文件检查）
|
||||
try:
|
||||
verify_doc = Document(output_file)
|
||||
@ -835,89 +783,8 @@ class DocumentService:
|
||||
if field_code:
|
||||
verify_placeholders_in_saved.add(field_code)
|
||||
|
||||
for table_idx, table in enumerate(verify_doc.tables):
|
||||
try:
|
||||
if not table.rows:
|
||||
continue
|
||||
|
||||
# 安全地获取表格行数
|
||||
try:
|
||||
row_count = len(table.rows)
|
||||
except Exception:
|
||||
continue
|
||||
|
||||
# 使用索引方式访问行，而不是迭代器
|
||||
for row_idx in range(row_count):
|
||||
try:
|
||||
# 使用索引访问行，而不是迭代器
|
||||
row = table.rows[row_idx]
|
||||
|
||||
if not hasattr(row, 'cells'):
|
||||
continue
|
||||
|
||||
try:
|
||||
# 先尝试获取cells的数量
|
||||
try:
|
||||
cell_count = len(row.cells)
|
||||
except (IndexError, AttributeError):
|
||||
cell_count = 0
|
||||
|
||||
if cell_count == 0:
|
||||
continue
|
||||
|
||||
# 使用索引方式访问cells，而不是迭代器
|
||||
cells = []
|
||||
for cell_idx in range(cell_count):
|
||||
try:
|
||||
cell = row.cells[cell_idx]
|
||||
cells.append(cell)
|
||||
except (IndexError, AttributeError):
|
||||
continue
|
||||
|
||||
if not cells:
|
||||
continue
|
||||
except (IndexError, AttributeError):
|
||||
continue
|
||||
|
||||
for cell_idx, cell in enumerate(cells):
|
||||
try:
|
||||
if not hasattr(cell, 'paragraphs'):
|
||||
continue
|
||||
|
||||
# 安全地获取paragraphs列表
|
||||
try:
|
||||
# 先尝试获取paragraphs的数量
|
||||
try:
|
||||
para_count = len(cell.paragraphs)
|
||||
except (IndexError, AttributeError):
|
||||
para_count = 0
|
||||
|
||||
paragraphs = []
|
||||
for para_idx in range(para_count):
|
||||
try:
|
||||
para = cell.paragraphs[para_idx]
|
||||
paragraphs.append(para)
|
||||
except (IndexError, AttributeError):
|
||||
continue
|
||||
except (IndexError, AttributeError):
|
||||
continue
|
||||
|
||||
for paragraph in paragraphs:
|
||||
try:
|
||||
text = paragraph.text
|
||||
matches = placeholder_pattern.findall(text)
|
||||
for match in matches:
|
||||
field_code = match.strip()
|
||||
if field_code:
|
||||
verify_placeholders_in_saved.add(field_code)
|
||||
except Exception:
|
||||
continue
|
||||
except Exception:
|
||||
continue
|
||||
except Exception:
|
||||
continue
|
||||
except Exception:
|
||||
continue
|
||||
# 跳过表格验证（表格中的占位符已在替换阶段处理）
|
||||
print(f"[DEBUG] 保存后验证：跳过表格验证（表格中的占位符已在替换阶段处理）")
|
||||
|
||||
if verify_placeholders_in_saved:
|
||||
print(f"[WARN] 保存后验证：文件中仍有占位符: {sorted(verify_placeholders_in_saved)}")
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user