diff --git a/migra/changes.py b/migra/changes.py index d8d7963..73dab48 100644 --- a/migra/changes.py +++ b/migra/changes.py @@ -240,7 +240,9 @@ def get_table_changes( rls_alter = v.alter_rls_statement statements.append(rls_alter) - seq_created, seq_dropped, seq_modified, _ = differences(sequences_from, sequences_target) + seq_created, seq_dropped, seq_modified, _ = differences( + sequences_from, sequences_target + ) for k in seq_created: seq_b = sequences_target[k] @@ -472,12 +474,12 @@ class Changes(object): self.i_target.enums, ) - elif name == 'sequences': + elif name == "sequences": return partial( statements_for_changes, getattr(self.i_from, name), getattr(self.i_target, name), - modifications=False + modifications=False, ) elif name in THINGS: diff --git a/migra/migra.py b/migra/migra.py index 68e20ef..ac04acb 100644 --- a/migra/migra.py +++ b/migra/migra.py @@ -1,8 +1,7 @@ from __future__ import unicode_literals -from sqlbag import raw_execute - from schemainspect import DBInspector, get_inspector +from sqlbag import raw_execute from .changes import Changes from .statements import Statements diff --git a/tests/test_migra.py b/tests/test_migra.py index 55a80c3..744c784 100644 --- a/tests/test_migra.py +++ b/tests/test_migra.py @@ -3,11 +3,11 @@ from __future__ import unicode_literals import io from pytest import raises +from schemainspect import get_inspector from sqlbag import S, load_sql_from_file, temporary_database from migra import Migration, Statements, UnsafeMigrationException from migra.command import parse_args, run -from schemainspect import get_inspector SQL = """select 1;