From 7168e5ce67119d0c171be7dac36bd7d21f762589 Mon Sep 17 00:00:00 2001
From: Andi Albrecht <albrecht.andi@gmail.com>
Date: Thu, 16 May 2019 07:56:57 +0200
Subject: Add support for time zone casts (fixes #489).

---
 sqlparse/engine/grouping.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

(limited to 'sqlparse/engine')

diff --git a/sqlparse/engine/grouping.py b/sqlparse/engine/grouping.py
index afc9123..ab97ec6 100644
--- a/sqlparse/engine/grouping.py
+++ b/sqlparse/engine/grouping.py
@@ -88,6 +88,19 @@ def group_typecasts(tlist):
     _group(tlist, sql.Identifier, match, valid_prev, valid_next, post)
 
 
+def group_tzcasts(tlist):
+    def match(token):
+        return token.ttype == T.Keyword.TZCast
+
+    def valid(token):
+        return token is not None
+
+    def post(tlist, pidx, tidx, nidx):
+        return pidx, nidx
+
+    _group(tlist, sql.Identifier, match, valid, valid, post)
+
+
 def group_period(tlist):
     def match(token):
         return token.match(T.Punctuation, '.')
@@ -358,6 +371,7 @@ def group(stmt):
         group_identifier,
         group_order,
         group_typecasts,
+        group_tzcasts,
         group_operator,
         group_comparison,
         group_as,
-- 
cgit v1.2.1