Improve formatting of statements with JSON operators (fixes andialbrecht#542).

andialbrecht · andialbrecht · commit 8c24779e027e · 2024-03-16T10:10:19.000+01:00
diff --git a/CHANGELOG b/CHANGELOG
@@ -13,6 +13,7 @@ Enhancements:
 * Support TypedLiterals in get_parameters (pr649, by Khrol).
 * Improve splitting of Transact SQL when using GO keyword (issue762).
 * Support for some JSON operators (issue682).
+* Improve formatting of statements containing JSON operators (issue542).
 
 Bug Fixes
 
diff --git a/sqlparse/engine/grouping.py b/sqlparse/engine/grouping.py
@@ -139,7 +139,12 @@ def post(tlist, pidx, tidx, nidx):
 
 def group_period(tlist):
     def match(token):
-        return token.match(T.Punctuation, '.')
+        for ttype, value in ((T.Punctuation, '.'),
+                             (T.Operator, '->'),
+                             (T.Operator, '->>')):
+            if token.match(ttype, value):
+                return True
+        return False
 
     def valid_prev(token):
         sqlcls = sql.SquareBrackets, sql.Identifier
@@ -153,7 +158,7 @@ def valid_next(token):
     def post(tlist, pidx, tidx, nidx):
         # next_ validation is being performed here. issue261
         sqlcls = sql.SquareBrackets, sql.Function
-        ttypes = T.Name, T.String.Symbol, T.Wildcard
+        ttypes = T.Name, T.String.Symbol, T.Wildcard, T.String.Single
         next_ = tlist[nidx] if nidx is not None else None
         valid_next = imt(next_, i=sqlcls, t=ttypes)
 
