From 22f370322412074174cde20ecfd14ec03657ab63 Mon Sep 17 00:00:00 2001
From: lyg <1543117173@qq.com>
Date: 星期一, 07 七月 2025 16:20:25 +0800
Subject: [PATCH] 生成数据库

---
 knowledgebase/doc/models.py |   20 ++++++++++++++++++--
 1 files changed, 18 insertions(+), 2 deletions(-)

diff --git a/knowledgebase/doc/models.py b/knowledgebase/doc/models.py
index 7eacdc8..c4daaf5 100644
--- a/knowledgebase/doc/models.py
+++ b/knowledgebase/doc/models.py
@@ -23,8 +23,11 @@
         num_level: int - 鍒楄〃搴忓彿绾у埆锛�0琛ㄧず姝f枃
         num: int - 鍒楄〃搴忓彿锛屽鏋滄槸鍒楄〃
         children: typing.List[ParagraphInfo] - 瀛愭钀藉垪琛�
-        refs: 寮曠敤鏂囨。
+        refs: 寮曠敤鍒楄〃
+        entities: 瀹炰綋璇嶅垪琛�
+        is_toc: 鏄惁鏄洰褰曟爣蹇�
     """
+    id: int
     text: str
     title_level: int
     title_num: str
@@ -33,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):
@@ -61,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):
         """
         娈佃惤淇℃伅
 
@@ -70,6 +82,7 @@
             title_level: int - 娈佃惤绾у埆锛�1-9绾ф爣棰橈紝0琛ㄧず姝f枃
             num: int - 鍒楄〃搴忓彿
             num_level: int - 鍒楄〃搴忓彿绾у埆
+            is_toc: bool - 鏄惁鏄洰褰�
         """
         self.text = text
         self.title_level = title_level
@@ -78,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}"

--
Gitblit v1.9.1