Helper function to return a token at a specific offset.

1 files changed, 9 insertions, 0 deletions

diff --git a/sqlparse/sql.py b/sqlparse/sql.py
index 86deb4e..16f055b 100644
--- a/sqlparse/sql.py
+++ b/sqlparse/sql.py
@@ -196,6 +196,15 @@ class TokenList(Token):
             val = val[1:-1]
         return val
 
+    def get_token_at_offset(self, offset):
+        """Returns the token that is on position offset."""
+        idx = 0
+        for token in self.flatten():
+            end = idx + len(token.value)
+            if idx <= offset <= end:
+                return token
+            idx = end
+
     def flatten(self):
         """Generator yielding ungrouped tokens.