diff options
Diffstat (limited to 'sqlparse/sql.py')
| -rw-r--r-- | sqlparse/sql.py | 9 |
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. |
