diff options
Diffstat (limited to 'alembic/autogenerate/api.py')
-rw-r--r-- | alembic/autogenerate/api.py | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/alembic/autogenerate/api.py b/alembic/autogenerate/api.py index 60e2100..cc5debe 100644 --- a/alembic/autogenerate/api.py +++ b/alembic/autogenerate/api.py @@ -13,7 +13,7 @@ from sqlalchemy.util import OrderedSet from .compare import _compare_tables from .render import _drop_table, _drop_column, _drop_index, _drop_constraint, \ _add_table, _add_column, _add_index, _add_constraint, _modify_col, \ - _add_fk_constraint, _drop_fk_constraint + _add_fk_constraint from .. import util log = logging.getLogger(__name__) @@ -294,7 +294,7 @@ def _invoke_adddrop_command(updown, args, autogen_context): "column": (_drop_column, _add_column), "index": (_drop_index, _add_index), "constraint": (_drop_constraint, _add_constraint), - "fk":(_drop_fk_constraint, _add_fk_constraint) + "fk": (_drop_constraint, _add_fk_constraint) } cmd_callables = _commands[cmd_type] @@ -346,8 +346,7 @@ def _group_diffs_by_table(diffs): "column": lambda diff: (diff[0], diff[1]), "index": lambda diff: (diff[0].table.schema, diff[0].table.name), "constraint": lambda diff: (diff[0].table.schema, diff[0].table.name), - "fk": lambda diff: - (diff[0].parent.table.schema, diff[0].parent.table.name) + "fk": lambda diff: (diff[0].parent.schema, diff[0].parent.name) } def _derive_table(diff): |