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: check sql only contain read operation #814

Browse Source
This commit is contained in:
junjun
2026-01-29 14:22:16 +08:00
committed by Junjun
parent 184e76ebc2
commit bf56c3e486
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/apps/db/db.py
+4 -3
View File
@@ -577,10 +577,11 @@ def exec_sql(ds: CoreDatasource | AssistantOutDsSchema, sql: str, origin_column=
raise Exception(str(ex))
def check_sql_read(sql: str, dialect=None):
def check_sql_read(sql: str, ds: CoreDatasource | AssistantOutDsSchema):
try:
ansi_escape = re.compile(r'\x1b\[[0-9;]*m')
sql = ansi_escape.sub('', sql)
dialect = None
if ds.type == "mysql" or ds.type == "doris" or ds.type == "starrocks":
dialect = 'mysql'
statements = sqlglot.parse(sql, dialect=dialect)