def db_reset(
load: str = None, dir: str = BACKUP_PATH, structure: bool = True, data: bool = False
):
"""Reset database, with options to load fresh data."""
if not app.debug:
prompt = (
"This deletes all data and resets the structure on %s.\nDo you want to continue?"
% app.db.engine
)
if not click.confirm(prompt):
click.echo("I did nothing.")
return
from flexmeasures.data.scripts.data_gen import reset_db
current_version = migrate.current()
reset_db(app.db)
migrate.stamp(current_version)
if load:
if not data and not structure:
click.echo("Neither --data nor --structure given ... loading nothing.")
return
from flexmeasures.data.scripts.data_gen import load_tables
load_tables(app.db, load, structure, data, dir)
def free_the_zoo(zoo_url):
confirm = str(input('Are you sure you want to clear and re-migrate the database? (y/N): ')).strip()
if confirm == 'y':
init_zoo_db(zoo_url)
if flask_migrate.current() is not None:
flask_migrate.downgrade(tag='base')
flask_migrate.upgrade()
migrate_models()
def reset():
"""Reset database data and re-create tables from data model."""
if not app.debug:
prompt = (
"This deletes all data and re-creates the tables on %s.\nDo you want to continue?"
% app.db.engine
)
if not click.confirm(prompt):
click.echo("I did nothing.")
return
from flexmeasures.data.scripts.data_gen import reset_db
current_version = migrate.current()
reset_db(app.db)
migrate.stamp(current_version)
def upgrade_db(first_attempt=True):
try:
upgrade()
except Exception:
print('Error: Could not run migrations')
if first_attempt and current() == None:
# The migrations failed and the database has no record of past migrations.
#
# This is likely due to the developer having a version of the database from before Flask-Migrate
# was introduced to the project.
#
# We try to automatically resolve this by telling alembic (the migration tool Flask-Migrate is based on)
# that this database equates to the first revision, and then try the migration again.
print(
'Retrying migration after stamping database with baseline revision'
)
stamp(revision='baseline')
upgrade_db(False)
else:
logging.error(traceback.format_exc())
raise
def database(action, revision, message):
with webui.app.app_context():
# creates folder /migrations with necessary files for migrations
if action == "init":
init_migration()
# creates migration with current state of models
elif action == "migrate":
comment = str(message) if message else ""
migrate(message=comment)
# upgrades current state to latest version of database or to defined revision
elif action == "upgrade":
if revision:
upgrade(revision=revision)
else:
upgrade()
# downgrades current state to previous version of database or to defined revision
elif action == "downgrade":
if revision:
downgrade(revision=revision)
else:
downgrade()
# shows current migration
elif action == "current":
return current()
# shows history of migrations until current migration
elif action == "history":
return history()
# creates migration manually with comment = message
elif action == "create-migration":
comment = str(message) if message else ""
return manual_migrate(message=comment)
def db_current(c, dir=DEFAULT_MIGRATION_DIR):
"""
Show current migration revision.
"""
with app.app_context():
migrate.current(directory=dir)
def current(args):
"""Get current database schema revision."""
flask_migrate.current(DIRECTORY)
def db_current(c, dir=DEFAULT_MIGRATION_DIR):
"""
Show current migration revision.
"""
with app.app_context():
migrate.current(directory=dir)
def mcurrent(directory=None, verbose=False, head_only=False):
current(directory=None, verbose=False, head_only=False)
from app import app, db
from flask_migrate import stamp, current
with app.app_context():
db.create_all()
print('Database tables:')
for table in db.metadata.sorted_tables:
print('- {}'.format(table))
stamp()
print('Alembic current revision set to:')
current()
def current(args):
"""Get current database schema revision."""
flask_migrate.current(DIRECTORY)