Fix Python3 compatibility

[sqlconvert.git] / scripts / group-file.py

diff --git a/scripts/group-file.py b/scripts/group-file.py
index 53a11f10fb4362b5df18da0a9174a3b5f613491f..5ab1f447ea2f48bf61c5dae14db4ccc8a29f4675 100755 (executable)
--- 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("----------")
@@ -24,9 +23,10 @@ def main(filename):
                     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))
 
 
 if __name__ == '__main__':