From 22f370322412074174cde20ecfd14ec03657ab63 Mon Sep 17 00:00:00 2001
From: lyg <1543117173@qq.com>
Date: 星期一, 07 七月 2025 16:20:25 +0800
Subject: [PATCH] 生成数据库
---
knowledgebase/db/models.py | 14 ++++++++++----
1 files changed, 10 insertions(+), 4 deletions(-)
diff --git a/knowledgebase/db/models.py b/knowledgebase/db/models.py
index 9577b28..db509dd 100644
--- a/knowledgebase/db/models.py
+++ b/knowledgebase/db/models.py
@@ -473,8 +473,14 @@
C_ACCESS = Column(Integer)
C_EDIT = Column(Integer)
+engine = None
-if os.path.isfile("db.db"):
- os.remove("db.db")
-engine = create_engine('sqlite:///db.db', echo=True)
-metadata.create_all(engine)
+def get_engine():
+ return engine
+
+def init_base_db(db_path: str):
+ global engine, metadata
+ if os.path.isfile(db_path):
+ os.remove(db_path)
+ engine = create_engine(f'sqlite:///{db_path}', echo=False)
+ metadata.create_all(engine)
--
Gitblit v1.9.1