155 lines
7.6 KiB
Python
155 lines
7.6 KiB
Python
from ninja_extra import api_controller, route, ControllerBase
|
||
from django.db import transaction
|
||
from pathlib import Path
|
||
from django.shortcuts import get_object_or_404
|
||
# 文档处理相关库
|
||
from docxtpl import DocxTemplate
|
||
from docx import Document
|
||
from docx.text.paragraph import Paragraph
|
||
from docx.table import Table
|
||
from docx.oxml.table import CT_Tbl
|
||
from docx.oxml.text.paragraph import CT_P
|
||
from docx.oxml.text.run import CT_R
|
||
from docx.oxml.shape import CT_Picture
|
||
from docx.parts.image import ImagePart
|
||
from docx.text.run import Run
|
||
from docx.shared import Cm
|
||
from docx.enum.text import WD_PARAGRAPH_ALIGNMENT
|
||
|
||
@route.get('/smDocument', url_name='create-smDocument')
|
||
@transaction.atomic
|
||
def create_smDocument(self, id: int):
|
||
"""生成最后说明文档"""
|
||
# 获取项目对象
|
||
project_obj = get_object_or_404(Project, id=id)
|
||
# 首先第二层模版所需变量
|
||
member = project_obj.member[0] if len(project_obj.member) > 0 else project_obj.duty_person
|
||
context = {'name': project_obj.name, 'is_JD': False, 'ident': project_obj.ident, 'sec_title': "公开",
|
||
'duty_person': project_obj.duty_person, 'member': member}
|
||
if project_obj.report_type == '9':
|
||
context['is_JD'] = True
|
||
# 提取第一轮测试中源代码 - 用户标识
|
||
round_1 = project_obj.pField.filter(key='0').first()
|
||
duty_so = round_1.rdField.filter(type='SO').first()
|
||
if not duty_so:
|
||
return ChenResponse(code=400, status=400, message="未找到第一轮测试中源代码被测件请添加")
|
||
context['user_ident'] = duty_so.ref
|
||
|
||
# ~~~~~~定义模版路径~~~~~~~
|
||
## 定义说明最后模版
|
||
sm_template_file = Path.cwd() / 'media' / 'form_template' / 'products' / '测试说明.docx'
|
||
# docx生成后给docxtpl文件路径
|
||
sm_to_tpl_file = Path.cwd() / 'media' / 'temp' / '测试说明.docx'
|
||
# 生成最后文档路径
|
||
sm_seitai_final_file = Path.cwd() / 'media' / 'final_seitai' / '测试说明.docx'
|
||
# output_dir根路径
|
||
output_files_path = Path.cwd() / 'media' / 'output_dir'
|
||
# 获取可能使用的被复制的docx文件
|
||
dg_copied_files = []
|
||
sm_copied_files = []
|
||
for file in output_files_path.iterdir():
|
||
if file.is_file():
|
||
if file.suffix == '.docx':
|
||
dg_copied_files.append(file)
|
||
elif file.is_dir():
|
||
if file.stem == 'sm':
|
||
for f in file.iterdir():
|
||
if f.suffix == '.docx':
|
||
sm_copied_files.append(f)
|
||
# 打开文档找到所有的“域”
|
||
doc = Document(sm_template_file.as_posix())
|
||
body = doc.element.body
|
||
sdt_element_list = body.xpath('./w:sdt')
|
||
# 找到sdt域的名称 -> 为了对应output_dir文件 / 储存所有output_dir图片
|
||
area_name_list = []
|
||
image_part_list = []
|
||
# 遍历所有控件 -> 放入area_name_list【这里准备提取公共代码】
|
||
for sdt_ele in sdt_element_list:
|
||
for elem in sdt_ele.iterchildren():
|
||
# 获取“域”的名称
|
||
if elem.tag.endswith('sdtPr'):
|
||
for el in elem.getchildren():
|
||
if el.tag.endswith('alias'):
|
||
if len(el.attrib.values()) > 0:
|
||
area_name = el.attrib.values()[0]
|
||
area_name_list.append(area_name)
|
||
# 开始替换里面的“域”
|
||
if elem.tag.endswith('sdtContent'):
|
||
elem.clear()
|
||
if len(area_name_list) > 0:
|
||
area_pop_name = area_name_list.pop(0)
|
||
# 取到“域名称”，这里先去找media/output_dir/sm下文件，然后找media/output下文件
|
||
copied_file_path = ""
|
||
for file in sm_copied_files:
|
||
if file.stem == area_pop_name:
|
||
copied_file_path = file
|
||
# 这里判断是否copied_file_path没取到文件，然后遍历output_dir下文件
|
||
if not copied_file_path:
|
||
for file in dg_copied_files:
|
||
if file.stem == area_pop_name:
|
||
copied_file_path = file
|
||
# 找到所需的文件，将其数据复制到对应area_name的“域”
|
||
if copied_file_path:
|
||
doc_copied = Document(copied_file_path)
|
||
copied_element_list = []
|
||
element_list = doc_copied.element.body.inner_content_elements
|
||
for elet in element_list:
|
||
if isinstance(elet, CT_P):
|
||
copied_element_list.append(Paragraph(elet, doc_copied))
|
||
if isinstance(elet, CT_Tbl):
|
||
copied_element_list.append(Table(elet, doc_copied))
|
||
for para_copied in copied_element_list:
|
||
elem.append(para_copied._element)
|
||
|
||
# 下面代码就是将图片全部提取到image_part_list，以便后续插入
|
||
doc_copied = Document(copied_file_path) # 需要重新获取否则namespace错误
|
||
copied_body = doc_copied.element.body
|
||
img_node_list = copied_body.xpath('.//pic:pic')
|
||
if not img_node_list:
|
||
pass
|
||
else:
|
||
for img_node in img_node_list:
|
||
img: CT_Picture = img_node
|
||
# 根据节点找到图片的关联id
|
||
embed = img.xpath('.//a:blip/@r:embed')[0]
|
||
# 这里得到ImagePart -> 马上要给新文档添加
|
||
related_part: ImagePart = doc_copied.part.related_parts[embed]
|
||
# doc_copied.part.related_parts是一个字典
|
||
image_part_list.append(related_part)
|
||
|
||
# 新文档添加sdt_element_list的图片
|
||
graph_node_list = body.xpath('.//pic:pic')
|
||
img_count = 0
|
||
if len(graph_node_list) == len(image_part_list):
|
||
for graph_node in graph_node_list:
|
||
image_run_node = getParentRunNode(graph_node)
|
||
image_run_node.clear()
|
||
copied_bytes_io = BytesIO(image_part_list[img_count].image.blob)
|
||
r_element = Run(image_run_node, doc)
|
||
inline_shape = r_element.add_picture(copied_bytes_io)
|
||
# 设置图片位置尺寸
|
||
source_width = inline_shape.width
|
||
inline_shape.width = Cm(12)
|
||
inline_shape.height = int(inline_shape.height * (inline_shape.width / source_width))
|
||
# 设置图片所在段落居中对齐
|
||
r_element.alignment = WD_PARAGRAPH_ALIGNMENT.CENTER
|
||
img_count += 1
|
||
else:
|
||
return ChenResponse(code=400, status=400, message='注意模版里面有自定义图片，请删除后重试!!!')
|
||
|
||
try:
|
||
doc.save(sm_to_tpl_file)
|
||
except PermissionError as e:
|
||
return ChenResponse(status=400, code=400, message="注意你打开了生成的文档，请关闭后再试，{0}".format(e))
|
||
|
||
doc = DocxTemplate(sm_to_tpl_file)
|
||
start = time.time()
|
||
doc.render(context) # 耗时最长，TODO:异步任务处理？或前端等待？
|
||
end = time.time()
|
||
print('渲染耗时：', end - start)
|
||
try:
|
||
doc.save(sm_seitai_final_file)
|
||
return ChenResponse(status=200, code=200, message="最终大纲生成成功！")
|
||
except PermissionError as e:
|
||
return ChenResponse(status=400, code=400, message="模版文件已打开，请关闭后再试，{0}".format(e))
|