From e70504bdee8b1c15ca4908d97a203098b2718f05 Mon Sep 17 00:00:00 2001 From: Robert Lechte Date: Wed, 1 Jul 2020 11:15:52 +1000 Subject: [PATCH] refmt --- migra/changes.py | 8 +++++--- migra/migra.py | 3 +-- tests/test_migra.py | 2 +- 3 files changed, 7 insertions(+), 6 deletions(-) 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;