import json import sys from docx import Document from docx.shared import Inches def get_command_argv_by_sys(): # 默认值-用于测试 template_path = "template.docx" filename_path = "demo.docx" datafile_path = "data.json" # 读取参数 number = len(sys.argv) if 2 == number: template_path = sys.argv[1] if 3 == number: template_path = sys.argv[1] filename_path = sys.argv[2] if 4 == number: template_path = sys.argv[1] filename_path = sys.argv[2] datafile_path = sys.argv[3] return template_path, filename_path, datafile_path def read_data(filepath): content = [] if filepath != "": try: with open(filepath, "r", encoding="utf-8") as file: content = json.load(file) finally: return content def replace(template_path, filename_path, data_json): try: document_file = Document(template_path) except Exception: return 2 # 段落替换 for paragraph in document_file.paragraphs: paragraph.text = paragraph.text.replace(" ", "") find = False for datum in data_json: key = datum["key"] if key not in paragraph.text: continue find = True if datum["type"] == "text": runs = paragraph.runs for i in range(len(runs)): if key in runs[i].text: runs[i].text = runs[i].text.replace(key, str(datum["value"])) elif datum["type"] == "number": paragraph.text = paragraph.text.replace(key, str(datum["value"])) elif datum["type"] == "picture": # 清空整个段落并插入图片（假设占位符独占段落） paragraph.clear() width = float(datum.get("width", 2.0)) paragraph.add_run().add_picture(datum["value"], width=Inches(width)) elif datum["type"] == "list": p_elem = paragraph._p parent = p_elem.getparent() index = parent.index(p_elem) values = datum["value"] # 删除原段落 parent.remove(p_elem) # 为每个值插入新段落 for i, val in enumerate(values): new_para = document_file.add_paragraph() new_para.text = str(val) parent.insert(index + i, new_para._p) elif datum["type"] == "object_list": p_elem = paragraph._p parent = p_elem.getparent() index = parent.index(p_elem) values = datum["value"] # 删除原段落（占位符） parent.remove(p_elem) # 为每个对象插入新段落 for i, obj_item in enumerate(values): new_para = document_file.add_paragraph() item_type = obj_item.get("type") if item_type == "text": new_para.add_run(str(obj_item["value"])) elif item_type == "number": new_para.add_run(str(obj_item["value"])) elif item_type == "picture": width = float(obj_item.get("width", 2.0)) new_para.add_run().add_picture(obj_item["value"], width=Inches(width)) else: new_para.add_run(str(obj_item.get("value", ""))) # 插入到原位置 parent.insert(index + i, new_para._p) if not find: paragraph.text = "" # 表格替换 for table in document_file.tables: # 字段 group = None fields = [] repeat = False for row in table.rows: if repeat: break for cell in row.cells: if "." in cell.text: split = cell.text.replace("}","").split(".") group = split[0] + "}" field = split[1] fields.append(field.strip()) repeat = True else: for paragraph in cell.paragraphs: for run in paragraph.runs: find = False for datum in data_json: if datum["key"] in cell.text: find = True if "text" == datum["type"]: run.text = run.text.replace(datum["key"], datum["value"]) elif "picture" == datum["type"]: paragraph = cell.paragraphs[0] paragraph.clear() paragraph.add_run().add_picture(datum["value"], width=Inches(datum["width"])) elif "number" == datum["type"]: run.text = run.text.replace(datum["key"], str(datum["value"])) if not find and run.text.startswith("{") and run.text.endswith("}"): run.text = "" # 替换 if repeat: for datum in data_json: rows = len(table.rows) if datum["key"] in group: values = datum["value"] # 读取表格格式模板（保存每列的格式信息） column_formats = [] if rows > 0: template_row = table.rows[0] for i, cell in enumerate(template_row.cells): if i < len(fields): # 只需要处理与数据字段对应的列 if cell.paragraphs: para = cell.paragraphs[0] # 保存列格式信息 column_format = { 'font_name': '', 'font_size': None, 'alignment': para.alignment, 'bold': False, 'italic': False, 'underline': False, } # 优先使用run的格式（如果有） if para.runs: run = para.runs[0] if run.font.name: column_format['font_name'] = run.font.name if run.font.size: column_format['font_size'] = run.font.size if run.font.bold is not None: column_format['bold'] = run.font.bold if run.font.italic is not None: column_format['italic'] = run.font.italic if run.font.underline is not None: column_format['underline'] = run.font.underline # 保存字体颜色 if run.font.color.rgb: column_format['font_color'] = run.font.color.rgb # 如果没有run，使用样式的格式 else: if para.style.font.name: column_format['font_name'] = para.style.font.name if para.style.font.size: column_format['font_size'] = para.style.font.size if para.style.font.bold is not None: column_format['bold'] = para.style.font.bold if para.style.font.italic is not None: column_format['italic'] = para.style.font.italic if para.style.font.underline is not None: column_format['underline'] = para.style.font.underline column_formats.append(column_format) # 处理每一行数据 for (i, item) in enumerate(values): if i < rows: row = table.rows[i] else: row = table.add_row() # 复制行高 if i == 0 and rows > 0: row.height = template_row.height row.height_rule = template_row.height_rule # 处理每一列数据和格式 for z in range(0, len(fields)): field = fields[z] cell = row.cells[z] # 设置单元格内容 if item.get(field) is not None: cell.text = str(item.get(field)) else: cell.text = "" # 应用表格格式 if z < len(column_formats) and cell.paragraphs: format_info = column_formats[z] para = cell.paragraphs[0] # 设置段落对齐方式 para.alignment = format_info['alignment'] # 设置字体格式 if para.runs: run = para.runs[0] if format_info['font_name']: run.font.name = format_info['font_name'] if format_info['font_size']: run.font.size = format_info['font_size'] run.font.bold = format_info['bold'] run.font.italic = format_info['italic'] run.font.underline = format_info['underline'] # 设置字体颜色 if 'font_color' in format_info: run.font.color.rgb = format_info['font_color'] document_file.save(filename_path) return 0 def main(): params = get_command_argv_by_sys() [template, filename, datafile] = params data = read_data(datafile) if len(data) == 0: return 1 else: return replace(template, filename, data) if __name__ == '__main__': print(main())