diff options
Diffstat (limited to 'sqlparse/sql.py')
| -rw-r--r-- | sqlparse/sql.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/sqlparse/sql.py b/sqlparse/sql.py index cee6af5..52b3bf1 100644 --- a/sqlparse/sql.py +++ b/sqlparse/sql.py @@ -140,6 +140,7 @@ class TokenList(Token): def __init__(self, tokens=None): self.tokens = tokens or [] + [setattr(token, 'parent', self) for token in tokens] super(TokenList, self).__init__(None, text_type(self)) def __str__(self): @@ -300,11 +301,13 @@ class TokenList(Token): def insert_before(self, where, token): """Inserts *token* before *where*.""" + token.parent = self self.tokens.insert(self.token_index(where), token) def insert_after(self, where, token, skip_ws=True): """Inserts *token* after *where*.""" next_token = self.token_next(where, skip_ws=skip_ws) + token.parent = self if next_token is None: self.tokens.append(token) else: |