X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;f=scripts%2Fgroup-sql.py;h=850dcb30b2ad1fd72f79274515feff33517f4c7b;hb=96d6304e4c08ef94bb7dfac3e3069fcb8c982747;hp=f1f09886ea6be49b31632f767bf640ed25b8b141;hpb=031cc0d6a41717d4c5d7c4659290e05810202eb9;p=sqlconvert.git

diff --git a/scripts/group-sql.py b/scripts/group-sql.py
index f1f0988..850dcb3 100755
--- a/scripts/group-sql.py
+++ b/scripts/group-sql.py
@@ -2,7 +2,6 @@
 from __future__ import print_function
 
 import sys
-from sqlparse import parse
 from mysql2sql.print_tokens import print_tokens
 from mysql2sql.process_tokens import requote_names, find_error, \
     StatementGrouper
@@ -11,7 +10,7 @@ from mysql2sql.process_tokens import requote_names, find_error, \
 def main(*queries):
     grouper = StatementGrouper()
     for query in queries:
-        grouper.process(parse(query)[0])
+        grouper.process_line(query)
         if grouper.statements:
             for statement in grouper.get_statements():
                 print("----------")
@@ -22,7 +21,10 @@ def main(*queries):
                 print()
                 statement._pprint_tree()
             print("----------")
-    grouper.close()
+    tokens = grouper.close()
+    for token in tokens:
+        print_tokens(token)
+        print(repr(token))
 
 
 def test():