Update keywords_lists.py

Added support of 'MERGE' command.
macbre · Sep 5, 2023 · 2ca440b · 2ca440b
1 parent cf09e64
commit 2ca440b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/sql_metadata/keywords_lists.py b/sql_metadata/keywords_lists.py
@@ -34,7 +34,7 @@
 }
 
 # next statement beginning after with statement
-WITH_ENDING_KEYWORDS = {"UPDATE", "SELECT", "DELETE", "REPLACE", "INSERT"}
+WITH_ENDING_KEYWORDS = {"UPDATE", "SELECT", "DELETE", "REPLACE", "INSERT", "MERGE"}
 
 # subquery preceding keywords
 SUBQUERY_PRECEDING_KEYWORDS = {
@@ -61,6 +61,7 @@
     "INTO": "insert",
     "SET": "update",
     "GROUPBY": "group_by",
+    "MERGE": "merge",
 }
 
 
@@ -77,6 +78,7 @@ class QueryType(str, Enum):
     CREATE = "CREATE TABLE"
     ALTER = "ALTER TABLE"
     DROP = "DROP TABLE"
+    MERGE = "MERGE"
 
 
 class TokenType(str, Enum):
@@ -104,6 +106,7 @@ class TokenType(str, Enum):
     "CREATETABLE": QueryType.CREATE,
     "ALTERTABLE": QueryType.ALTER,
     "DROPTABLE": QueryType.DROP,
+    "MERGE": QueryType.MERGE,
 }
 
 # all the keywords we care for - rest is ignored in assigning