2 from __future__ import print_function
6 from sqlparse import parse
7 from sqlconvert.print_tokens import print_tokens
8 from sqlconvert.process_tokens import find_error
11 def parse_queries(*queries):
13 for parsed in parse(query, encoding='utf-8'):
15 if find_error(parsed):
16 print("ERRORS IN QUERY")
17 print_tokens(parsed, encoding='utf-8')
23 def parse_test(_args):
25 "SELECT * FROM `mytable`; -- line-comment",
26 "INSERT into /* inline comment */ mytable VALUES (1, 'one')",
27 "/*! directive*/ INSERT INTO `MyTable` (`Id`, `Name`) "
33 parse_queries(*args.lines)
37 infile = open(args.filename, 'rt')
38 lines = infile.readlines()
43 if __name__ == '__main__':
44 main_parser = argparse.ArgumentParser(description='Parse')
45 subparsers = main_parser.add_subparsers(help='Commands')
47 parser = subparsers.add_parser('sql', help='SQL from command line')
48 parser.add_argument('lines', nargs='+', help='SQL lines')
49 parser.set_defaults(func=parse_sql)
51 parser = subparsers.add_parser('test', help='SQL from test data')
52 parser.set_defaults(func=parse_test)
54 parser = subparsers.add_parser('file', help='SQL from a file')
55 parser.add_argument('filename', help='SQL file')
56 parser.set_defaults(func=parse_file)
58 args = main_parser.parse_args()