[sqllab] db migration - setting Database.allow_run_sync=True (#1174)

This commit is contained in:
Maxime Beauchemin 2016-09-22 14:09:00 -07:00 committed by GitHub
parent 1fa18922fa
commit 8cb0bea57c
1 changed files with 41 additions and 0 deletions

View File

@ -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