fix(2.x): session get bind signature

CHANGES.rst

@@ -1,3 +1,11 @@
+Version 2.5.2
+-------------
+
+Unreleased
+
+-   Fix session ``get_bind`` signature for SQLAlchemy 1.4. :issue:`953`
+
+
 Version 2.5.1
 -------------
 

flask_sqlalchemy/__init__.py

@@ -184,7 +184,12 @@ def __init__(self, db, autocommit=False, autoflush=True, **options):
             bind=bind, binds=binds, **options
         )
 
-    def get_bind(self, mapper=None, clause=None):
+    def get_bind(
+        self,
+        mapper=None,
+        *args,
+        **kwargs
+    ):
         """Return the engine or connection for a given model or
         table, using the ``__bind_key__`` if it is set.
         """
@@ -202,7 +207,12 @@ def get_bind(self, mapper=None, clause=None):
             if bind_key is not None:
                 state = get_state(self.app)
                 return state.db.get_engine(self.app, bind=bind_key)
-        return SessionBase.get_bind(self, mapper, clause)
+        return SessionBase.get_bind(
+            self,
+            mapper,
+            *args,
+            **kwargs
+        )
 
 
 class _SessionSignalEvents(object):

tests/test_sessions.py

@@ -1,4 +1,5 @@
 import sqlalchemy as sa
+from sqlalchemy.engine import Engine
 from sqlalchemy.orm import sessionmaker
 
 import flask_sqlalchemy as fsa
@@ -63,3 +64,8 @@ class QazWsx(db.Model):
 
 def test_listen_to_session_event(db):
     sa.event.listen(db.session, 'after_commit', lambda session: None)
+
+
+def test_session_get_bind(app, db):
+    with app.test_request_context():
+        assert isinstance(db.session.get_bind(), Engine)