diff options
| author | Jenkins <jenkins@review.openstack.org> | 2014-03-04 20:38:53 +0000 |
|---|---|---|
| committer | Gerrit Code Review <review@openstack.org> | 2014-03-04 20:38:53 +0000 |
| commit | 716124f2adfe2945ff03d6501a7eb8636b271bf7 (patch) | |
| tree | eeb564a5a276764238966f01e1f7437e2cfdf377 | |
| parent | f944414bd4b393ecbcab1854f970989df90d783c (diff) | |
| parent | 12a6bcfa8c1ead90f22135ea7ab75352ed06d6b2 (diff) | |
| download | sqalchemy-migrate-716124f2adfe2945ff03d6501a7eb8636b271bf7.tar.gz | |
Merge "Conditionally import ibmdb2/ibm_db_sa"
| -rw-r--r-- | migrate/changeset/databases/visitor.py | 14 |
1 files changed, 11 insertions, 3 deletions
diff --git a/migrate/changeset/databases/visitor.py b/migrate/changeset/databases/visitor.py index 224c40f..c70aa6b 100644 --- a/migrate/changeset/databases/visitor.py +++ b/migrate/changeset/databases/visitor.py @@ -8,8 +8,7 @@ from migrate.changeset.databases import (sqlite, postgres, mysql, oracle, - firebird, - ibmdb2) + firebird) # Map SA dialects to the corresponding Migrate extensions @@ -21,10 +20,19 @@ DIALECTS = { "mysql": mysql.MySQLDialect, "oracle": oracle.OracleDialect, "firebird": firebird.FBDialect, - "ibm_db_sa": ibmdb2.IBMDBDialect } +# NOTE(mriedem): We have to conditionally check for DB2 in case ibm_db_sa +# isn't available since ibm_db_sa is not packaged in sqlalchemy like the +# other dialects. +try: + from migrate.changeset.databases import ibmdb2 + DIALECTS["ibm_db_sa"] = ibmdb2.IBMDBDialect +except ImportError: + pass + + def get_engine_visitor(engine, name): """ Get the visitor implementation for the given database engine. |