From e1b5482b1413591260f41ceb50e97091928f195a Mon Sep 17 00:00:00 2001
From: Andi Albrecht <albrecht.andi@gmail.com>
Date: Sun, 8 Feb 2015 16:52:15 +0100
Subject: (postgresql) Add support for square bracket array indexing, by
 darikg.

Pull request #170 with trivial conflicts resolved.
---
 sqlparse/sql.py | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'sqlparse/sql.py')

diff --git a/sqlparse/sql.py b/sqlparse/sql.py
index 6174db0..c974b35 100644
--- a/sqlparse/sql.py
+++ b/sqlparse/sql.py
@@ -502,6 +502,13 @@ class Identifier(TokenList):
             return None
         return ordering.value.upper()
 
+    def get_array_indices(self):
+        """Returns an iterator of index expressions as strings"""
+
+        # Use [1:-1] index to discard the square brackets
+        return (tok.value[1:-1] for tok in self.tokens
+                if tok.ttype in T.ArrayIndex)
+
 
 class IdentifierList(TokenList):
     """A list of :class:`~sqlparse.sql.Identifier`\'s."""
-- 
cgit v1.2.1