我想通过Bash脚本自动创建Django用户。我发现这个片段几乎符合我的需要:
echo "from django.contrib.auth.models import User; User.objects.create_superuser('admin', 'admin@example.com', 'pass')" |\
python manage.py shell
如果用户已经存在,我如何修改它以使它成为一个nop?
答案 0 :(得分:7)
manage.py shell
您可以使用QuerySet API方法检查用户是否存在,如果不存在则创建用户。此外,将代码放在heredoc中可能更容易:
cat <<EOF | python manage.py shell
from django.contrib.auth import get_user_model
User = get_user_model() # get the currently active user model,
User.objects.filter(username='admin').exists() or \
User.objects.create_superuser('admin', 'admin@example.com', 'pass')
EOF
另一个更易于维护的选项是为您的Django应用添加自定义management command。根据文档调整示例,修改yourapp/management/commands/ensure_adminuser.py
,如下所示:
from django.core.management.base import BaseCommand
from django.contrib.auth import get_user_model
class Command(BaseCommand):
help = "Creates an admin user non-interactively if it doesn't exist"
def add_arguments(self, parser):
parser.add_argument('--username', help="Admin's username")
parser.add_argument('--email', help="Admin's email")
parser.add_argument('--password', help="Admin's password")
def handle(self, *args, **options):
User = get_user_model()
if not User.objects.filter(username=options['username']).exists():
User.objects.create_superuser(username=options['username'],
email=options['email'],
password=options['password'])
然后可以使用python manage.py ensure_adminuser --username=admin --email=admin@example.com --password=pass
从您的Bash脚本调用新的自定义命令。
答案 1 :(得分:1)
即使您的应用中有自定义用户模型,此bash功能也可以解决问题:
create-superuser () {
local username="$1"
local email="$2"
local password="$3"
cat <<EOF | python manage.py shell
from django.contrib.auth import get_user_model
User = get_user_model()
if not User.objects.filter(username="$username").exists():
User.objects.create_superuser("$username", "$email", "$password")
else:
print('User "{}" exists already, not created'.format("$username"))
EOF
}
向Eugene Yarmash致信最初的想法。
答案 2 :(得分:0)
您可以使用get_or_create()。如果存在则不会执行任何操作,否则会创建一个。
您必须手动将is_staff
和is_superuser
设置为True