scripts/print_subtree.py

   1 #! /usr/bin/env python
   2 from __future__ import print_function
   3
   4 import sys
   5 from sqlparse import parse
   6 from mysql2sql.process_tokens import requote_names
   7 from mysql2sql.print_tokens import print_tokens
   8
   9
  10 def test():
  11     for query in (
  12         "SELECT * FROM `mytable`; -- line-comment",
  13         "INSERT into /* inline comment */ mytable VALUES (1, 'one')",
  14         "/*! directive*/ INSERT INTO `MyTable` (`Id`, `Name`) "
  15         "VALUES (1, 'one')"
  16     ):
  17         print("----------")
  18         for parsed in parse(query):
  19             requote_names(parsed)
  20             print_tokens(parsed)
  21             print()
  22             parsed._pprint_tree()
  23     print("----------")
  24
  25
  26 def main(query):
  27     parsed = parse(query)[0]
  28     requote_names(parsed)
  29     print_tokens(parsed)
  30     print()
  31     parsed._pprint_tree()
  32
  33 if __name__ == '__main__':
  34     if len(sys.argv) != 2:
  35         sys.exit("Usage: %s [-t | sql_query_string]" % sys.argv[0])
  36     if sys.argv[1] == '-t':
  37         test()
  38     else:
  39         query = sys.argv[1]
  40         main(query)