python类get_bind()的实例源码

def downgrade():
    session = Session(bind=op.get_bind())
    pods = session.query(MenuItem).filter(MenuItem.name=='Pods').one()
    pods.path = '/'
    setts = session.query(MenuItem).filter(MenuItem.name=='Settings').first()
    if setts is not None:
        session.add_all(MenuItemRole(role=role, menuitem=setts)
            for role in session.query(Role).filter(Role.rolename.in_(['User', 'TrialUser'])))
    session.commit()

def upgrade():
    ### commands auto generated by Alembic - please adjust! ###
    op.add_column(
        'persistent_disk',
        sa.Column('state', sa.Integer(), nullable=False, server_default='0')
    )
    conn = op.get_bind()
    conn.execute("UPDATE persistent_disk SET state=0")
    ### end Alembic commands ###

def upgrade():
    ### commands auto generated by Alembic - please adjust! ###
    op.add_column('kubes', sa.Column('included_traffic', sa.Integer(), server_default='0', nullable=False))
    conn = op.get_bind()
    conn.execute("UPDATE kubes SET included_traffic=total_traffic")
    op.drop_column('kubes', 'total_traffic')
    ### end Alembic commands ###

def upgrade():
    bind = op.get_bind()
    session = Session(bind=bind)

    ### commands auto generated by Alembic - please adjust! ###
    op.add_column(
        'container_states',
        sa.Column('docker_id', sa.String(length=80), server_default='unknown',
                  nullable=False)
    )
    op.execute("ALTER TABLE container_states DROP CONSTRAINT container_states_pkey, "\
               "ADD CONSTRAINT container_states_pkey PRIMARY KEY "\
                    "(pod_id, container_name, docker_id, kubes, start_time);")
    # op.execute()
    try:
        # Try to get docker_id for current container states.
        # 1. Extract all DB states without end time
        # 2. Get pods information for those containers from kubes-api
        # 3. Set docker_id for selected DB-states
        states = session.query(ContainerState).filter(
            ContainerState.end_time == None)
        containers = {
            (item.pod_id, item.container_name): item for item in states
        }
        container_ids = _get_container_ids()
        for key, state in containers:
            if key not in container_ids:
                continue
            state.docker_id = container_ids[key]
        session.commit()
    except Exception as err:
        # We will not break the migration in case of failed update of
        # some docker_id fields. Just warn about it.
        logger.warning(
            u'Failed to set actual docker_id for currently running containers: %s',
            err)

    ### end Alembic commands ###

def upgrade():
    bind = op.get_bind()
    session = Session(bind=bind)
    session._model_changes = False  # workaround for Flask-SQLAlchemy

    m1 = Notification(type='info',
                     message='CLN_NOTIFICATION',
                     description='')
    session.add(m1)
    session.commit()

def downgrade():
    bind = op.get_bind()
    session = Session(bind=bind)
    session._model_changes = False  # workaround for Flask-SQLAlchemy
    m = session.query(Notification).filter_by(message='CLN_NOTIFICATION').first()
    if m is not None:
        session.delete(m)
    session.commit()

def upgrade():
    conn = op.get_bind()
    conn.execute("UPDATE packages SET period='month' WHERE id=0")

def downgrade():
    conn = op.get_bind()
    conn.execute("UPDATE packages SET period='hour' WHERE id=0")

def upgrade():
    session = sa.orm.sessionmaker()(bind=op.get_bind())
    permission = session.query(Permission)\
        .join(Role, Role.id == Permission.role_id).join(Resource, Permission.resource_id == Resource.id)\
        .filter(Role.rolename == 'TrialUser').filter(Resource.name == 'pods')\
        .filter(Permission.name == 'create').one()
    permission.allow = True
    session.commit()

def downgrade():
    session = sa.orm.sessionmaker()(bind=op.get_bind())
    permission = session.query(Permission)\
        .join(Role, Role.id == Permission.role_id).join(Resource, Permission.resource_id == Resource.id)\
        .filter(Role.rolename == 'TrialUser').filter(Resource.name == 'pods')\
        .filter(Permission.name == 'create').one()
    permission.allow = False
    session.commit()

def upgrade():
    conn = op.get_bind()
    op.drop_column('predefined_apps', 'user_id')
    op.add_column('pods', sa.Column(
        'template_plan_name', sa.String(24), nullable=True))
    op.create_unique_constraint('resource_role_name_unique', 'rbac_permission',
                                ['resource_id', 'role_id', 'name'])
    op.add_column(
        'system_settings', sa.Column('setting_group', sa.Text, default=''))

def downgrade():
    conn = op.get_bind()
    op.add_column('predefined_apps', sa.Column(
        'user_id',
        sa.Integer,
        sa.ForeignKey('users.id'),
        nullable=False,
        server_default='1'))
    op.drop_column('pods', 'template_plan_name')
    op.drop_constraint('resource_role_name_unique', 'rbac_permission')
    op.drop_column('system_settings', 'setting_group')

def downgrade():
    bind = op.get_bind()
    Base.metadata.bind = bind
    ### commands auto generated by Alembic - please adjust! ###
    op.add_column(u'predefined_apps', sa.Column('template', sa.TEXT(),
                                                nullable=True))
    op.drop_column(u'predefined_apps', 'is_deleted')

    downgrade_data(bind)

    op.alter_column(u'predefined_apps', u'template', nullable=False)

    op.drop_column(u'pods', 'template_version_id')
    op.drop_table('predefined_app_templates')
    ### end Alembic commands ###

def upgrade():
    conn = op.get_bind()
    conn.execute("DELETE FROM session_data")
    op.drop_column('session_data', 'data')
    op.add_column('session_data', sa.Column('user_id', sa.Integer(), nullable=False))
    op.add_column('session_data', sa.Column('role_id', sa.Integer(), nullable=False))

def upgrade():
    op.add_column('kubes', sa.Column('disk_space_units', sa.String(3),
                  server_default='MB', nullable=False))

    session = sa.orm.sessionmaker()(bind=op.get_bind())

    for kube in session.query(Kube).all():
        kube.disk_space /= 2 ** 20

    session.commit()

def downgrade():
    session = sa.orm.sessionmaker()(bind=op.get_bind())

    for kube in session.query(Kube).all():
        kube.disk_space *= 2 ** 20

    session.commit()

    op.drop_column('kubes', 'disk_space_units')

def upgrade():
    op.add_column('pods', sa.Column('template_id', sa.Integer(), nullable=True))
    bind = op.get_bind()
    session = sessionmaker()(bind=bind)
    ku = session.query(User).filter(
        User.username == KUBERDOCK_INTERNAL_USER
    ).first()
    if not ku:
        logger.warning('Internal user not found: %s', KUBERDOCK_INTERNAL_USER)
        return

    if not ku.password_hash:
        ku.password = uuid.uuid4().hex
    ku.get_token()
    session.commit()

def upgrade():
    op.add_column('predefined_apps', sa.Column('qualifier', sa.String(40),
                  nullable=True, index=True))

    Pa = sa.Table('predefined_apps', sa.MetaData(), sa.Column('qualifier'))
    session = sa.orm.Session(bind=op.get_bind())
    for pa in session.query(Pa):
        sha = sha1()
        sha.update(str(datetime.now()))
        pa.qualifier = sha.hexdigest()
    session.commit()

    op.alter_column('predefined_apps', 'qualifier', server_default='', nullable=False)

def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    conn = op.get_bind()
    conn.execute(text("""
    INSERT INTO "Permission" (name, default_value, course_permission)
    SELECT 'can_edit_own_password', true, false WHERE NOT EXISTS
        (SELECT 1 FROM "Permission" WHERE name = 'can_edit_own_password')
    """))
    # ### end Alembic commands ###

def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    conn = op.get_bind()
    conn.execute(text("""
    UPDATE "File"
    SET name =
    CASE
        WHEN extension = '' THEN name
        ELSE name || '.' || extension
    END
    WHERE is_directory = false;
    """))
    op.drop_column('File', 'extension')
    # ### end Alembic commands ###