From f6c3a0f124becdaeacfb755d7ba61ffc7c032337 Mon Sep 17 00:00:00 2001
From: Andi Albrecht <albrecht.andi@gmail.com>
Date: Sun, 27 Jan 2013 10:21:37 +0100
Subject: Helper function to return a token at a specific offset.

---
 sqlparse/sql.py | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'sqlparse/sql.py')

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.
 
-- 
cgit v1.2.1