summaryrefslogtreecommitdiff
path: root/sqlparse/sql.py
diff options
context:
space:
mode:
Diffstat (limited to 'sqlparse/sql.py')
-rw-r--r--sqlparse/sql.py9
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.