summaryrefslogtreecommitdiff
path: root/migrate/changeset/databases/firebird.py
diff options
context:
space:
mode:
Diffstat (limited to 'migrate/changeset/databases/firebird.py')
-rw-r--r--migrate/changeset/databases/firebird.py10
1 files changed, 6 insertions, 4 deletions
diff --git a/migrate/changeset/databases/firebird.py b/migrate/changeset/databases/firebird.py
index 5eacd58..a8831bf 100644
--- a/migrate/changeset/databases/firebird.py
+++ b/migrate/changeset/databases/firebird.py
@@ -2,11 +2,13 @@
Firebird database specific implementations of changeset classes.
"""
-from migrate.changeset import ansisql, exceptions
-
-# TODO: SQLA 0.6 has not migrated the FB dialect over yet
+from migrate.changeset import ansisql, exceptions, SQLA_06
from sqlalchemy.databases import firebird as sa_base
-FBSchemaGenerator = sa_base.FBSchemaGenerator
+
+if SQLA_06:
+ FBSchemaGenerator = sa_base.FBDDLCompiler
+else:
+ FBSchemaGenerator = sa_base.FBSchemaGenerator
class FBColumnGenerator(FBSchemaGenerator, ansisql.ANSIColumnGenerator):
"""Firebird column generator implementation."""