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

Author: vimalloc

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)

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")
 

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'