Source code for keystone.common.sql.migrate_repo.versions.021_add_trust_to_token

# Copyright 2013 OpenStack Foundation
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.

import sqlalchemy
from sqlalchemy import exc


[docs]def downgrade_token_table_with_column_drop(meta, migrate_engine): token_table = sqlalchemy.Table('token', meta, autoload=True) #delete old tokens, as the format has changed. #We don't guarantee that existing tokens will be #usable after a migration token_table.delete() token_table.drop_column( sqlalchemy.Column('trust_id', sqlalchemy.String(64), nullable=True)) token_table.drop_column( sqlalchemy.Column('user_id', sqlalchemy.String(64)))
[docs]def create_column_forgiving(migrate_engine, table, column): try: table.create_column(column) except exc.OperationalError as e: if (e.args[0].endswith('duplicate column name: %s' % column.name) and migrate_engine.name == "sqlite"): #sqlite does not drop columns, so if we have already #done a downgrade and are now upgrading, we will hit #this: the SQLite driver previously reported success #dropping the columns but it hasn't. pass else: raise
[docs]def upgrade_token_table(meta, migrate_engine): #delete old tokens, as the format has changed. #The existing tokens will not #support some of the list functions token_table = sqlalchemy.Table('token', meta, autoload=True) token_table.delete() create_column_forgiving( migrate_engine, token_table, sqlalchemy.Column('trust_id', sqlalchemy.String(64), nullable=True)) create_column_forgiving( migrate_engine, token_table, sqlalchemy.Column('user_id', sqlalchemy.String(64)))
[docs]def upgrade(migrate_engine): meta = sqlalchemy.MetaData() meta.bind = migrate_engine upgrade_token_table(meta, migrate_engine)
[docs]def downgrade(migrate_engine): meta = sqlalchemy.MetaData() meta.bind = migrate_engine downgrade_token_table_with_column_drop(meta, migrate_engine)