X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;f=scripts%2Fgroup-sql.py;h=953e9e7cfc66a03304af31cdc7473ec584fd1a7f;hb=0f209248832190291a510febd790e71a815e496a;hp=0eab6b4042253cba3a629df455841924e2935c22;hpb=3151032d036f9a66bad633dc1018395a14f46bac;p=sqlconvert.git

diff --git a/scripts/group-sql.py b/scripts/group-sql.py
index 0eab6b4..953e9e7 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("----------")
@@ -23,9 +22,10 @@ def main(*queries):
                 statement._pprint_tree()
             print("----------")
     tokens = grouper.close()
-    for token in tokens:
-        print_tokens(token)
-        print(repr(token))
+    if tokens:
+        for token in tokens:
+            print_tokens(token)
+            print(repr(token))
 
 
 def test():