Use parent's type instead of generic Token in formatter.

Trundle · Trundle · commit 7c24edfd15c5 · 2009-08-23T14:17:19.000+02:00
diff --git a/bpython/formatter.py b/bpython/formatter.py
@@ -105,10 +105,9 @@ def format(self, tokensource, outfile):
             if text == '\n':
                 continue
 
-            if token in self.f_strings:
-                o +=  "%s\x03%s\x04" % (self.f_strings[token], text )
-            else:
-                o += "%s\x03%s\x04" % (self.f_strings[Token], text )
+            while token not in self.f_strings:
+                token = token.parent
+            o +=  "%s\x03%s\x04" % (self.f_strings[token], text)
         outfile.write(o.rstrip())
 
 # vim: sw=4 ts=4 sts=4 ai et
