Exemplo n.º 1
0
def db(app, request):
    """Session-wide test database."""
    if os.path.exists(TESTDB):
        os.unlink(TESTDB)

    def teardown():
        _db.drop_all()
        os.unlink(TESTDB)

    _db.app = app
    _db.create_all()

    request.addfinalizer(teardown)
    return _db
Exemplo n.º 2
0
def db_create():
    with gendb.app_context():
        db.create_all()
    if not os.path.exists(SQLALCHEMY_MIGRATE_REPO):
        api.create(SQLALCHEMY_MIGRATE_REPO, 'database repository')
        api.version_control(
            SQLALCHEMY_DATABASE_URI,
            SQLALCHEMY_MIGRATE_REPO
        )
    else:
        api.version_control(
            SQLALCHEMY_DATABASE_URI,
            SQLALCHEMY_MIGRATE_REPO,
            api.version(SQLALCHEMY_MIGRATE_REPO)
        )