X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;f=scripts%2Fprint_subtree.py;h=100fa649c313a2713cda1e9b753e1bc565831bef;hb=db60d46e72f1cbec5a45010abadd0b66434ea8a7;hp=d6c3f743810c601a9b7f796f544fc434f402fb54;hpb=ef0d69a5069eef56d4601deb0848059b7271b907;p=sqlconvert.git

diff --git a/scripts/print_subtree.py b/scripts/print_subtree.py
index d6c3f74..100fa64 100755
--- a/scripts/print_subtree.py
+++ b/scripts/print_subtree.py
@@ -4,7 +4,7 @@ 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():
@@ -14,25 +14,31 @@ def test():
         "/*! directive*/ INSERT INTO `MyTable` (`Id`, `Name`) "
         "VALUES (1, 'one')"
     ):
-        print("----------")
         for parsed in parse(query):
+            print("----------")
             requote_names(parsed)
             print_tokens(parsed)
-            print_subtree(parsed)
+            print()
+            parsed._pprint_tree()
     print("----------")
 
 
-def main(query):
-    parsed = parse(query)[0]
-    requote_names(parsed)
-    print_tokens(parsed)
-    print_subtree(parsed)
+def main(*queries):
+    for query in queries:
+        for parsed in parse(query):
+            print("----------")
+            requote_names(parsed)
+            print_tokens(parsed)
+            print()
+            parsed._pprint_tree()
+        print("----------")
 
 if __name__ == '__main__':
-    if len(sys.argv) != 2:
-        sys.exit("Usage: %s [-t | sql_query_string]" % sys.argv[0])
+    if len(sys.argv) <= 1:
+        sys.exit("Usage: %s [-t | sql_query_string [; sql_query_string ...]]" %
+                 sys.argv[0])
     if sys.argv[1] == '-t':
         test()
     else:
-        query = sys.argv[1]
-        main(query)
+        queries = sys.argv[1:]
+        main(*queries)