diff --git a/budget/run.py b/budget/run.py index 807ad12..8f29f35 100644 --- a/budget/run.py +++ b/budget/run.py @@ -3,7 +3,7 @@ import warnings from flask import Flask, g, request, session from flask.ext.babel import Babel -from flask.ext.migrate import Migrate, upgrade +from flask.ext.migrate import Migrate, upgrade, stamp from raven.contrib.flask import Sentry from web import main, db, mail @@ -15,6 +15,15 @@ from utils import minimal_round app = Flask(__name__) +def pre_alembic_db(): + """ Checks if we are migrating from a pre-alembic ihatemoney + """ + con = db.engine.connect() + tables_exist = db.engine.dialect.has_table(con, 'project') + alembic_setup = db.engine.dialect.has_table(con, 'alembic_version') + return tables_exist and not alembic_setup + + def configure(): """ A way to (re)configure the app, specially reset the settings """ @@ -49,6 +58,11 @@ db.app = app # db migrations migrate = Migrate(app, db) +if pre_alembic_db(): + with app.app_context(): + # fake the first migration + stamp(revision='b9a10d5d63ce') + # auto-execute migrations on runtime with app.app_context(): upgrade()