refactor(db): 重命名数据库引擎加载函数
- 将 get_sqlite_engine 函数重命名为 load_sqlite_engine - 更新了相关模块中的导入和调用 - 这个改动是为了更好地反映函数的实际功能,提高代码可读性
This commit is contained in:
parent
7242a2ce03
commit
feaea1fb64
@ -1,9 +1,9 @@
|
|||||||
from .init_db import get_sqlite_engine, initialize_sqlite_db
|
from .init_db import load_sqlite_engine, initialize_sqlite_db
|
||||||
from .prompt_store import get_prompt_tinydb, initialize_prompt_store
|
from .prompt_store import get_prompt_tinydb, initialize_prompt_store
|
||||||
from .dataset_store import get_all_dataset
|
from .dataset_store import get_all_dataset
|
||||||
|
|
||||||
__all__ = [
|
__all__ = [
|
||||||
"get_sqlite_engine",
|
"load_sqlite_engine",
|
||||||
"initialize_sqlite_db",
|
"initialize_sqlite_db",
|
||||||
"get_prompt_tinydb",
|
"get_prompt_tinydb",
|
||||||
"initialize_prompt_store",
|
"initialize_prompt_store",
|
||||||
|
@ -14,7 +14,7 @@ from schema.dataset_generation import APIProvider
|
|||||||
# 全局变量,用于存储数据库引擎实例
|
# 全局变量,用于存储数据库引擎实例
|
||||||
_engine: Optional[Engine] = None
|
_engine: Optional[Engine] = None
|
||||||
|
|
||||||
def get_sqlite_engine(workdir: str) -> Engine:
|
def load_sqlite_engine(workdir: str) -> Engine:
|
||||||
"""
|
"""
|
||||||
获取数据库引擎实例。如果引擎尚未创建,则创建一个新的引擎并返回。
|
获取数据库引擎实例。如果引擎尚未创建,则创建一个新的引擎并返回。
|
||||||
|
|
||||||
@ -74,6 +74,6 @@ if __name__ == "__main__":
|
|||||||
# 定义工作目录路径
|
# 定义工作目录路径
|
||||||
workdir = os.path.join(os.path.dirname(__file__), "..", "workdir")
|
workdir = os.path.join(os.path.dirname(__file__), "..", "workdir")
|
||||||
# 获取数据库引擎
|
# 获取数据库引擎
|
||||||
engine = get_sqlite_engine(workdir)
|
engine = load_sqlite_engine(workdir)
|
||||||
# 初始化数据库
|
# 初始化数据库
|
||||||
initialize_sqlite_db(engine)
|
initialize_sqlite_db(engine)
|
@ -1,4 +1,4 @@
|
|||||||
from db import get_sqlite_engine, get_prompt_tinydb, get_all_dataset
|
from db import load_sqlite_engine, get_prompt_tinydb, get_all_dataset
|
||||||
from tools import scan_docs_directory
|
from tools import scan_docs_directory
|
||||||
|
|
||||||
_prompt_store = None
|
_prompt_store = None
|
||||||
@ -10,7 +10,7 @@ _workdir = None
|
|||||||
def init_global_var(workdir="workdir"):
|
def init_global_var(workdir="workdir"):
|
||||||
global _prompt_store, _sql_engine, _datasets, _workdir
|
global _prompt_store, _sql_engine, _datasets, _workdir
|
||||||
_prompt_store = get_prompt_tinydb(workdir)
|
_prompt_store = get_prompt_tinydb(workdir)
|
||||||
_sql_engine = get_sqlite_engine(workdir)
|
_sql_engine = load_sqlite_engine(workdir)
|
||||||
_datasets = get_all_dataset(workdir)
|
_datasets = get_all_dataset(workdir)
|
||||||
_workdir = workdir
|
_workdir = workdir
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user