From: Matt Riedemann <mriedem@us.ibm.com>
Date: Fri, 17 Apr 2015 15:47:35 +0000 (-0700)
Subject: Register ibm-db-alembic import for DB2 migrations
X-Git-Url: https://review.fuel-infra.org/gitweb?a=commitdiff_plain;h=952e556e4e6b0e25ca584481ef6745fc08ace970;p=openstack-build%2Fneutron-build.git

Register ibm-db-alembic import for DB2 migrations

This patch adds the optional runtime dependency to support db2
for alembic migration, will not break the default mysql path.

This is needed for the IBM DB2 third party CI to work on this
project.

Closes-Bug: 1442524

Change-Id: I9db112dbdc2d02df88ad676a6b21018f1d6f1724
---

diff --git a/neutron/db/migration/alembic_migrations/env.py b/neutron/db/migration/alembic_migrations/env.py
index 9966f55e7..7ea6c5ca5 100644
--- a/neutron/db/migration/alembic_migrations/env.py
+++ b/neutron/db/migration/alembic_migrations/env.py
@@ -23,6 +23,13 @@ from sqlalchemy import event
 from neutron.db.migration.models import head  # noqa
 from neutron.db import model_base
 
+try:
+    # NOTE(mriedem): This is to register the DB2 alembic code which
+    # is an optional runtime dependency.
+    from ibm_db_alembic.ibm_db import IbmDbImpl  # noqa # pylint: disable=unused-import
+except ImportError:
+    pass
+
 
 MYSQL_ENGINE = None