Mercurial > hg > sqlex
view 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 |
line wrap: on
line source
#!/usr/bin/env python # -*- coding: utf-8 -*- """ sql(ite) explorer/exporter """ # imports import argparse import os import sys def ensure_dir(directory): """ensure a directory exists""" if os.path.exists(directory): if not os.path.isdir(directory): raise OSError("Not a directory: '{}'".format(directory)) return directory os.makedirs(directory) return directory class SQLExParser(argparse.ArgumentParser): """CLI option parser""" def __init__(self, **kwargs): kwargs.setdefault('formatter_class', argparse.RawTextHelpFormatter) kwargs.setdefault('description', __doc__) argparse.ArgumentParser.__init__(self, **kwargs) self.add_argument('db', help="sqlite `.db` file") self.options = None def parse_args(self, *args, **kw): options = argparse.ArgumentParser.parse_args(self, *args, **kw) self.validate(options) self.options = options return options def validate(self, options): """validate options""" def main(args=sys.argv[1:]): """CLI""" # parse command line options parser = SQLExParser() options = parser.parse_args(args) if __name__ == '__main__': main()