Pretty format comments in identifier lists (fixes issue59).

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