生成数据库

lyg

2 天以前 22f370322412074174cde20ecfd14ec03657ab63

 knowledgebase/doc/models.py

@@ -6,6 +6,7 @@
# @description: 文档相关数据类
from dataclasses import dataclass
import typing
from enum import Enum

from knowledgebase.db.doc_db_models import TEntity

@@ -22,8 +23,11 @@
        num_level: int - 列表序号级别，0表示正文
        num: int - 列表序号，如果是列表
        children: typing.List[ParagraphInfo] - 子段落列表
        refs: 引用文档
        refs: 引用列表
        entities: 实体词列表
        is_toc: 是否是目录标志
    """
    id: int
    text: str
    title_level: int
    title_num: str
@@ -32,6 +36,15 @@
    children: typing.List
    refs: typing.List
    entities: typing.List[TEntity]
    is_toc: bool

    @property
    def title(self):
        if self.title_level:
            text = self.full_text
            idx = text.index('\n')
            return text[0:idx]
        return ''

    @property
    def full_text(self):
@@ -60,7 +73,7 @@
                full_text = full_text + "\n" + child.full_text_with_children
        return full_text

    def __init__(self, text: str, title_level: int, num=0, num_level=0):
    def __init__(self, text: str, title_level: int, num=0, num_level=0, is_toc=False):
        """
        段落信息

@@ -69,6 +82,7 @@
            title_level: int - 段落级别，1-9级标题，0表示正文
            num: int - 列表序号
            num_level: int - 列表序号级别
            is_toc: bool - 是否是目录
        """
        self.text = text
        self.title_level = title_level
@@ -77,6 +91,9 @@
        self.num_level = num_level
        self.children: typing.List[ParagraphInfo] = []
        self.entities: typing.List[TEntity] = []
        self.refs: typing.List[ParagraphInfo] = []
        self.is_toc = is_toc
        self.id = 0

    def __str__(self):
        return f"{self.full_text}"
@@ -115,3 +132,15 @@
        self.file = file
        self.file_type = file_type
        self.paragraphs: typing.List[ParagraphInfo] = paragraphs


class _DocType:
    tm_outline = '遥测大纲'
    user_requirements = '用户需求'
    tm_pkt_design = '源包设计'
    bus_comm_proto = '总线通信协议'
    tc_format = '指令格式'
    tc_cmd_table = '遥控指令表'


DocType = _DocType()