mysql和pgsql支持with查询

hhyo · Jan 17, 2025 · 82c4042 · 82c4042
1 parent 69e29b1
commit 82c4042
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/sql/engines/mysql.py b/sql/engines/mysql.py
@@ -559,7 +559,7 @@ def query_check(self, db_name=None, sql=""):
         except IndexError:
             result["bad_query"] = True
             result["msg"] = "没有有效的SQL语句"
-        if re.match(r"^select|^show|^explain", sql, re.I) is None:
+        if re.match(r"^select|^show|^explain|^with", sql, re.I) is None:
             result["bad_query"] = True
             result["msg"] = "不支持的查询语法类型!"
         if "*" in sql:

diff --git a/sql/engines/pgsql.py b/sql/engines/pgsql.py
@@ -165,7 +165,7 @@ def query_check(self, db_name=None, sql=""):
         except IndexError:
             result["bad_query"] = True
             result["msg"] = "没有有效的SQL语句"
-        if re.match(r"^select|^explain", sql, re.I) is None:
+        if re.match(r"^select|^explain|^with", sql, re.I) is None:
             result["bad_query"] = True
             result["msg"] = "不支持的查询语法类型!"
         if "*" in sql: