解决clickhouse表名大小写敏感问题

hhyo · May 10, 2024 · 823b3fc · 823b3fc
1 parent b61bdf0
commit 823b3fc
Showing 1 changed file with 17 additions and 11 deletions.
diff --git a/sql/engines/clickhouse.py b/sql/engines/clickhouse.py
@@ -255,7 +255,7 @@ def execute_check(self, db_name=None, sql=""):
         for statement in sql_list:
             statement = statement.rstrip(";")
             # 禁用语句
-            if re.match(r"^select|^show", statement.lower()):
+            if re.match(r"^select|^show", statement, re.M | re.IGNORECASE):
                 result = ReviewResult(
                     id=line,
                     errlevel=2,
@@ -273,11 +273,13 @@ def execute_check(self, db_name=None, sql=""):
                     sql=statement,
                 )
             # alter语句
-            elif re.match(r"^alter", statement.lower()):
+            elif re.match(r"^alter", statement, re.M | re.IGNORECASE):
                 # alter table语句
-                if re.match(r"^alter\s+table\s+(.+?)\s+", statement.lower()):
+                if re.match(
+                    r"^alter\s+table\s+(.+?)\s+", statement, re.M | re.IGNORECASE
+                ):
                     table_name = re.match(
-                        r"^alter\s+table\s+(.+?)\s+", statement.lower(), re.M
+                        r"^alter\s+table\s+(.+?)\s+", statement, re.M | re.IGNORECASE
                     ).group(1)
                     if "." not in table_name:
                         table_name = f"{db_name}.{table_name}"
@@ -298,7 +300,8 @@ def execute_check(self, db_name=None, sql=""):
                             # delete与update语句，实际是alter语句的变种
                             if re.match(
                                 r"^alter\s+table\s+(.+?)\s+(delete|update)\s+",
-                                statement.lower(),
+                                statement,
+                                re.M | re.IGNORECASE,
                             ):
                                 if not table_engine.endswith("MergeTree"):
                                     result = ReviewResult(
@@ -328,9 +331,11 @@ def execute_check(self, db_name=None, sql=""):
                 else:
                     result = self.explain_check(check_result, db_name, line, statement)
             # truncate语句
-            elif re.match(r"^truncate\s+table\s+(.+?)(\s|$)", statement.lower()):
+            elif re.match(
+                r"^truncate\s+table\s+(.+?)(\s|$)", statement, re.M | re.IGNORECASE
+            ):
                 table_name = re.match(
-                    r"^truncate\s+table\s+(.+?)(\s|$)", statement.lower(), re.M
+                    r"^truncate\s+table\s+(.+?)(\s|$)", statement, re.M | re.IGNORECASE
                 ).group(1)
                 if "." not in table_name:
                     table_name = f"{db_name}.{table_name}"
@@ -358,15 +363,16 @@ def execute_check(self, db_name=None, sql=""):
                         sql=statement,
                     )
             # insert语句，explain无法正确判断，暂时只做表存在性检查与简单关键字匹配
-            elif re.match(r"^insert", statement.lower()):
+            elif re.match(r"^insert", statement, re.M | re.IGNORECASE):
                 if re.match(
                     r"^insert\s+into\s+([a-zA-Z_][0-9a-zA-Z_.]+)([\w\W]*?)(values|format|select)(\s+|\()",
-                    statement.lower(),
+                    statement,
+                    re.M | re.IGNORECASE,
                 ):
                     table_name = re.match(
                         r"^insert\s+into\s+([a-zA-Z_][0-9a-zA-Z_.]+)([\w\W]*?)(values|format|select)(\s+|\()",
-                        statement.lower(),
-                        re.M,
+                        statement,
+                        re.M | re.IGNORECASE,
                     ).group(1)
                     if "." not in table_name:
                         table_name = f"{db_name}.{table_name}"