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