443b3e3262
Log messages are no longer being translated. This removes all use of the _LE, _LI, and _LW translation markers to simplify logging and to avoid confusion with new contributions. See: http://lists.openstack.org/pipermail/openstack-i18n/2016-November/002574.html http://lists.openstack.org/pipermail/openstack-dev/2017-March/113365.html Closes-Bug: #1674577 Change-Id: I7553ae7c222f61b6796b72ac4a2b744d3e08fee6
80 lines
2.7 KiB
Python
80 lines
2.7 KiB
Python
# 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 logging
|
|
import os
|
|
|
|
from migrate.versioning import version as migrate_version
|
|
|
|
from oslo_db.sqlalchemy import migration
|
|
from oslo_db.sqlalchemy.migration_cli import ext_base
|
|
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
|
|
class MigrateExtension(ext_base.MigrationExtensionBase):
|
|
"""Extension to provide sqlalchemy-migrate features.
|
|
|
|
:param migration_config: Stores specific configuration for migrations
|
|
:type migration_config: dict
|
|
"""
|
|
|
|
order = 1
|
|
|
|
def __init__(self, engine, migration_config):
|
|
self.engine = engine
|
|
self.repository = migration_config.get('migration_repo_path', '')
|
|
self.init_version = migration_config.get('init_version', 0)
|
|
|
|
@property
|
|
def enabled(self):
|
|
return os.path.exists(self.repository)
|
|
|
|
def upgrade(self, version):
|
|
version = None if version == 'head' else version
|
|
return migration.db_sync(
|
|
self.engine, self.repository, version,
|
|
init_version=self.init_version)
|
|
|
|
def downgrade(self, version):
|
|
try:
|
|
# version for migrate should be valid int - else skip
|
|
if version in ('base', None):
|
|
version = self.init_version
|
|
version = int(version)
|
|
return migration.db_sync(
|
|
self.engine, self.repository, version,
|
|
init_version=self.init_version)
|
|
except ValueError:
|
|
LOG.error(
|
|
'Migration number for migrate plugin must be valid '
|
|
'integer or empty, if you want to downgrade '
|
|
'to initial state'
|
|
)
|
|
raise
|
|
|
|
def version(self):
|
|
return migration.db_version(
|
|
self.engine, self.repository, init_version=self.init_version)
|
|
|
|
def has_revision(self, rev_id):
|
|
collection = migrate_version.Collection(self.repository)
|
|
try:
|
|
collection.version(rev_id)
|
|
return True
|
|
except (KeyError, ValueError):
|
|
# NOTE(breton): migrate raises KeyError if an int is passed but not
|
|
# found in the list of revisions and ValueError if non-int is
|
|
# passed. Both mean there is no requested revision.
|
|
return False
|