Django 只允许超级用户登录

在本文中，我们将介绍如何在Django中只允许超级用户登录的方法。通常情况下，Django允许任何已注册的用户登录，但有时我们需要限制只有超级用户才能访问特定的功能或页面。通过以下步骤，我们可以实现这个需求。

创建一个中间件

第一步是创建一个中间件，并在其中添加逻辑以检查用户是否是超级用户。在Django中，中间件能够在请求到达视图之前或之后执行一些操作。

首先，在Django项目的根目录中的任何地方创建一个名为middlewares.py的文件，然后添加以下代码：

from django.http import HttpResponseForbidden

class SuperUserMiddleware:
    def __init__(self, get_response):
        self.get_response = get_response

    def __call__(self, request):
        if not request.user.is_superuser:
            return HttpResponseForbidden()

        response = self.get_response(request)
        return response

上述代码中，SuperUserMiddleware类是一个中间件类，它检查request.user是否为超级用户。如果不是超级用户，则返回HttpResponseForbidden，即禁止访问。

注册中间件

接下来，我们需要将自定义的中间件注册到Django项目中。在项目的settings.py文件中，找到MIDDLEWARE设置项，并在其中添加'yourproject.middlewares.SuperUserMiddleware'。

MIDDLEWARE = [
    ...
    'yourproject.middlewares.SuperUserMiddleware',
    ...
]

确保将yourproject替换为你的项目的文件夹名。

测试登录限制

一旦我们完成了上述步骤，就可以测试限制只有超级用户才能登录的功能了。

首先，确保你有一个超级用户。在终端中，运行以下命令来创建一个超级用户：

python manage.py createsuperuser

按照提示输入用户名、电子邮件地址和密码。

接下来，启动Django开发服务器：

python manage.py runserver

打开浏览器，并输入http://localhost:8000/admin/来访问Django的管理页面。使用刚刚创建的超级用户进行登录。

现在，我们尝试使用一个非超级用户进行登录，你将会看到收到了403 Forbidden的错误页面。

自定义错误消息

为了改进用户体验，我们可以自定义并显示一个错误消息，以说明为什么访问被拒绝。

在SuperUserMiddleware类中，我们可以添加自定义错误消息的逻辑。修改中间件的__call__方法如下：

from django.http import HttpResponseForbidden
from django.contrib import messages

class SuperUserMiddleware:
    def __init__(self, get_response):
        self.get_response = get_response

    def __call__(self, request):
        if not request.user.is_superuser:
            messages.error(request, 'Only super users are allowed to access this page.')
            return HttpResponseForbidden()

        response = self.get_response(request)
        return response

在上述代码中，我们导入了messages模块，并在不是超级用户的情况下添加了一个错误消息。

要使这个功能生效，需要保证在settings.py文件中设置了django.contrib.messages.middleware.MessageMiddleware：

MIDDLEWARE = [
    ...
    'django.contrib.sessions.middleware.SessionMiddleware',
    'django.contrib.messages.middleware.MessageMiddleware',
    'yourproject.middlewares.SuperUserMiddleware',
    ...
]