全局设置要求登录的所有页面？

Question

我想将未经身份验证的用户的访问重定向到登录页面,之后应将登录的用户重定向到最初请求的页面.

根据文档,使用@user_passes_test装饰器很容易实现.但似乎我必须装饰每个视图,这是疯狂的,有太多而且容易出错.

有什么好方法可以全局启用此功能(除了一小组固定视图,例如login)？也就是说,默认情况下,只需登录即可显式处理匿名查看.

Answer 1

from django.shortcuts import redirect
from django.conf import settings


class LoginRequiredMiddleware:
    def __init__(self, get_response):
        self.get_response = get_response
        self.login_url = settings.LOGIN_URL
        self.open_urls = [self.login_url] + \
                         getattr(settings, 'OPEN_URLS', [])

    def __call__(self, request):
        if not request.user.is_authenticated \
        and not request.path_info in self.open_urls:
            return redirect(self.login_url+'?next='+request.path)

        return self.get_response(request)

Answer 2

我解决这个问题的方法是使用 mixin 类和装饰器（或任何你需要的代码）。虽然你必须记住调用该super(Class, self).get(...)函数，所以我想它毕竟没有那么不同。

另一方面，我发现拥有一组可以执行不同操作的 mixin 非常适合获得一个非常简单的视图，无需太多代码即可完成很多工作。

编辑

这就是我在上一个项目中所做的：

class BaseAuthMixin(object):
    def auth_check(self, user):
        return True

    def dispatch(self, request, *args, **kwargs):
        if not self.auth_check(request.user):
            from django.http import HttpResponseRedirect
            from django.contrib.auth import logout

            is_web = False
            is_live = False

            if hasattr(self, 'get_url_name'):
                from django.core.urlresolvers import reverse
                from django.core.urlresolvers import NoReverseMatch

                try:
                    next = reverse(self.get_url_name(), kwargs=kwargs)
                except NoReverseMatch:
                    next = ''
                else:
                    next= '?next=' + next

            logout(request)

            redirect_url = settings.LOGIN_URL
            redirect_url += next

            return HttpResponseRedirect(redirect_url)
        else:
            return super(BaseAuthMixin, self).dispatch(request, *args, **kwargs)

class LoginRequiredMixin(BaseAuthMixin):
    """
    Check if the view needs the user to be logged in.
    """
    def auth_check(self, user):
        if not super(LoginRequiredMixin, self).auth_check(user):
            return False
        else:
            if hasattr(self, 'login_required'):
                if self.login_required and not user.is_authenticated():
                    return False
        return True

class MyDefaultMixin(LoginRequiredMixin):
    """
    Mixin that inherits from all common view mixins.
    """
    pass

然后视图类使用上面的内容（我使用带有基于类的视图的 Django 1.3）：

from django.views.generic import TemplateView

class SomeViewClass(TemplateView, MyDefaultMixin):
    # Used by 'LoginRequiredMixin' to check if a user has to be logged in
    login_required = True

    # Template for the Django TemplateView
    template_name = "some_view_template.html"

您需要一个视图来处理登录（URL 位于 中settings.LOGIN_URL），其中包含一个带有名为 的隐藏字段的表单next。该字段必须通过上下文变量设置为成功登录后要转到的页面。

如果所有视图都继承自基本 mixin（MyDefaultMixin在我上面的代码中），它将自动检查用户是否已登录，视图包含一个名为 的属性login_required，并且该属性设置为True。

可能有更好的方法来做到这一点，但这就是我所做的，而且效果很好。