fixbug

pengkang1991 · Nov 29, 2023 · 15fa320 · 15fa320
1 parent 9e729b3
commit 15fa320
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/sql/engines/mssql.py b/sql/engines/mssql.py
@@ -311,15 +311,15 @@ def query_check(self, db_name=None, sql=""):
             result["msg"] = keyword_warning
         return result
 
-def filter_sql(sql="", limit_num=0):
-    sql_lower = sql.lower()
-    # 对查询sql增加limit限制
-    if re.match(r"^select", sql_lower):
-        if sql_lower.find(" top ") == -1:
-            if sql_lower.find(" distinct ") > 0:
-                return sql_lower.replace("distinct", "distinct top {}".format(limit_num))
-            return sql_lower.replace("select", "select top {}".format(limit_num))
-    return sql.strip()
+    def filter_sql(sql="", limit_num=0):
+        sql_lower = sql.lower()
+        # 对查询sql增加limit限制
+        if re.match(r"^select", sql_lower):
+            if sql_lower.find(" top ") == -1:
+                if sql_lower.find(" distinct ") > 0:
+                    return sql_lower.replace("distinct", "distinct top {}".format(limit_num))
+                return sql_lower.replace("select", "select top {}".format(limit_num))
+        return sql.strip()
 
     def query(
         self,