Как передать объект запроса Django в функцию вызова user_passes_test
Я использую декоратор Django user_passes_test, чтобы проверить разрешение пользователя.
@user_passes_test(lambda u: has_add_permission(u, "project"))
def create_project(request):
......
Я вызываю функцию обратного вызова has_add_permission, которая принимает два аргумента User и String. Я хотел бы передать объект запроса вместе с тем, что это возможно? Кроме того, кто-нибудь может рассказать мне, как мы можем напрямую обращаться к объекту User внутри декоратора.
Ответы
Ответ 1
Нет, вы не можете передать запрос user_passes_test. Чтобы понять, почему и как это работает, просто перейдите к источнику :
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. Как вы можете видеть, тестовая функция, переданная декоратору (в вашем случае lambda u: has_add_permission(u, "project")), передается только один аргумент, request.user. Теперь, конечно, можно написать свой собственный декоратор (даже скопировать этот код напрямую и просто его изменить), чтобы передать сам request, но вы не можете выполнить его с реализацией по умолчанию user_passes_test.
Ответ 2
Обратите внимание, что Django 1.9 представил UserPassesTestMixin, который использует метод test_func в качестве тестовой функции. Это означает, что запрос доступен в self.request. Итак, вы можете сделать что-то вроде этого:
class MyView(UserPassesTestMixin, View):
def test_func(self):
return has_add_permission(self.request.user, self.request)
Это работает только с представлениями на основе классов.
Ответ 3
Я нашел редактирование user_passes_test, чтобы украшенная функция работала на request, а не request.user не слишком сложна. У меня есть короткая версия в в этом сообщении в блоге о декораторе-декораторе, но для потомков, здесь мой полный отредактированный код:
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
В основном, единственное, что я сделал, это изменить if test_func(request.user): на if test_func(request):.
