在 Django 中自动创建超级用户命令

Vaibhav Vaibhav 2021年6月29日
在 Django 中自动创建超级用户命令

在 Django 中,拥有管理员权限的用户称为超级用户。超级用户可以做任何事情。他们可以创建新用户、删除现有用户、操纵其他用户的信息等。

在 Django 中,要创建超级用户,我们通常使用命令行命令 python manage.py createsuperuser,然后再输入一些内容:用户名、电子邮件和密码。

但是,如果我们可以自动化整个过程或使用一行代码完成它,那不是很神奇吗?

在本文中,我们将学习自动创建超级用户。

Django Shell 是一个命令行 Python 工具,可使用 Django 的数据库 API 与数据库进行交互。使用 Django Shell,我们可以在数据库中创建数据,修改现有数据,甚至删除现有数据。此外,我们还可以创建新用户和超级用户。

要使用 Django Shell 创建超级用户,请使用以下脚本。

echo "from django.contrib.auth import get_user_model; User = get_user_model(); User.objects.create_superuser('userUsername', 'userEmail', 'userPassword')" | python manage.py shell

此脚本首先导入一个方法 get_user_model,以获取 Django 使用的用户模型。

此方法检索对实际用户模型对象的引用并将其存储在变量 User 中。

使用这个引用,它然后使用 create_superuser() 创建一个新的超级用户。此方法接受用户名、电子邮件和密码。

请注意,这些字段对 Django 的默认用户模型有效。如果你使用自定义用户模型,则用于创建用户或超级用户的字段将相应更改。

使用 Python 脚本创建超级用户

由于 Django 模型在 Django 项目中随处可见,我们可以使用它们来创建用户和超级用户。我们可以创建一个接受一些输入的函数,然后创建一个使用它们来完成这个任务的超级用户。

from django.contrib.auth.models import User


def createSuperUser(username, password, email="", firstName="", lastName=""):
    invalidInputs = ["", None]

    if username.strip() in invalidInputs or password.strip() in invalidInputs:
        return None

    user = User(
        username=username,
        email=email,
        first_name=firstName,
        last_name=lastName,
    )
    user.set_password(password)
    user.is_superuser = True
    user.is_staff = True
    user.save()

    return user

该函数有两个强制参数,即用户名和密码,以及三个可选参数。它根据传递的值创建一个超级用户,并返回对新形成的超级用户的引用。

作者: Vaibhav Vaibhav
Vaibhav Vaibhav avatar Vaibhav Vaibhav avatar

Vaibhav is an artificial intelligence and cloud computing stan. He likes to build end-to-end full-stack web and mobile applications. Besides computer science and technology, he loves playing cricket and badminton, going on bike rides, and doodling.