Add isort

Author: LuckySting

README.md

@@ -56,6 +56,7 @@ $ tox -e style
 
 Reformat code:
 ```bash
+$ tox -e isort
 $ tox -e black-format
 ```
 

examples/example.py

@@ -1,11 +1,10 @@
 import datetime
 import logging
-import sqlalchemy as sa
-from sqlalchemy import orm, exc, sql
-from sqlalchemy import Table, Column, Integer, String, Float, TIMESTAMP
 
+import sqlalchemy as sa
 from fill_tables import fill_all_tables, to_days
-from models import Base, Series, Episodes
+from models import Base, Episodes, Series
+from sqlalchemy import TIMESTAMP, Column, Float, Integer, String, Table, exc, orm, sql
 
 
 def describe_table(engine, name):

examples/fill_tables.py

@@ -1,7 +1,6 @@
 import iso8601
-
 import sqlalchemy as sa
-from models import Base, Series, Seasons, Episodes
+from models import Base, Episodes, Seasons, Series
 
 
 def to_days(date):

examples/models.py

@@ -1,7 +1,6 @@
 import sqlalchemy.orm as orm
 from sqlalchemy import Column, Integer, Unicode
 
-
 Base = orm.declarative_base()
 
 

pyproject.toml

@@ -1,2 +1,5 @@
 [tool.black]
 line-length = 120
+
+[tool.isort]
+profile = "black"

test-requirements.txt

@@ -10,3 +10,4 @@ flake8==3.9.2
 black==23.3.0
 pytest-cov
 pytest-asyncio
+isort==5.13.2

test/test_core.py

@@ -4,17 +4,15 @@
 from typing import NamedTuple
 
 import pytest
-
 import sqlalchemy as sa
-from sqlalchemy import Table, Column, Integer, Unicode, String
-from sqlalchemy.testing.fixtures import TestBase, TablesTest, config
-
 import ydb
+from sqlalchemy import Column, Integer, String, Table, Unicode
+from sqlalchemy.testing.fixtures import TablesTest, TestBase, config
 from ydb._grpc.v4.protos import ydb_common_pb2
 
-from ydb_sqlalchemy import dbapi, IsolationLevel
-from ydb_sqlalchemy.sqlalchemy import types
+from ydb_sqlalchemy import IsolationLevel, dbapi
 from ydb_sqlalchemy import sqlalchemy as ydb_sa
+from ydb_sqlalchemy.sqlalchemy import types
 
 
 def clear_sql(stm):

test/test_inspect.py

@@ -1,6 +1,5 @@
 import sqlalchemy as sa
-
-from sqlalchemy import Table, Column, Integer, Unicode, Numeric
+from sqlalchemy import Column, Integer, Numeric, Table, Unicode
 from sqlalchemy.testing.fixtures import TablesTest
 
 

test/test_suite.py

@@ -1,55 +1,86 @@
 import pytest
 import sqlalchemy as sa
 import sqlalchemy.testing.suite.test_types
-
+from sqlalchemy.testing import is_false, is_true
 from sqlalchemy.testing.suite import *  # noqa: F401, F403
-
-from sqlalchemy.testing import is_true, is_false
-from sqlalchemy.testing.suite import eq_, testing, inspect, provide_metadata, config, requirements, fixtures
-from sqlalchemy.testing.suite import func, column, literal_column, select, exists
-from sqlalchemy.testing.suite import MetaData, Column, Table, Integer, String
-
-from sqlalchemy.testing.suite.test_select import (
-    ExistsTest as _ExistsTest,
-    LikeFunctionsTest as _LikeFunctionsTest,
-    CompoundSelectTest as _CompoundSelectTest,
+from sqlalchemy.testing.suite import (
+    Column,
+    Integer,
+    MetaData,
+    String,
+    Table,
+    column,
+    config,
+    eq_,
+    exists,
+    fixtures,
+    func,
+    inspect,
+    literal_column,
+    provide_metadata,
+    requirements,
+    select,
+    testing,
+)
+from sqlalchemy.testing.suite.test_ddl import (
+    LongNameBlowoutTest as _LongNameBlowoutTest,
+)
+from sqlalchemy.testing.suite.test_deprecations import (
+    DeprecatedCompoundSelectTest as _DeprecatedCompoundSelectTest,
+)
+from sqlalchemy.testing.suite.test_dialect import (
+    DifficultParametersTest as _DifficultParametersTest,
+)
+from sqlalchemy.testing.suite.test_dialect import EscapingTest as _EscapingTest
+from sqlalchemy.testing.suite.test_insert import (
+    InsertBehaviorTest as _InsertBehaviorTest,
 )
 from sqlalchemy.testing.suite.test_reflection import (
-    HasTableTest as _HasTableTest,
-    HasIndexTest as _HasIndexTest,
     ComponentReflectionTest as _ComponentReflectionTest,
-    CompositeKeyReflectionTest as _CompositeKeyReflectionTest,
+)
+from sqlalchemy.testing.suite.test_reflection import (
     ComponentReflectionTestExtra as _ComponentReflectionTestExtra,
+)
+from sqlalchemy.testing.suite.test_reflection import (
+    CompositeKeyReflectionTest as _CompositeKeyReflectionTest,
+)
+from sqlalchemy.testing.suite.test_reflection import HasIndexTest as _HasIndexTest
+from sqlalchemy.testing.suite.test_reflection import HasTableTest as _HasTableTest
+from sqlalchemy.testing.suite.test_reflection import (
     QuotedNameArgumentTest as _QuotedNameArgumentTest,
 )
+from sqlalchemy.testing.suite.test_results import RowFetchTest as _RowFetchTest
+from sqlalchemy.testing.suite.test_select import (
+    CompoundSelectTest as _CompoundSelectTest,
+)
+from sqlalchemy.testing.suite.test_select import ExistsTest as _ExistsTest
+from sqlalchemy.testing.suite.test_select import (
+    FetchLimitOffsetTest as _FetchLimitOffsetTest,
+)
+from sqlalchemy.testing.suite.test_select import JoinTest as _JoinTest
+from sqlalchemy.testing.suite.test_select import LikeFunctionsTest as _LikeFunctionsTest
+from sqlalchemy.testing.suite.test_select import OrderByLabelTest as _OrderByLabelTest
+from sqlalchemy.testing.suite.test_types import BinaryTest as _BinaryTest
+from sqlalchemy.testing.suite.test_types import DateTest as _DateTest
 from sqlalchemy.testing.suite.test_types import (
-    IntegerTest as _IntegerTest,
-    NumericTest as _NumericTest,
-    BinaryTest as _BinaryTest,
-    TrueDivTest as _TrueDivTest,
-    TimeTest as _TimeTest,
-    StringTest as _StringTest,
-    NativeUUIDTest as _NativeUUIDTest,
-    TimeMicrosecondsTest as _TimeMicrosecondsTest,
     DateTimeCoercedToDateTimeTest as _DateTimeCoercedToDateTimeTest,
-    DateTest as _DateTest,
+)
+from sqlalchemy.testing.suite.test_types import (
     DateTimeMicrosecondsTest as _DateTimeMicrosecondsTest,
-    DateTimeTest as _DateTimeTest,
-    TimestampMicrosecondsTest as _TimestampMicrosecondsTest,
 )
-from sqlalchemy.testing.suite.test_dialect import (
-    EscapingTest as _EscapingTest,
-    DifficultParametersTest as _DifficultParametersTest,
+from sqlalchemy.testing.suite.test_types import DateTimeTest as _DateTimeTest
+from sqlalchemy.testing.suite.test_types import IntegerTest as _IntegerTest
+from sqlalchemy.testing.suite.test_types import NativeUUIDTest as _NativeUUIDTest
+from sqlalchemy.testing.suite.test_types import NumericTest as _NumericTest
+from sqlalchemy.testing.suite.test_types import StringTest as _StringTest
+from sqlalchemy.testing.suite.test_types import (
+    TimeMicrosecondsTest as _TimeMicrosecondsTest,
 )
-from sqlalchemy.testing.suite.test_select import (
-    JoinTest as _JoinTest,
-    OrderByLabelTest as _OrderByLabelTest,
-    FetchLimitOffsetTest as _FetchLimitOffsetTest,
+from sqlalchemy.testing.suite.test_types import (
+    TimestampMicrosecondsTest as _TimestampMicrosecondsTest,
 )
-from sqlalchemy.testing.suite.test_insert import InsertBehaviorTest as _InsertBehaviorTest
-from sqlalchemy.testing.suite.test_ddl import LongNameBlowoutTest as _LongNameBlowoutTest
-from sqlalchemy.testing.suite.test_results import RowFetchTest as _RowFetchTest
-from sqlalchemy.testing.suite.test_deprecations import DeprecatedCompoundSelectTest as _DeprecatedCompoundSelectTest
+from sqlalchemy.testing.suite.test_types import TimeTest as _TimeTest
+from sqlalchemy.testing.suite.test_types import TrueDivTest as _TrueDivTest
 
 from ydb_sqlalchemy.sqlalchemy import types as ydb_sa_types
 

test_dbapi/test_dbapi.py

@@ -1,11 +1,11 @@
+from contextlib import suppress
+
 import pytest
 import pytest_asyncio
-
+import sqlalchemy.util as util
 import ydb
-import ydb_sqlalchemy.dbapi as dbapi
 
-from contextlib import suppress
-import sqlalchemy.util as util
+import ydb_sqlalchemy.dbapi as dbapi
 
 
 class BaseDBApiTestSuit: