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琛ㄧず姝ｆ枃
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琛ㄧず姝ｆ枃
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