From 6f2f3481f4b6fba5961bf66635cc05b315a53d09 Mon Sep 17 00:00:00 2001 From: M Clark Date: Thu, 21 Jan 2016 15:44:19 +0800 Subject: [PATCH] register self._state when using init_app Brings `security.init_app(app)` inline with `security = Security(app)` by registering security._state upon security.init_app. Without this change `security.send_mail_task` will give an infinite recursion error. --- flask_security/core.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/flask_security/core.py b/flask_security/core.py index eec3092..0ebc66a 100644 --- a/flask_security/core.py +++ b/flask_security/core.py @@ -423,7 +423,7 @@ class Security(object): identity_loaded.connect_via(app)(_on_identity_loaded) - state = _get_state(app, datastore, + self._state = _get_state(app, datastore, login_form=login_form, confirm_register_form=confirm_register_form, register_form=register_form, @@ -435,11 +435,11 @@ class Security(object): anonymous_user=anonymous_user) if register_blueprint: - app.register_blueprint(create_blueprint(state, __name__)) + app.register_blueprint(create_blueprint(self._state, __name__)) app.context_processor(_context_processor) state.render_template = self.render_template - app.extensions['security'] = state + app.extensions['security'] = self._state return state