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()
