From 8cb0bea57c67fba230346d387f1d950abe7790bd Mon Sep 17 00:00:00 2001 From: Maxime Beauchemin Date: Thu, 22 Sep 2016 14:09:00 -0700 Subject: [PATCH] [sqllab] db migration - setting Database.allow_run_sync=True (#1174) --- .../eca4694defa7_sqllab_setting_defaults.py | 41 +++++++++++++++++++ 1 file changed, 41 insertions(+) create mode 100644 caravel/migrations/versions/eca4694defa7_sqllab_setting_defaults.py diff --git a/caravel/migrations/versions/eca4694defa7_sqllab_setting_defaults.py b/caravel/migrations/versions/eca4694defa7_sqllab_setting_defaults.py new file mode 100644 index 0000000000..41c84267d0 --- /dev/null +++ b/caravel/migrations/versions/eca4694defa7_sqllab_setting_defaults.py @@ -0,0 +1,41 @@ +"""sqllab_setting_defaults + +Revision ID: eca4694defa7 +Revises: 5e4a03ef0bf0 +Create Date: 2016-09-22 11:31:50.543820 + +""" +from alembic import op +from caravel import db +from sqlalchemy.ext.declarative import declarative_base +from sqlalchemy import (Column, Integer, Boolean) + +# revision identifiers, used by Alembic. +revision = 'eca4694defa7' +down_revision = '5e4a03ef0bf0' + +Base = declarative_base() + + +class Database(Base): + + """An ORM object that stores Database related information""" + + __tablename__ = 'dbs' + id = Column(Integer, primary_key=True) + allow_run_sync = Column(Boolean, default=True) + + +def upgrade(): + bind = op.get_bind() + session = db.Session(bind=bind) + + for obj in session.query(Database).all(): + obj.allow_run_sync = True + + session.commit() + session.close() + + +def downgrade(): + pass