我正在尝试创建一个类似于createuperuser的命令,它将使用两个参数(用户名和密码)
它在django 1.7中工作正常,但不在1.8. (我也在使用python3.4)
这是我写的代码
MyApp的/管理/命令/ createmysuperuser.py
from django.core.management.base import BaseCommand, CommandError
from django.contrib.auth.models import User
class Command(BaseCommand):
help = 'Create a super user'
def handle(self, *args, **options):
if len(args) != 2:
raise CommandError('need exactly two arguments for username and password')
username, password = args
u, created = User.objects.get_or_create(username=username)
if created:
u.is_superuser = True
u.is_staff = True
u.set_password(password)
u.save()
else:
raise CommandError("user '%s' already exist" % username)
return "Password changed successfully for user '%s'" % u.username
当我尝试运行这个命令
$python manage.py createmysuperuser myuser mypassword
我得到这个错误
usage: manage.py createmysuperuser [-h] [--version] [-v {0,1,2,3}]
[--settings SETTINGS]
[--pythonpath PYTHONPATH] [--traceback]
[--no-color]
manage.py createmysuperuser: error: unrecognized arguments: myuser mypassword
但是当我不传递任何参数时,会引发预期的CommandError.
CommandError:用户名和密码只需要两个参数