From 74ae5e14264138ca014fa7ee77e38cfe856d08dc Mon Sep 17 00:00:00 2001
From: carry <2641257231@qq.com>
Date: Wed, 9 Apr 2025 08:12:59 +0800
Subject: [PATCH] =?UTF-8?q?refactor(db):=20=E9=87=8D=E5=91=BD=E5=90=8D?=
 =?UTF-8?q?=E6=95=B0=E6=8D=AE=E5=BA=93=E5=BC=95=E6=93=8E=E8=8E=B7=E5=8F=96?=
 =?UTF-8?q?=E5=87=BD=E6=95=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

将 get_engine 函数重命名为 get_sqlite_engine，以更清晰地表示其功能和用途。
- 更新了 db/__init__.py 中的导入和 __all__ 列表
- 修改了 db/init_db.py 中的函数定义
- 更新了前端设置页面和全局变量中的导入和函数调用

此更改提高了代码的可读性和维护性，特别是在将来可能添加其他类型数据库引擎的情况下。
---
 db/__init__.py           | 4 ++--
 db/init_db.py            | 4 ++--
 frontend/setting_page.py | 2 +-
 global_var.py            | 4 ++--
 4 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/db/__init__.py b/db/__init__.py
index aeed376..8fe1308 100644
--- a/db/__init__.py
+++ b/db/__init__.py
@@ -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']
\ No newline at end of file
+__all__ = ['get_sqlite_engine', 'initialize_sqlite_db']
\ No newline at end of file
diff --git a/db/init_db.py b/db/init_db.py
index f05df0b..2e49dc5 100644
--- a/db/init_db.py
+++ b/db/init_db.py
@@ -14,7 +14,7 @@ from schema.dataset_generation import APIProvider
 # 全局变量，用于存储数据库引擎实例
 _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")
     # 获取数据库引擎
-    engine = get_engine(workdir)
+    engine = get_sqlite_engine(workdir)
     # 初始化数据库
     initialize_sqlite_db(engine)
\ No newline at end of file
diff --git a/frontend/setting_page.py b/frontend/setting_page.py
index 437063f..a2b0885 100644
--- a/frontend/setting_page.py
+++ b/frontend/setting_page.py
@@ -1,7 +1,7 @@
 import gradio as gr
 from typing import List
 from sqlmodel import Session, select
-from db import get_engine
+from db import get_sqlite_engine
 from schema import APIProvider
 import os
 from global_var import sql_engine
diff --git a/global_var.py b/global_var.py
index 7a494b3..ee0ef40 100644
--- a/global_var.py
+++ b/global_var.py
@@ -1,3 +1,3 @@
-from db import get_engine
+from db import get_sqlite_engine
 
-sql_engine = get_engine("workdir")
\ No newline at end of file
+sql_engine = get_sqlite_engine("workdir")
\ No newline at end of file