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 find_error, StatementGrouper
   7
   8
   9 def main(filename):
  10     grouper = StatementGrouper(encoding='utf-8')
  11     with open(filename) as infile:
  12         for line in infile:
  13             grouper.process_line(line)
  14             if grouper.statements:
  15                 for statement in grouper.get_statements():
  16                     print("----------")
  17                     if find_error(statement):
  18                         print("ERRORS IN QUERY")
  19                     print_tokens(statement, encoding='utf-8')
  20                     print()
  21                     statement._pprint_tree()
  22                 print("----------")
  23     tokens = grouper.close()
  24     if tokens:
  25         for token in tokens:
  26             print_tokens(token, encoding='utf-8')
  27             print(repr(token))
  28
  29
  30 if __name__ == '__main__':
  31     if len(sys.argv) <= 1:
  32         sys.exit("Usage: %s file" % sys.argv[0])
  33     main(sys.argv[1])