From 8ecc3b9a78af453023494d26c505bc86dc84089c Mon Sep 17 00:00:00 2001 From: Matt Wright Date: Wed, 13 Mar 2013 12:09:28 -0400 Subject: [PATCH] Add user to request context for http basic and token auth --- flask_security/decorators.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/flask_security/decorators.py b/flask_security/decorators.py index 229e2c6..0ea1105 100644 --- a/flask_security/decorators.py +++ b/flask_security/decorators.py @@ -11,7 +11,7 @@ from functools import wraps -from flask import current_app, Response, request, redirect +from flask import current_app, Response, request, redirect, _request_ctx_stack from flask.ext.login import current_user, login_required from flask.ext.principal import RoleNeed, Permission, Identity, identity_changed from werkzeug.local import LocalProxy @@ -61,6 +61,7 @@ def _check_token(): if utils.md5(user.password) == data[1]: app = current_app._get_current_object() + _request_ctx_stack.top.user = user identity_changed.send(app, identity=Identity(user.id)) return True @@ -72,6 +73,7 @@ def _check_http_auth(): if user and utils.verify_and_update_password(auth.password, user): _security.datastore.commit() app = current_app._get_current_object() + _request_ctx_stack.top.user = user identity_changed.send(app, identity=Identity(user.id)) return True