summaryrefslogtreecommitdiff
path: root/sqlparse/sql.py
diff options
context:
space:
mode:
Diffstat (limited to 'sqlparse/sql.py')
-rw-r--r--sqlparse/sql.py10
1 files changed, 9 insertions, 1 deletions
diff --git a/sqlparse/sql.py b/sqlparse/sql.py
index 72609e2..9c7aeee 100644
--- a/sqlparse/sql.py
+++ b/sqlparse/sql.py
@@ -137,7 +137,7 @@ class TokenList(Token):
if tokens is None:
tokens = []
self.tokens = tokens
- Token.__init__(self, None, None)
+ Token.__init__(self, None, unicode(self))
def __unicode__(self):
return ''.join(unicode(x) for x in self.flatten())
@@ -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