X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;f=scripts%2Fprint_subtree.py;h=f1da6de13caf4f515be588764e96e715124479aa;hb=51018ab80b945523dc678b80d4dd28191b3fd50c;hp=da5eae8c5fbdac1dc48db6a26cc20f5ce686a627;hpb=3289238688d9c3dfefccf143d21f6c406faad9e4;p=sqlconvert.git

diff --git a/scripts/print_subtree.py b/scripts/print_subtree.py
index da5eae8..f1da6de 100755
--- a/scripts/print_subtree.py
+++ b/scripts/print_subtree.py
@@ -1,9 +1,10 @@
 #! /usr/bin/env python
+from __future__ import print_function
 
 import sys
 from sqlparse import parse
 from mysql2sql.process_tokens import requote_names
-from mysql2sql.print_tokens import print_tokens, print_subtree
+from mysql2sql.print_tokens import print_tokens
 
 
 def test():
@@ -13,19 +14,21 @@ def test():
         "/*! directive*/ INSERT INTO `MyTable` (`Id`, `Name`) "
         "VALUES (1, 'one')"
     ):
-        print "----------"
+        print("----------")
         for parsed in parse(query):
             requote_names(parsed)
             print_tokens(parsed)
-            print_subtree(parsed)
-    print "----------"
+            print()
+            parsed._pprint_tree()
+    print("----------")
 
 
 def main(query):
     parsed = parse(query)[0]
     requote_names(parsed)
     print_tokens(parsed)
-    print_subtree(parsed)
+    print()
+    parsed._pprint_tree()
 
 if __name__ == '__main__':
     if len(sys.argv) != 2: