summaryrefslogtreecommitdiff
path: root/sqlparse/sql.py
diff options
context:
space:
mode:
authorAndi Albrecht <albrecht.andi@gmail.com>2013-08-30 12:31:57 +0200
committerAndi Albrecht <albrecht.andi@gmail.com>2013-08-30 12:31:57 +0200
commit32ab737bc4029ea0602c71fd37a0f46a3889a154 (patch)
tree8bcb65a732a6a461d2e6f6c8b1d09d4e016a0ee2 /sqlparse/sql.py
parentd4b8e6e03e6c8d610da458593166be9a0d5d014f (diff)
downloadsqlparse-32ab737bc4029ea0602c71fd37a0f46a3889a154.tar.gz
Add skip_ws flag to Token.insert_after (fixes #108).
Diffstat (limited to 'sqlparse/sql.py')
-rw-r--r--sqlparse/sql.py4
1 files changed, 2 insertions, 2 deletions
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: