diff --git a/flask_security/core.py b/flask_security/core.py index d0a632d..9b776ae 100644 --- a/flask_security/core.py +++ b/flask_security/core.py @@ -232,14 +232,11 @@ class Security(object): url_prefix=cv('URL_PREFIX', app=app)) app.register_blueprint(bp) - if not hasattr(app, 'extensions'): - app.extensions = {} - kwargs = {} for key, value in get_config(app).items(): kwargs[key.lower()] = value - app.extensions['security'] = _SecurityState( + state = _SecurityState( app=app, datastore=datastore, auth_provider=AuthenticationProvider(), @@ -251,6 +248,11 @@ class Security(object): token_auth_serializer=_get_token_auth_serializer(app), **kwargs) + if not hasattr(app, 'extensions'): + app.extensions = {} + + app.extensions['security'] = state + class AuthenticationProvider(object): """The default authentication provider implementation."""