Ejemplo n.º 1
0
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)
Ejemplo n.º 2
0
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)