def user_passes_test(test_func, login_url=None, redirect_field_name=REDIRECT_FIELD_NAME):
"""
Decorator for views that checks that the user passes the given test,
redirecting to the log-in page if necessary. The test should be a callable
that takes the user object and returns True if the user passes.
"""
def decorator(view_func):
@wraps(view_func, assigned=available_attrs(view_func))
def _wrapped_view(request, *args, **kwargs):
if test_func(request.user):
return view_func(request, *args, **kwargs)
path = request.build_absolute_uri()
# If the login url is the same scheme and net location then just
# use the path as the "next" url.
login_scheme, login_netloc = urlparse.urlparse(login_url or
settings.LOGIN_URL)[:2]
current_scheme, current_netloc = urlparse.urlparse(path)[:2]
if ((not login_scheme or login_scheme == current_scheme) and
(not login_netloc or login_netloc == current_netloc)):
path = request.get_full_path()
from django.contrib.auth.views import redirect_to_login
return redirect_to_login(path, login_url, redirect_field_name)
return _wrapped_view
return decorator
我发现编辑user_passes_test使修饰函数在request上操作而不是request.user并不太困难。我在this blog post中有一个关于视图装饰器装饰器的简短版本,但是对于子孙后代,这里是我的完整编辑代码:
def request_passes_test(test_func, login_url=None, redirect_field_name=REDIRECT_FIELD_NAME):
"""
Decorator for views that checks that the request passes the given test,
redirecting to the log-in page if necessary. The test should be a callable
that takes the request object and returns True if the request passes.
"""
def decorator(view_func):
@wraps(view_func, assigned=available_attrs(view_func))
def _wrapped_view(request, *args, **kwargs):
if test_func(request):
return view_func(request, *args, **kwargs)
path = request.build_absolute_uri()
# urlparse chokes on lazy objects in Python 3, force to str
resolved_login_url = force_str(
resolve_url(login_url or settings.LOGIN_URL))
# If the login url is the same scheme and net location then just
# use the path as the "next" url.
login_scheme, login_netloc = urlparse(resolved_login_url)[:2]
current_scheme, current_netloc = urlparse(path)[:2]
if ((not login_scheme or login_scheme == current_scheme) and
(not login_netloc or login_netloc == current_netloc)):
path = request.get_full_path()
from django.contrib.auth.views import redirect_to_login
return redirect_to_login(
path, resolved_login_url, redirect_field_name)
return _wrapped_view
return decorator
注意,Django 1.9引入了^{} ,它使用方法
test_func
作为测试函数。这意味着请求在self.request
中可用。所以你可以这样做:但是,这只适用于基于类的视图。
不,不能将请求传递给
user_passes_test
。要理解它为什么以及如何工作,只需转到source:这是装饰器后面的代码。如您所见,传递给decorator的测试函数(在您的例子中,
lambda u: has_add_permission(u, "project")
)只传递了一个参数request.user
。现在,当然可以编写自己的decorator(甚至直接复制并修改代码)来传递request
本身,但是不能使用默认的user_passes_test
实现。我发现编辑
user_passes_test
使修饰函数在request
上操作而不是request.user
并不太困难。我在this blog post中有一个关于视图装饰器装饰器的简短版本,但是对于子孙后代,这里是我的完整编辑代码:我唯一做的就是把
if test_func(request.user):
改成if test_func(request):
。相关问题 更多 >
编程相关推荐