From d8a0300bab3a77fba7fa6014ab7fa7c5ae388aaa Mon Sep 17 00:00:00 2001 From: Andi Albrecht Date: Sun, 12 Jul 2009 10:15:19 +0200 Subject: Add parent relationships. --- sqlparse/sql.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) (limited to 'sqlparse') diff --git a/sqlparse/sql.py b/sqlparse/sql.py index 1c99a53..b5ba902 100644 --- a/sqlparse/sql.py +++ b/sqlparse/sql.py @@ -15,11 +15,12 @@ class Token(object): the type of the token. """ - __slots__ = ('value', 'ttype',) + __slots__ = ('value', 'ttype', 'parent') def __init__(self, ttype, value): self.value = value self.ttype = ttype + self.parent = None def __str__(self): return unicode(self).encode('latin-1') @@ -268,6 +269,9 @@ class TokenList(Token): for t in tokens: self.tokens.remove(t) grp = grp_cls(tokens) + for token in tokens: + token.parent = grp + grp.parent = self self.tokens.insert(idx, grp) return grp -- cgit v1.2.1