demo/group-sql.py

   1 #! /usr/bin/env python
   2 from __future__ import print_function
   3
   4 import sys
   5 from mysql2sql.print_tokens import print_tokens
   6 from mysql2sql.process_tokens import requote_names, find_error, \
   7     StatementGrouper
   8
   9
  10 def main(*queries):
  11     grouper = StatementGrouper()
  12     for query in queries:
  13         grouper.process_line(query)
  14         if grouper.statements:
  15             for statement in grouper.get_statements():
  16                 print("----------")
  17                 if find_error(statement):
  18                     print("ERRORS IN QUERY")
  19                 requote_names(statement)
  20                 print_tokens(statement)
  21                 print()
  22                 statement._pprint_tree()
  23             print("----------")
  24     tokens = grouper.close()
  25     if tokens:
  26         for token in tokens:
  27             print_tokens(token)
  28             print(repr(token))
  29
  30
  31 def test():
  32     main(
  33         "SELECT * FROM `mytable`; -- line-comment",
  34         "INSERT into /* inline comment */ mytable VALUES (1, 'one');",
  35         "/*! directive*/ INSERT INTO `MyTable` (`Id`, `Name`) "
  36         "VALUES (1, 'one');"
  37     )
  38
  39
  40 if __name__ == '__main__':
  41     if len(sys.argv) <= 1:
  42         sys.exit("Usage: %s [-t | sql_query_string [; sql_query_string ...]]" %
  43                  sys.argv[0])
  44     if sys.argv[1] == '-t':
  45         test()
  46     else:
  47         queries = sys.argv[1:]
  48         main(*queries)