Adjust for SQLAlchemy 1

Author: edgarrmondragon

pyproject.toml

@@ -110,6 +110,7 @@ skip-string-normalization = true
 ignore = [
   "ANN101",  # missing-type-self
   "ANN102",  # missing-type-cls
+  "FIX002",  # line-contains-todo
 ]
 line-length = 100
 select = [

src/sqlean_driver/__init__.py

@@ -84,7 +84,7 @@ class SQLeanDialect(SQLiteDialect_pysqlite):
 
     driver = "sqlean"
     supports_statement_cache = True
-    type_compiler_cls = SQLeanTypeCompiler
+    type_compiler = SQLeanTypeCompiler
 
     @classmethod
     def dbapi(cls) -> ModuleType:  # type: ignore[override]

src/sqlean_driver/custom_types.py

@@ -71,9 +71,13 @@ def result_processor(
         """Return a result processor."""
         return none_or_ip_interface
 
+    # TODO(edgarrmondragon): Add missing type parameters:
+    # > sqltypes.Indexable.Comparator[IPAddress]
+    # > sqltypes.Concatenable.Comparator[IPAddress]
+    # https://github.com/edgarrmondragon/sqlean-driver/issues/37
     class Comparator(
-        sqltypes.Indexable.Comparator[IPAddress],
-        sqltypes.Concatenable.Comparator[IPAddress],
+        sqltypes.Indexable.Comparator,  # type: ignore[type-arg]
+        sqltypes.Concatenable.Comparator,  # type: ignore[type-arg]
     ):
         """Comparator for the INET type."""
 

tests/test_types.py

@@ -87,7 +87,7 @@ def test_ipaddr_types(
     """Test that the types work."""
     engine = create_engine("sqlite+sqlean:///:memory:?extensions=ipaddr")
     metadata.create_all(engine)
-    with engine.connect() as conn:
+    with engine.connect() as conn, conn.begin():
         conn.execute(table.insert(), data)
         result = conn.execute(query)
         assert result.fetchone() == expected
@@ -112,7 +112,7 @@ def test_uuid_types(
     """Test that the types work."""
     engine = create_engine("sqlite+sqlean:///:memory:?extensions=uuid")
     metadata.create_all(engine)
-    with engine.connect() as conn:
+    with engine.connect() as conn, conn.begin():
         conn.execute(table.insert(), data)
         result = conn.execute(query)
         assert result.fetchone() == expected