scripts/group-file.py

   1 #! /usr/bin/env python
   2 from __future__ import print_function
   3
   4 import sys
   5 from sqlparse import parse
   6 from mysql2sql.print_tokens import print_tokens
   7 from mysql2sql.process_tokens import requote_names, find_error, \
   8     StatementGrouper
   9
  10
  11 def main(filename):
  12     grouper = StatementGrouper()
  13     with open(filename) as infile:
  14         for line in infile:
  15             grouper.process(parse(line)[0])
  16             if grouper.statements:
  17                 for statement in grouper.get_statements():
  18                     print("----------")
  19                     if find_error(statement):
  20                         print("ERRORS IN QUERY")
  21                     requote_names(statement)
  22                     print_tokens(statement)
  23                     print()
  24                     statement._pprint_tree()
  25                 print("----------")
  26     grouper.close()
  27
  28
  29 if __name__ == '__main__':
  30     if len(sys.argv) <= 1:
  31         sys.exit("Usage: %s file" % sys.argv[0])
  32     main(sys.argv[1])