Example #1
0
from mig.models import MigrationData

from talkatv import db
from talkatv.models import MODELS
from talkatv.migrations import MIGRATIONS

root_log = logging.getLogger()
root_log.setLevel(logging.DEBUG)
logging.basicConfig()

_log = logging.getLogger(__name__)


def check_or_create_mig_data():
    if not db.engine.dialect.has_table(db.session, 'mig__data'):
        _log.info('Creating MigrationData table...')
        MigrationData.__table__.create(db.engine)

        # Create the first migration, so that mig doesn't init.
        migration = MigrationData(name=u'__main__', version=0)
        db.session.add(migration)
        db.session.commit()


if __name__ == '__main__':
    if db.engine.dialect.has_table(db.session, 'user'):
        # The DB is already populated, check if migrations are active
        check_or_create_mig_data()

    run(db.engine, u'__main__', MODELS, MIGRATIONS)
Example #2
0
def main():
    run(db.engine, u'__main__', models.MODELS, migrations.MIGRATIONS)
    run(db.engine, u'www', wwwmodels.MODELS, wwwmigrations.MIGRATIONS)