mirror of https://github.com/openspug/spug
28 lines
1.3 KiB
Python
28 lines
1.3 KiB
Python
# Copyright: (c) OpenSpug Organization. https://github.com/openspug/spug
|
||
# Copyright: (c) <spug.dev@gmail.com>
|
||
# Released under the AGPL-3.0 License.
|
||
from django.core.management.base import BaseCommand
|
||
from apps.account.models import User
|
||
|
||
|
||
class Command(BaseCommand):
|
||
help = '创建账户'
|
||
|
||
def add_arguments(self, parser):
|
||
parser.add_argument('-u', required=True, metavar='username', help='账户名称')
|
||
parser.add_argument('-p', required=True, metavar='password', help='账户密码')
|
||
parser.add_argument('-n', default='', metavar='nickname', help='账户昵称')
|
||
parser.add_argument('-s', default=False, action='store_true', help='是否是超级用户（默认否）')
|
||
|
||
def handle(self, *args, **options):
|
||
if User.objects.filter(username=options['u'], deleted_by_id__isnull=True).exists():
|
||
return self.stderr.write(self.style.ERROR(f'已存在登录名为【{options["u"]}】的用户'))
|
||
User.objects.create(
|
||
username=options['u'],
|
||
nickname=options['n'],
|
||
password_hash=User.make_password(options['p']),
|
||
is_supper=options['s'],
|
||
)
|
||
self.stdout.write(self.style.SUCCESS('创建成功'))
|
||
self.stdout.write(self.style.WARNING('废弃警告，v3.0.0之后将会移除该命令，请使用 python manage.py user add 来代替！'))
|