Parse square brackets as a group just like parens

- add class sql.SquareBrackets
- replace group_parenthesis() with more generic group_brackets(), which groups square and round brackets, so each can contain groups of the other

1 files changed, 9 insertions, 0 deletions

diff --git a/sqlparse/sql.py b/sqlparse/sql.py
index 8492c5e..25d5243 100644
--- a/sqlparse/sql.py
+++ b/sqlparse/sql.py
@@ -542,6 +542,15 @@ class Parenthesis(TokenList):
         return self.tokens[1:-1]
 
 
+class SquareBrackets(TokenList):
+    """Tokens between square brackets"""
+
+    __slots__ = ('value', 'ttype', 'tokens')
+
+    @property
+    def _groupable_tokens(self):
+        return self.tokens[1:-1]
+
 class Assignment(TokenList):
     """An assignment like 'var := val;'"""
     __slots__ = ('value', 'ttype', 'tokens')