129 lines
5.1 KiB
Python
129 lines
5.1 KiB
Python
from docx import Document
|
||
from docxtpl import DocxTemplate
|
||
from stuctor.constants import PREFIX
|
||
from docx.oxml.text.paragraph import CT_P
|
||
from docx.table import Table
|
||
from stuctor.item import TitleItem, TableItemList, Item
|
||
from nicegui import ui
|
||
from pathlib import Path
|
||
|
||
template_path = Path.cwd() / "templates"
|
||
|
||
class Operator:
|
||
"""
|
||
参数1: 传入被解析文档名称，不加.docx
|
||
参数2: 传入说明模版文档名称，不加.docx
|
||
参数3: 传入记录模版文档名称，不加.docx
|
||
属性1: item_list - 储存所有Item对象的列表
|
||
"""
|
||
# 标题过滤
|
||
TITLE_LIST = ['3', '30', '4', '40', '5', '50', '6', '60', '7', '70']
|
||
BASIC_CAPTER = '6.2'
|
||
|
||
def __init__(self, doc_name: str, tp_shuoming_name: str, tp_jilu_name: str, logger: ui.log) -> None:
|
||
# 记录器
|
||
self.logger = logger
|
||
# 被解析文档对象
|
||
self.doc = Document(doc_name)
|
||
# 说明模版文件
|
||
self.tp_shuom = DocxTemplate(template_path / (tp_shuoming_name+'.docx'))
|
||
# 记录模版文件
|
||
self.tp_jilu = DocxTemplate(template_path / (tp_jilu_name+'.docx'))
|
||
# 储存所有item对象的列表
|
||
self.item_list: list[Item] = []
|
||
# 储存大纲章节号排布
|
||
self.c3 = 0
|
||
self.c4 = 0
|
||
self.c5 = 0
|
||
self.c6 = 0
|
||
self.c7 = 0
|
||
self.last_title_item = None
|
||
|
||
def main_parse(self, parse_content_strategy: str = 'FPGA'):
|
||
"""默认是未解析的，调用该函数开始解析"""
|
||
# 解析开始-这里有点耦合了，其实该请TitleItem对象去处理的
|
||
for ele in self.doc._element.body.iter(): # type: ignore
|
||
if ele.tag.endswith('}p'):
|
||
for child in ele.iter():
|
||
if child.tag.endswith("pStyle"):
|
||
rank = child.get(f"{PREFIX}val")
|
||
title = self.__parse_ele_title(ele)
|
||
# 标题只有30，40，5，6
|
||
if rank in self.TITLE_LIST:
|
||
current_title_item = TitleItem(rank, title)
|
||
self.item_list.append(current_title_item)
|
||
# 解析当前章节号
|
||
self.__parse_current_capter(current_title_item)
|
||
elif ele.tag.endswith('}tbl'):
|
||
table = Table(ele, self.doc)
|
||
if table.cell(0, 0).text == "测试项名称":
|
||
self.item_list.append(TableItemList(table, self.last_title_item, self.logger, parse_content_strategy))
|
||
|
||
# 私有：解析标题文字
|
||
def __parse_ele_title(self, ele: CT_P):
|
||
t_ele = ele.findall(f".//{PREFIX}t") # type: ignore
|
||
title = ''
|
||
for i in range(len(t_ele)):
|
||
if not t_ele[i].text.isdigit():
|
||
title = title + t_ele[i].text
|
||
return title
|
||
|
||
# 私有：解析当前章节号-根据其rank
|
||
def __parse_current_capter(self, current_title_item: TitleItem):
|
||
rank = current_title_item.type
|
||
if rank == '30' or rank == '3':
|
||
self.c3 += 1
|
||
self.c4 = 0
|
||
self.c5 = 0
|
||
self.c6 = 0
|
||
self.c7 = 0
|
||
current_title_item.capter = ".".join([self.BASIC_CAPTER, str(self.c3)])
|
||
elif rank == '40' or rank == '4':
|
||
self.c4 += 1
|
||
self.c5 = 0
|
||
self.c6 = 0
|
||
self.c7 = 0
|
||
current_title_item.capter = '.'.join([self.BASIC_CAPTER, str(self.c3), str(self.c4)])
|
||
elif rank == '50' or rank == '5':
|
||
self.c5 += 1
|
||
self.c6 = 0
|
||
self.c7 = 0
|
||
current_title_item.capter = '.'.join([self.BASIC_CAPTER, str(self.c3), str(self.c4), str(self.c5)])
|
||
elif rank == '60' or rank == '6':
|
||
self.c6 += 1
|
||
self.c7 = 0
|
||
current_title_item.capter = '.'.join([self.BASIC_CAPTER, str(self.c3), str(self.c4), str(self.c6)])
|
||
elif rank == '70' or rank == '7':
|
||
self.c7 += 1
|
||
current_title_item.capter = '.'.join([self.BASIC_CAPTER, str(self.c3), str(self.c4), str(self.c6), str(self.c7)])
|
||
|
||
self.last_title_item = current_title_item
|
||
|
||
# 最后渲染
|
||
def render(self):
|
||
context = self.__create_context()
|
||
self.tp_shuom.render(context, autoescape = True)
|
||
self.tp_shuom.save("测试说明-输出.docx")
|
||
self.tp_jilu.render(context, autoescape = True)
|
||
self.tp_jilu.save("测试记录-输出.docx")
|
||
|
||
# 构造context
|
||
def __create_context(self):
|
||
data_list = []
|
||
for item in self.item_list:
|
||
if isinstance(item, TableItemList):
|
||
for it in item.table_item:
|
||
temp_item = {
|
||
'type': 'table',
|
||
'name': it.name,
|
||
'ident': it.ident,
|
||
'zhui': item.zhui,
|
||
'zong': it.zong,
|
||
'step': it.step,
|
||
'csh': it.csh
|
||
}
|
||
data_list.append(temp_item)
|
||
elif isinstance(item, TitleItem):
|
||
data_list.append({'type': item.type, 'title': item.title})
|
||
return {"datas": data_list}
|