def unregister_models(base=model_base.SavannaBase): """Unregister Models, useful clearing out data before testing.""" try: engine = session.get_engine(sqlite_fk=True) base.metadata.drop_all(engine) except Exception: LOG.exception("Database exception")
def drop_db(): try: engine = db_session.get_engine(sqlite_fk=True) m.Cluster.metadata.drop_all(engine) except Exception as e: LOG.exception("Database shutdown exception: %s", e) return False return True
def setup_db(): try: engine = db_session.get_engine(sqlite_fk=True) m.Cluster.metadata.create_all(engine) except sa.exc.OperationalError as e: LOG.exception("Database registration exception: %s", e) return False return True
def drop_db(): try: engine = db_session.get_engine(sqlite_fk=True) m.Cluster.metadata.drop_all(engine) except Exception as e: LOG.error("Database shutdown exception: %s", e) return False return True
def setup_db(): try: engine = db_session.get_engine(sqlite_fk=True) m.Cluster.metadata.create_all(engine) except sa.exc.OperationalError as e: LOG.error("Database registration exception: %s", e) return False return True
def register_models(base=model_base.SavannaBase): """Register Models and create properties.""" try: engine = session.get_engine(sqlite_fk=True) base.metadata.create_all(engine) except sql.exc.OperationalError as e: LOG.info("Database registration exception: %s", e) return False return True
def configure_db(): """Configure database. Establish the database, create an engine if needed, and register the models. """ global _DB_ENGINE if not _DB_ENGINE: session.set_defaults('sqlite:////tmp/savanna-server.db', 'savanna-server.db') _DB_ENGINE = session.get_engine(sqlite_fk=True) register_models()