如何限制django中超级用户的数量

Question

在我的 django 项目中，我希望只有一个超级用户，并且不能通过 python manage.py createsuperuser

创建更多超级用户

有可能吗？如果可能怎么做？

Answer 1

任何能够运行python manage.py createsuperuser 的人都应该能够运行python manage.py dbshell 并在数据库中手动创建超级用户。所以，无论如何，这应该是一个值得信赖的人。

如果只有受信任的人可以添加超级用户，那么只需告诉他们不要创建多个超级用户（尽管我想知道仅限于一个超级用户的目的是什么）。

不过，如果你想防止python manage.py createsuperuser错误地创建多个超级用户，你可以override this command：

from django.contrib.auth.management.commands import createsuperuser
from django.core.management.base import CommandError

class Command(createsuperuser.Command):
    def handle(self, *args, **options):
        if self.UserModel.objects.filter(is_superuser=True).exists():
            raise CommandError("There is no room for two, go your way!")
        super().handle(*args, **options)

请注意，这不会阻止从 django 管理界面将用户设置为超级用户。

如果你想完全做到不能创建两个超级用户，可以直接在数据库级别添加约束。

另一种方法是继承django.contrib.auth.models.User 并定义：

SUPERUSER_ID = 1  # or whatever

@property
def is_superuser(self):
    return self.id == self.SUPERUSER_ID

Answer 2

您可以编写一个脚本来检查超级用户的数量。假设您想要 10 个超级用户，那么每次创建超级用户时都会计算其是否超过 10 个，并相应地给出错误/成功消息。

您可以按如下方式计算超级用户数：

    from django.contrib.auth.models import User
    from django.http import HttpResponse

    user_obj = User.objects.all()
    c = 0
    for i in user_obj:
        if i.is_superuser():
            c += 1

    if c > 10:
        return HttpResponse('Cannot add anymore superusers')
    else:
        new_user  = User.objects.create_user(username = name, password = password)

当然，您必须制作一个表格来接受用户名和密码，但我已经给出了基本的想法。
您还可以使用 python 的 threading 库来使事情异步