annotate sqlex/main.py @ 0:a05c70cc24be

initial commit
author Jeff Hammel <k0scist@gmail.com>
date Fri, 31 Mar 2017 18:21:15 -0700
parents
children 1cfdb859f9d2
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
1 #!/usr/bin/env python
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
2 # -*- coding: utf-8 -*-
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
3
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
4 """
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
5 sql(ite) explorer/exporter
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
6 """
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
7
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
8 # imports
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
9 import argparse
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
10 import os
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
11 import sys
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
12
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
13 def ensure_dir(directory):
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
14 """ensure a directory exists"""
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
15 if os.path.exists(directory):
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
16 if not os.path.isdir(directory):
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
17 raise OSError("Not a directory: '{}'".format(directory))
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
18 return directory
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
19 os.makedirs(directory)
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
20 return directory
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
21
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
22
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
23 class SQLExParser(argparse.ArgumentParser):
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
24 """CLI option parser"""
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
25
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
26 def __init__(self, **kwargs):
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
27 kwargs.setdefault('formatter_class', argparse.RawTextHelpFormatter)
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
28 kwargs.setdefault('description', __doc__)
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
29 argparse.ArgumentParser.__init__(self, **kwargs)
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
30 self.add_argument('db', help="sqlite `.db` file")
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
31 self.options = None
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
32
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
33 def parse_args(self, *args, **kw):
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
34 options = argparse.ArgumentParser.parse_args(self, *args, **kw)
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
35 self.validate(options)
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
36 self.options = options
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
37 return options
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
38
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
39 def validate(self, options):
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
40 """validate options"""
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
41
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
42 def main(args=sys.argv[1:]):
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
43 """CLI"""
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
44
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
45 # parse command line options
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
46 parser = SQLExParser()
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
47 options = parser.parse_args(args)
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
48
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
49 if __name__ == '__main__':
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
50 main()
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
51
a05c70cc24be initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
52