klinsmaya/SQLBot
1
0
Fork 0
mirror of https://gh-proxy.org/https://github.com/klinsmaya/SQLBot.git synced 2026-11-08 01:06:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: fix mysql check status error

Browse Source
This commit is contained in:
junjun
2025-09-01 10:24:13 +08:00
parent a0ca17d3af
commit ad16f6221d
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/apps/db/db.py
+4 -1
View File
@@ -106,6 +106,8 @@ def check_connection(trans: Trans, ds: CoreDatasource, is_raise: bool = False):
conn = get_engine(ds, 10)
try:
with conn.connect() as connection:
if ds.type == 'mysql':
connection.execute(text('select 1')).fetchall()
SQLBotLogUtil.info("success")
return True
except Exception as e:
@@ -158,7 +160,8 @@ def check_connection(trans: Trans, ds: CoreDatasource, is_raise: bool = False):
def get_version(ds: CoreDatasource | AssistantOutDsSchema):
conf = None
if isinstance(ds, CoreDatasource):
conf = DatasourceConf(**json.loads(aes_decrypt(ds.configuration))) if ds.type != "excel" else get_engine_config()
conf = DatasourceConf(
**json.loads(aes_decrypt(ds.configuration))) if ds.type != "excel" else get_engine_config()
if isinstance(ds, AssistantOutDsSchema):
conf = DatasourceConf()
conf.host = ds.host