diff options
| author | Ann Kamyshnikova <akamyshnikova@mirantis.com> | 2014-11-07 11:53:31 +0300 |
|---|---|---|
| committer | Ann Kamyshnikova <akamyshnikova@mirantis.com> | 2014-11-07 11:53:31 +0300 |
| commit | 9b55ad2123fae5557705249b13ce6071208d98b3 (patch) | |
| tree | f195e1c1793373ddc872918a6e97b7c71fec4592 /alembic/autogenerate/api.py | |
| parent | 3c475c1332aff37c6aadd8599f0855dd305e35d4 (diff) | |
| download | alembic-9b55ad2123fae5557705249b13ce6071208d98b3.tar.gz | |
Update and add some tests for checking fk
fixes issue #178
Diffstat (limited to 'alembic/autogenerate/api.py')
| -rw-r--r-- | alembic/autogenerate/api.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/alembic/autogenerate/api.py b/alembic/autogenerate/api.py index 65ea3ec..72c22a1 100644 --- a/alembic/autogenerate/api.py +++ b/alembic/autogenerate/api.py @@ -8,7 +8,8 @@ from sqlalchemy.engine.reflection import Inspector 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_table, _add_column, _add_index, _add_constraint, _modify_col, \ + _add_fk_constraint from .. import util log = logging.getLogger(__name__) @@ -260,6 +261,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_constraint, _add_fk_constraint) } cmd_callables = _commands[cmd_type] |