demo/group-file.py

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