From 32ab737bc4029ea0602c71fd37a0f46a3889a154 Mon Sep 17 00:00:00 2001 From: Andi Albrecht Date: Fri, 30 Aug 2013 12:31:57 +0200 Subject: Add skip_ws flag to Token.insert_after (fixes #108). --- sqlparse/sql.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'sqlparse') diff --git a/sqlparse/sql.py b/sqlparse/sql.py index f8f4cc9..3f1e98b 100644 --- a/sqlparse/sql.py +++ b/sqlparse/sql.py @@ -376,9 +376,9 @@ class TokenList(Token): """Inserts *token* before *where*.""" self.tokens.insert(self.token_index(where), token) - def insert_after(self, where, token): + def insert_after(self, where, token, skip_ws=True): """Inserts *token* after *where*.""" - next_token = self.token_next(where) + next_token = self.token_next(where, skip_ws=skip_ws) if next_token is None: self.tokens.append(token) else: -- cgit v1.2.1