FlaskPlugin: use get_id() instead of id attr

Following Flask-Login documentation we should use current_user.get_id() instead of current_user.id. (closes #149) https://flask-login.readthedocs.io/en/latest/#your-user-class Signed-off-by: Jiri Kuncar <[email protected]>
kvesteri · Feb 10, 2022 · 9cc42a1 · 9cc42a1
1 parent 6c16fcd
commit 9cc42a1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/sqlalchemy_continuum/plugins/flask.py b/sqlalchemy_continuum/plugins/flask.py
@@ -36,7 +36,7 @@ def fetch_current_user_id():
     if _app_ctx_stack.top is None or _request_ctx_stack.top is None:
         return
     try:
-        return current_user.id
+        return current_user.get_id()
     except AttributeError:
         return
 

diff --git a/tests/plugins/test_flask.py b/tests/plugins/test_flask.py
@@ -1,7 +1,7 @@
 import os
 
 from flask import Flask, url_for
-from flask_login import LoginManager
+from flask_login import LoginManager, UserMixin
 from flask_sqlalchemy import SQLAlchemy, _SessionSignalEvents
 from flexmock import flexmock
 
@@ -76,7 +76,7 @@ def logout(self, user=None):
     def create_models(self):
         TestCase.create_models(self)
 
-        class User(self.Model):
+        class User(self.Model, UserMixin):
             __tablename__ = 'user'
             __versioned__ = {
                 'base_classes': (self.Model, )