117 lines
4.5 KiB
Python
117 lines
4.5 KiB
Python
|
from typing import Dict
|
|||
|
|
import os
|
|||
|
|
from docx import Document
|
|||
|
|
from tools.document_tools import add_picture_to_table, add_table_to_document
|
|||
|
|
def add_jf_picture_table(
|
|||
|
|
typical_picture_dict: Dict[str, str],
|
|||
|
|
defect_picture_dict: Dict[str, str],
|
|||
|
|
TYPICAL_MUBAN_DIR: str,
|
|||
|
|
DEFECT_MUBAN_DIR: str,
|
|||
|
|
output_dir: str,
|
|||
|
|
):
|
|||
|
|
"""添加金风版本的图片展示表格"""
|
|||
|
|
# 初始化文档对象
|
|||
|
|
doc = Document()
|
|||
|
|
target_filename = os.path.join(output_dir, "output.docx")
|
|||
|
|
|
|||
|
|
# 处理典型图
|
|||
|
|
typical_data = []
|
|||
|
|
defect_keys = [] # 记录有缺陷的典型图key
|
|||
|
|
|
|||
|
|
# 准备典型图数据
|
|||
|
|
for key, value in typical_picture_dict.items():
|
|||
|
|
if "损伤" in key or "损伤" in value:
|
|||
|
|
# 情况二：有缺陷
|
|||
|
|
defect_count = int(value.split("损伤类型")[1].split("处")[0])
|
|||
|
|
typical_data.append([key, "损伤{}处，详见下表".format(defect_count), value])
|
|||
|
|
defect_keys.append((key, defect_count))
|
|||
|
|
else:
|
|||
|
|
# 情况一：正常图片
|
|||
|
|
typical_data.append([key, value, ""]) # 第三行留空
|
|||
|
|
|
|||
|
|
# 添加典型图表格
|
|||
|
|
if typical_data:
|
|||
|
|
# 将数据转换为适合表格的格式（3行×n列）
|
|||
|
|
# 需要将数据从行优先转为列优先
|
|||
|
|
cols = len(typical_data)
|
|||
|
|
table_data = [[], [], []]
|
|||
|
|
for col_data in typical_data:
|
|||
|
|
for i in range(3):
|
|||
|
|
table_data[i].append(col_data[i] if i < len(col_data) else "")
|
|||
|
|
|
|||
|
|
# 添加典型图表格
|
|||
|
|
add_table_to_document(
|
|||
|
|
target_filename=target_filename,
|
|||
|
|
source_filename=TYPICAL_MUBAN_DIR,
|
|||
|
|
rows=3,
|
|||
|
|
cols=cols,
|
|||
|
|
table_num=0,
|
|||
|
|
data=table_data
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 添加典型图中的图片
|
|||
|
|
doc = Document(target_filename)
|
|||
|
|
for col_idx, (key, value) in enumerate(typical_picture_dict.items()):
|
|||
|
|
if "损伤" not in key and "损伤" not in value: # 只有正常图片才添加
|
|||
|
|
add_picture_to_table(
|
|||
|
|
target_doc=doc,
|
|||
|
|
target_filename=target_filename,
|
|||
|
|
row=1, # 第二行是图片
|
|||
|
|
col=col_idx,
|
|||
|
|
image_path=value
|
|||
|
|
)
|
|||
|
|
doc.save(target_filename)
|
|||
|
|
|
|||
|
|
# 处理缺陷图
|
|||
|
|
if defect_keys:
|
|||
|
|
# 遍历所有有缺陷的典型图
|
|||
|
|
for typical_key, defect_count in defect_keys:
|
|||
|
|
# 计算需要多少个缺陷图表格（每表5列）
|
|||
|
|
table_count = (defect_count + 4) // 5 # 向上取整
|
|||
|
|
|
|||
|
|
for table_idx in range(table_count):
|
|||
|
|
# 计算当前表格的列数
|
|||
|
|
cols_in_table = min(5, defect_count - table_idx * 5)
|
|||
|
|
|
|||
|
|
# 准备缺陷图数据
|
|||
|
|
defect_data = []
|
|||
|
|
for i in range(cols_in_table):
|
|||
|
|
defect_idx = table_idx * 5 + i
|
|||
|
|
defect_key = list(defect_picture_dict.keys())[defect_idx]
|
|||
|
|
defect_value = list(defect_picture_dict.values())[defect_idx]
|
|||
|
|
defect_data.append([defect_key, defect_value])
|
|||
|
|
|
|||
|
|
# 将数据转换为适合表格的格式（2行×n列）
|
|||
|
|
table_data = [[], []]
|
|||
|
|
for col_data in defect_data:
|
|||
|
|
for i in range(2):
|
|||
|
|
table_data[i].append(col_data[i] if i < len(col_data) else "")
|
|||
|
|
|
|||
|
|
# 添加缺陷图表格
|
|||
|
|
add_table_to_document(
|
|||
|
|
target_filename=target_filename,
|
|||
|
|
source_filename=DEFECT_MUBAN_DIR,
|
|||
|
|
rows=2,
|
|||
|
|
cols=cols_in_table,
|
|||
|
|
table_num=0,
|
|||
|
|
data=table_data
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 添加缺陷图中的图片
|
|||
|
|
doc = Document(target_filename)
|
|||
|
|
for col_idx in range(cols_in_table):
|
|||
|
|
defect_idx = table_idx * 5 + col_idx
|
|||
|
|
defect_value = list(defect_picture_dict.values())[defect_idx]
|
|||
|
|
add_picture_to_table(
|
|||
|
|
target_doc=doc,
|
|||
|
|
target_filename=target_filename,
|
|||
|
|
row=1, # 第二行是图片
|
|||
|
|
col=col_idx,
|
|||
|
|
image_path=defect_value
|
|||
|
|
)
|
|||
|
|
doc.save(target_filename)
|
|||
|
|
|
|||
|
|
return target_filename
|
|||
|
|
|
|||
|
|
|
|||
|
|
add_jf_picture_table({""})
|