0
|
1 #!/usr/bin/env python
|
|
2 # -*- coding: utf-8 -*-
|
|
3
|
|
4 """
|
|
5 sql(ite) explorer/exporter
|
|
6 """
|
|
7
|
|
8 # imports
|
|
9 import argparse
|
|
10 import os
|
|
11 import sys
|
3
|
12 from .model import SQLEx
|
|
13
|
0
|
14
|
|
15 def ensure_dir(directory):
|
|
16 """ensure a directory exists"""
|
|
17 if os.path.exists(directory):
|
|
18 if not os.path.isdir(directory):
|
|
19 raise OSError("Not a directory: '{}'".format(directory))
|
|
20 return directory
|
|
21 os.makedirs(directory)
|
|
22 return directory
|
|
23
|
|
24
|
|
25 class SQLExParser(argparse.ArgumentParser):
|
|
26 """CLI option parser"""
|
|
27
|
|
28 def __init__(self, **kwargs):
|
|
29 kwargs.setdefault('formatter_class', argparse.RawTextHelpFormatter)
|
|
30 kwargs.setdefault('description', __doc__)
|
|
31 argparse.ArgumentParser.__init__(self, **kwargs)
|
1
|
32 self.add_argument('db',
|
|
33 help="sqlite `.db` file")
|
3
|
34 self.add_argument('--tables', '--list-tables', dest='list_tables',
|
|
35 action='store_true', default=False,
|
|
36 help="list tables and exit")
|
0
|
37 self.options = None
|
|
38
|
|
39 def parse_args(self, *args, **kw):
|
|
40 options = argparse.ArgumentParser.parse_args(self, *args, **kw)
|
|
41 self.validate(options)
|
|
42 self.options = options
|
|
43 return options
|
|
44
|
|
45 def validate(self, options):
|
|
46 """validate options"""
|
|
47
|
1
|
48 try:
|
3
|
49 open(options.db).close()
|
1
|
50 except Exception as e:
|
|
51 self.error("Could not open '{}': {}".format(options.db, e))
|
|
52
|
2
|
53
|
0
|
54 def main(args=sys.argv[1:]):
|
|
55 """CLI"""
|
|
56
|
|
57 # parse command line options
|
|
58 parser = SQLExParser()
|
|
59 options = parser.parse_args(args)
|
|
60
|
3
|
61 # connect to database
|
|
62 db = SQLEx(options.db)
|
|
63
|
|
64 if options.list_tables:
|
|
65 tables = db.tables()
|
|
66 print ('\n'.join(tables))
|
|
67 return
|
|
68
|
0
|
69 if __name__ == '__main__':
|
|
70 main()
|