Store JWTManager in app.extensions['flask-jwt-extended'] instead of a…

…pp.jwt_manager
vimalloc · Aug 25, 2017 · 043ba23 · 043ba23
1 parent 018c2fc
commit 043ba23
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 5 deletions.
diff --git a/flask_jwt_extended/jwt_manager.py b/flask_jwt_extended/jwt_manager.py
@@ -64,7 +64,9 @@ def init_app(self, app):
         :param app: A flask application
         """
         # Save this so we can use it later in the extension
-        app.jwt_manager = self
+        if not hasattr(app, 'extensions'):   # pragma: no cover
+            app.extensions = {}
+        app.extensions['flask-jwt-extended'] = self
 
         # Set all the default configurations for this extension
         self._set_default_configuration_options(app)

diff --git a/flask_jwt_extended/utils.py b/flask_jwt_extended/utils.py
@@ -70,8 +70,8 @@ def decode_token(encoded_token):
 
 def _get_jwt_manager():
     try:
-        return current_app.jwt_manager
-    except AttributeError:  # pragma: no cover
+        return current_app.extensions['flask-jwt-extended']
+    except KeyError:  # pragma: no cover
         raise RuntimeError("You must initialize a JWTManager with this flask "
                            "application before using this method")
 

diff --git a/tests/test_jwt_manager.py b/tests/test_jwt_manager.py
@@ -24,11 +24,11 @@ def _parse_callback_result(self, result):
     def test_init_app(self):
         jwt_manager = JWTManager()
         jwt_manager.init_app(self.app)
-        self.assertIsInstance(jwt_manager, JWTManager)
+        self.assertEqual(jwt_manager, self.app.extensions['flask-jwt-extended'])
 
     def test_class_init(self):
         jwt_manager = JWTManager(self.app)
-        self.assertIsInstance(jwt_manager, JWTManager)
+        self.assertEqual(jwt_manager, self.app.extensions['flask-jwt-extended'])
 
     def test_default_user_claims_callback(self):
         identity = 'foobar'