summaryrefslogtreecommitdiff
path: root/sqlparse/sql.py
diff options
context:
space:
mode:
Diffstat (limited to 'sqlparse/sql.py')
-rw-r--r--sqlparse/sql.py8
1 files changed, 8 insertions, 0 deletions
diff --git a/sqlparse/sql.py b/sqlparse/sql.py
index 72609e2..3ea1c6f 100644
--- a/sqlparse/sql.py
+++ b/sqlparse/sql.py
@@ -322,6 +322,14 @@ class TokenList(Token):
"""Inserts *token* before *where*."""
self.tokens.insert(self.token_index(where), token)
+ def insert_after(self, where, token):
+ """Inserts *token* after *where*."""
+ next_token = self.token_next(where)
+ if next_token is None:
+ self.tokens.append(token)
+ else:
+ self.tokens.insert(self.token_index(next_token), token)
+
def has_alias(self):
"""Returns ``True`` if an alias is present."""
return self.get_alias() is not None