From ef20fbf8f462807c86417d668fdc4d4a601f1c8f Mon Sep 17 00:00:00 2001 From: junjun Date: Mon, 2 Feb 2026 17:39:57 +0800 Subject: [PATCH] fix: check sql only contain read operation --- backend/apps/db/db.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/backend/apps/db/db.py b/backend/apps/db/db.py index 0334b060..29885332 100644 --- a/backend/apps/db/db.py +++ b/backend/apps/db/db.py @@ -592,9 +592,9 @@ def exec_sql(ds: CoreDatasource | AssistantOutDsSchema, sql: str, origin_column= def check_sql_read(sql: str, ds: CoreDatasource | AssistantOutDsSchema): try: dialect = None - if ds.type == "mysql" or ds.type == "doris" or ds.type == "starrocks": + if equals_ignore_case(ds.type, 'mysql', 'doris', 'starrocks'): dialect = 'mysql' - elif ds.type == "sqlServer": + elif equals_ignore_case(ds.type, 'sqlServer'): dialect = 'tsql' statements = sqlglot.parse(sql, dialect=dialect)