refactor(db): 重命名数据库引擎获取函数
将 get_engine 函数重命名为 get_sqlite_engine,以更清晰地表示其功能和用途。 - 更新了 db/__init__.py 中的导入和 __all__ 列表 - 修改了 db/init_db.py 中的函数定义 - 更新了前端设置页面和全局变量中的导入和函数调用 此更改提高了代码的可读性和维护性,特别是在将来可能添加其他类型数据库引擎的情况下。
This commit is contained in:
parent
0a6ae7a4ee
commit
74ae5e1426
@ -1,3 +1,3 @@
|
|||||||
from .init_db import get_engine, initialize_sqlite_db
|
from .init_db import get_sqlite_engine, initialize_sqlite_db
|
||||||
|
|
||||||
__all__ = ['get_engine', 'initialize_sqlite_db']
|
__all__ = ['get_sqlite_engine', 'initialize_sqlite_db']
|
@ -14,7 +14,7 @@ from schema.dataset_generation import APIProvider
|
|||||||
# 全局变量,用于存储数据库引擎实例
|
# 全局变量,用于存储数据库引擎实例
|
||||||
_engine: Optional[Engine] = None
|
_engine: Optional[Engine] = None
|
||||||
|
|
||||||
def get_engine(workdir: str) -> Engine:
|
def get_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_engine(workdir)
|
engine = get_sqlite_engine(workdir)
|
||||||
# 初始化数据库
|
# 初始化数据库
|
||||||
initialize_sqlite_db(engine)
|
initialize_sqlite_db(engine)
|
@ -1,7 +1,7 @@
|
|||||||
import gradio as gr
|
import gradio as gr
|
||||||
from typing import List
|
from typing import List
|
||||||
from sqlmodel import Session, select
|
from sqlmodel import Session, select
|
||||||
from db import get_engine
|
from db import get_sqlite_engine
|
||||||
from schema import APIProvider
|
from schema import APIProvider
|
||||||
import os
|
import os
|
||||||
from global_var import sql_engine
|
from global_var import sql_engine
|
||||||
|
@ -1,3 +1,3 @@
|
|||||||
from db import get_engine
|
from db import get_sqlite_engine
|
||||||
|
|
||||||
sql_engine = get_engine("workdir")
|
sql_engine = get_sqlite_engine("workdir")
|
Loading…
x
Reference in New Issue
Block a user