装饰 login_required Django 装饰器

Question

我正在尝试装饰 login_required Django 装饰器。我一直在看这个问题decorating decorators: try to get my head around understanding it。但是，这种情况有点不同，因为我不想修改 login_required 装饰器，而是向它添加更多功能。

明确地说，我正在尝试创建一个装饰器来装饰视图函数（第一个参数是正常请求）并检查用户参数。如果可能的话，我不想创建我自己的装饰器版本Django: Tweaking @login_required decorator，因为我认为这是一个安全关键的方法，将来可能会被修改。

这就是我目前所得到的，但它不起作用。

from django.conf import settings
from django.contrib.auth.decorators import login_required, REDIRECT_FIELD_NAME
from django.http.response import HttpResponseRedirect
from django.utils.decorators import available_attrs
from functools import wraps


def login_and_def_pass_required(view_function=None, redirect_field_name=REDIRECT_FIELD_NAME, login_url=None):
    def decorate_view(view_function):
        @wraps(view_function, assigned=available_attrs(view_function))
        def _double_wrapped_view(request, *args, **kwargs):
            actual_decorator = login_required(view_function, redirect_field_name, login_url)
            if request.user.temporary_password:
                return HttpResponseRedirect(settings.SET_PERMANENT_PASSWORD_URL)
            return actual_decorator
        return _double_wrapped_view
    return decorate_view

这会引发异常“'function' 对象没有属性 'get'”。

我正在使用 Django 1.8.4 Python 3.4

Answer 1

您可以将 user_passes_test 与 login_required 配合使用以实现所需的行为：

def check_permanent_password(user):
    return not user.temporary_password

@login_required(login_url)
@user_passes_test(check_temporary_password, login_url=settings.SET_PERMANENT_PASSWORD_URL)
def view(request):
    # your view

装饰 login_required Django 装饰器

decorating login_required Django decorator

python

django

django-authentication

python-decorators