add PL/pgSQL syntax

Demetrio92 · web-flow · commit c48a221e72df · 2017-08-07T18:02:51.000+02:00
tested with mxtr/SQLTools
diff --git a/sqlparse/keywords.py b/sqlparse/keywords.py
@@ -14,6 +14,7 @@ def is_keyword(value):
     val = value.upper()
     return (KEYWORDS_COMMON.get(val) or
             KEYWORDS_ORACLE.get(val) or
+            KEYWORDS_PLPGSQL.get(val) or
             KEYWORDS.get(val, tokens.Name)), value
 
 
@@ -800,3 +801,18 @@ def is_keyword(value):
     'UNLIMITED': tokens.Keyword,
     'UNLOCK': tokens.Keyword,
 }
+
+# PostgreSQL Syntax
+KEYWORDS_PLPGSQL = {
+    'PARTITION':    tokens.Keyword,
+    'OVER':         tokens.Keyword,
+    'PERFORM':      tokens.Keyword,
+    'NOTICE':       tokens.Keyword,
+    'PLPGSQL':      tokens.Keyword,
+    'INHERIT':      tokens.Keyword,
+    'INDEXES':      tokens.Keyword,
+
+    'FOR':          tokens.Keyword,
+    'IN':           tokens.Keyword,
+    'LOOP':         tokens.Keyword,
+}
