Mercurial > repos > jjohnson > query_tabular

comparison sqlite_to_tabular.py @ 20:ab27c4bd14b9 draft

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Uploaded
author jjohnson
date Fri, 14 Jul 2017 11:39:27 -0400
parents
children bed5018e7ae3
comparison
equal deleted inserted replaced
19:9d9ab2c69014 20:ab27c4bd14b9
1 #!/usr/bin/env python
2
3 from __future__ import print_function
4
5 import optparse
6 import os.path
7 import sys
8
9 from query_db import describe_tables, get_connection, run_query
10
11
12 def __main__():
13 # Parse Command Line
14 parser = optparse.OptionParser()
15 parser.add_option('-s', '--sqlitedb', dest='sqlitedb', default=None,
16 help='The SQLite Database')
17 parser.add_option('-q', '--query', dest='query', default=None,
18 help='SQL query')
19 parser.add_option('-Q', '--query_file', dest='query_file', default=None,
20 help='SQL query file')
21 parser.add_option('-n', '--no_header', dest='no_header', default=False,
22 action='store_true',
23 help='Include a column headers line')
24 parser.add_option('-o', '--output', dest='output', default=None,
25 help='Output file for query results')
26 (options, args) = parser.parse_args()
27
28 # determine output destination
29 if options.output is not None:
30 try:
31 outputPath = os.path.abspath(options.output)
32 outputFile = open(outputPath, 'w')
33 except Exception as e:
34 print("failed: %s" % e, file=sys.stderr)
35 exit(3)
36 else:
37 outputFile = sys.stdout
38
39 query = None
40 if (options.query_file is not None):
41 with open(options.query_file, 'r') as fh:
42 query = ''
43 for line in fh:
44 query += line
45 elif (options.query is not None):
46 query = options.query
47
48 if (query is None):
49 try:
50 describe_tables(get_connection(options.sqlitedb), outputFile)
51 except Exception as exc:
52 print("Error: %s" % exc, file=sys.stderr)
53 exit(0)
54 else:
55 try:
56 run_query(get_connection(options.sqlitedb), query, outputFile,
57 no_header=options.no_header)
58 except Exception as exc:
59 print("Error: %s" % exc, file=sys.stderr)
60 exit(1)
61
62
63 if __name__ == "__main__":
64 __main__()