db: replace SQLAlchemy deprecated contextual_connect method

petl-developers · Mar 24, 2021 · 46a98e6 · 46a98e6
1 parent ff6e7a3
commit 46a98e6
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/petl/io/db.py b/petl/io/db.py
@@ -183,7 +183,7 @@ def _iter_dbapi_cursor(cursor, query, *args, **kwargs):
 
 
 def _iter_sqlalchemy_engine(engine, query, *args, **kwargs):
-    return _iter_sqlalchemy_connection(engine.contextual_connect(), query,
+    return _iter_sqlalchemy_connection(engine.connect(), query,
                                        *args, **kwargs)
 
 
@@ -541,7 +541,7 @@ def _todb_dbapi_cursor(table, cursor, tablename, schema=None, commit=True,
 def _todb_sqlalchemy_engine(table, engine, tablename, schema=None, commit=True,
                             truncate=False):
 
-    _todb_sqlalchemy_connection(table, engine.contextual_connect(), tablename,
+    _todb_sqlalchemy_connection(table, engine.connect(), tablename,
                                 schema=schema, commit=commit, truncate=truncate)
 
 

diff --git a/petl/io/db_create.py b/petl/io/db_create.py
@@ -370,7 +370,7 @@ def _execute_sqlalchemy_connection(sql, connection, commit):
 
 
 def _execute_sqlalchemy_engine(sql, engine, commit):
-    _execute_sqlalchemy_connection(sql, engine.contextual_connect(), commit)
+    _execute_sqlalchemy_connection(sql, engine.connect(), commit)
 
 
 def _execute_sqlalchemy_session(sql, session, commit):

diff --git a/petl/io/db_utils.py b/petl/io/db_utils.py
@@ -22,7 +22,7 @@ def _is_dbapi_cursor(dbo):
 
 
 def _is_sqlalchemy_engine(dbo):
-    return (_hasmethods(dbo, 'execute', 'contextual_connect', 'raw_connection')
+    return (_hasmethods(dbo, 'execute', 'connect', 'raw_connection')
             and _hasprop(dbo, 'driver'))