Add new group for MySQL specific keywords.

andialbrecht · Mar 16, 2024 · 6b10952 · 6b10952
1 parent 8c24779
commit 6b10952
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/sqlparse/keywords.py b/sqlparse/keywords.py
@@ -486,7 +486,6 @@
     'ROUTINE_CATALOG': tokens.Keyword,
     'ROUTINE_NAME': tokens.Keyword,
     'ROUTINE_SCHEMA': tokens.Keyword,
-    'ROW': tokens.Keyword,
     'ROWS': tokens.Keyword,
     'ROW_COUNT': tokens.Keyword,
     'RULE': tokens.Keyword,
@@ -829,6 +828,11 @@
     'UNLOCK': tokens.Keyword,
 }
 
+# MySQL
+KEYWORDS_MYSQL = {
+    'ROW': tokens.Keyword,
+}
+
 # PostgreSQL Syntax
 KEYWORDS_PLPGSQL = {
     'CONFLICT': tokens.Keyword,

diff --git a/sqlparse/lexer.py b/sqlparse/lexer.py
@@ -62,6 +62,7 @@ def default_initialization(self):
         self.set_SQL_REGEX(keywords.SQL_REGEX)
         self.add_keywords(keywords.KEYWORDS_COMMON)
         self.add_keywords(keywords.KEYWORDS_ORACLE)
+        self.add_keywords(keywords.KEYWORDS_MYSQL)
         self.add_keywords(keywords.KEYWORDS_PLPGSQL)
         self.add_keywords(keywords.KEYWORDS_HQL)
         self.add_keywords(keywords.KEYWORDS_MSACCESS)