Merge pull request #88 from schireson/dc/sqlachemy-2-compat

Author: DanCardin

.github/workflows/build.yml

@@ -23,9 +23,9 @@ jobs:
       fail-fast: false
       matrix:
         python-version: ["3.7", "3.8", "3.9", "3.10", "3.11"]
-        pytest-version: ["6.2", "7.0"]
-        pytest-asyncio-version: ["0.16", "0.19"]
-        sqlalchemy-version: ["1.3", "1.4"]
+        pytest-version: ["6.2.0", "7.0.0"]
+        pytest-asyncio-version: ["0.16.0", "0.19.0"]
+        sqlalchemy-version: ["1.3.0", "1.4.0", "2.0.0"]
 
     steps:
       - uses: actions/checkout@v3

Makefile

@@ -8,20 +8,20 @@ build:
 	poetry build
 
 test:
-	COVERAGE_PROCESS_START="$(PWD)/pyproject.toml" \
+	SQLALCHEMY_WARN_20=1 COVERAGE_PROCESS_START="$(PWD)/pyproject.toml" \
 	coverage run -m pytest src tests -vv
 	coverage combine
 	coverage report -i
 	coverage xml
 
 lint:
-	ruff src tests || exit 1
-	black --check src tests || exit 1
+	ruff src tests examples || exit 1
+	black --check src tests examples || exit 1
 	mypy src tests || exit 1
 
 format:
-	ruff --fix src tests
-	black src tests
+	ruff --fix src tests examples
+	black src tests examples
 
 publish: build
 	poetry publish -u __token__ -p '${PYPI_TOKEN}' --no-interaction

docs/source/asyncio.rst

@@ -109,7 +109,8 @@ thus fixes :ref:`test_downgrade_leaves_no_trace`.
        if isinstance(connectable, AsyncEngine):
            asyncio.run(run_async_migrations(connectable))
        else:
-           do_run_migrations(connectable)
+           with connectable.connect() as connection:
+               do_run_migrations(connection)
    
    
    # Then use their setup for async connection/running of the migration

docs/source/experimental_tests.rst

@@ -71,7 +71,11 @@ after its definition.
 
    import sqlalchemy
    from sqlalchemy import Column, types
-   from sqlalchemy.ext.declarative import declarative_base
+
+   try:
+       from sqlalchemy.orm import declarative_base
+   except ImportError:
+       from sqlalchemy.ext.declarative import declarative_base
 
    Base = declarative_base()
 

examples/test_alternative_script_location/alembic/env.py

@@ -1,9 +1,8 @@
 from logging.config import fileConfig
 
 from alembic import context
-from sqlalchemy import engine_from_config, pool
-
 from models import Base
+from sqlalchemy import engine_from_config, pool
 
 fileConfig(context.config.config_file_name)
 target_metadata = Base.metadata

examples/test_alternative_script_location/alembic/versions/aaaaaaaaaaaa_create_foo.py

@@ -1,5 +1,5 @@
-from alembic import op
 import sqlalchemy as sa
+from alembic import op
 
 revision = "aaaaaaaaaaaa"
 down_revision = None

examples/test_alternative_script_location/alembic/versions/bbbbbbbbbbbb_update_foo.py

@@ -1,4 +1,5 @@
 from alembic import op
+from sqlalchemy import text
 
 revision = "bbbbbbbbbbbb"
 down_revision = "aaaaaaaaaaaa"
@@ -8,7 +9,7 @@
 
 def upgrade():
     conn = op.get_bind()
-    result = conn.execute("SELECT * FROM foo").fetchall()
+    result = conn.execute(text("SELECT * FROM foo")).fetchall()
     assert len(result) == 0
 
 

examples/test_alternative_script_location/models.py

@@ -1,6 +1,11 @@
 import sqlalchemy
 from sqlalchemy import Column, types
-from sqlalchemy.ext.declarative import declarative_base
+
+try:
+    from sqlalchemy.orm import declarative_base
+except ImportError:
+    from sqlalchemy.ext.declarative import declarative_base
+
 
 Base = declarative_base()
 

examples/test_ambiguous_downgrade_history/migrations/env.py

@@ -1,9 +1,8 @@
 from logging.config import fileConfig
 
 from alembic import context
-from sqlalchemy import engine_from_config, pool
-
 from models import Base
+from sqlalchemy import engine_from_config, pool
 
 fileConfig(context.config.config_file_name)
 target_metadata = Base.metadata

examples/test_ambiguous_downgrade_history/migrations/versions/10483a8dd3c8.py

@@ -1,14 +1,14 @@
-"""
-Revision ID: 10483a8dd3c8
+"""Revision ID: 10483a8dd3c8
 Revises: first
-Create Date: 2020-07-01 16:44:05.244310
+Create Date: 2020-07-01 16:44:05.244310.
 """
 # revision identifiers, used by Alembic.
-revision = '10483a8dd3c8'
-down_revision = 'first'
+revision = "10483a8dd3c8"
+down_revision = "first"
 branch_labels = None
 depends_on = None
 
+
 def upgrade():
     pass