79 lines
2.7 KiB
Python
79 lines
2.7 KiB
Python
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
|
|
|
# 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
|
|
|
|
|
|
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)))
|
|
|
|
|
|
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
|
|
|
|
|
|
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)))
|
|
|
|
|
|
def upgrade(migrate_engine):
|
|
meta = sqlalchemy.MetaData()
|
|
meta.bind = migrate_engine
|
|
upgrade_token_table(meta, migrate_engine)
|
|
|
|
|
|
def downgrade(migrate_engine):
|
|
meta = sqlalchemy.MetaData()
|
|
meta.bind = migrate_engine
|
|
downgrade_token_table_with_column_drop(meta, migrate_engine)
|