From e4bccf45a6e82d4738aab1b5eded3fa208a76875 Mon Sep 17 00:00:00 2001 From: Andi Albrecht Date: Thu, 19 Apr 2012 22:02:08 +0200 Subject: Pretty format comments in identifier lists (fixes issue59). --- sqlparse/sql.py | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'sqlparse/sql.py') 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 -- cgit v1.2.1