def main(): usage = """Data Model DDL and ERD Generator Usage: main.py (ddl | erd | start | test) [<args>...] """ # noqa from docopt import docopt # Ignore command name. argv = sys.argv[1:] args = docopt(usage, argv=argv, version=__version__, options_first=True) # Trim subcommand. sub_argv = argv[1:] if args['ddl']: ddl.main(sub_argv) elif args['erd']: erd.main(sub_argv) elif args['start']: service.main(sub_argv) elif args['test']: test.main(sub_argv)
def erd_route(model, version): ext = request.args.get('format') or 'png' filename = '%s_%s.%s' % (model, version, ext) filepath = '/'.join([app.instance_path, filename]) try: os.makedirs(app.instance_path) except OSError: pass erd.main([model, version, filepath]) return send_file(filepath)