X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;f=scripts%2Fgroup-file.py;h=5ab1f447ea2f48bf61c5dae14db4ccc8a29f4675;hb=0f209248832190291a510febd790e71a815e496a;hp=41f9b3318b47248100e892db2d4362201961594f;hpb=031cc0d6a41717d4c5d7c4659290e05810202eb9;p=sqlconvert.git

diff --git a/scripts/group-file.py b/scripts/group-file.py
index 41f9b33..5ab1f44 100755
--- a/scripts/group-file.py
+++ b/scripts/group-file.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
@@ -12,7 +11,7 @@ def main(filename):
     grouper = StatementGrouper()
     with open(filename) as infile:
         for line in infile:
-            grouper.process(parse(line)[0])
+            grouper.process_line(line)
             if grouper.statements:
                 for statement in grouper.get_statements():
                     print("----------")
@@ -23,7 +22,11 @@ def main(filename):
                     print()
                     statement._pprint_tree()
                 print("----------")
-    grouper.close()
+    tokens = grouper.close()
+    if tokens:
+        for token in tokens:
+            print_tokens(token)
+            print(repr(token))
 
 
 if __name__ == '__main__':