在 Django 中自动创建超级用户命令
Vaibhav Vaibhav
2021年6月29日
在 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