From 5f0df02ef527c7f3996b7af9068300d244850e74 Mon Sep 17 00:00:00 2001
From: YM <479443481@qq.com>
Date: 星期三, 14 五月 2025 10:34:15 +0800
Subject: [PATCH] 拆分DB生成流程

---
 knowledgebase/gen_base_db/db_generate.py |   17 ++++++++++++++++-
 1 files changed, 16 insertions(+), 1 deletions(-)

diff --git a/knowledgebase/gen_base_db/db_generate.py b/knowledgebase/gen_base_db/db_generate.py
index cbfa28a..87422aa 100644
--- a/knowledgebase/gen_base_db/db_generate.py
+++ b/knowledgebase/gen_base_db/db_generate.py
@@ -7,4 +7,19 @@
 
 class DbGenerate:
     def __init__(self):
-        pass
\ No newline at end of file
+        pass
+    def run(self,type):
+        # 鏍规嵁鐢熸垚鐨勭粨鏋勫寲鏁版嵁锛屾搷浣滄暟鎹簱鐢熸垚DB
+        if type is not None:
+            if type == 1:
+                self.handle_yc_database()
+            if type == 2:
+                self.handle_yk_database()
+        else:
+            self.handle_yc_database()
+            self.handle_yk_database()
+    
+    # 閬ユ祴-start
+
+
+    # 閬ユ祴-end
\ No newline at end of file

--
Gitblit v1.9.1