From 991a96ddb733fdc459002cc01546f1a7ed396b45 Mon Sep 17 00:00:00 2001 From: virusdefender <1670873886@qq.com> Date: Tue, 26 Jan 2016 13:12:23 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=20initadmin=20=E5=91=BD?= =?UTF-8?q?=E4=BB=A4?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- utils/management/__init__.py | 1 + utils/management/commands/__init__.py | 1 + utils/management/commands/initadmin.py | 16 ++++++++++++++++ 3 files changed, 18 insertions(+) create mode 100644 utils/management/__init__.py create mode 100644 utils/management/commands/__init__.py create mode 100644 utils/management/commands/initadmin.py diff --git a/utils/management/__init__.py b/utils/management/__init__.py new file mode 100644 index 00000000..9bad5790 --- /dev/null +++ b/utils/management/__init__.py @@ -0,0 +1 @@ +# coding=utf-8 diff --git a/utils/management/commands/__init__.py b/utils/management/commands/__init__.py new file mode 100644 index 00000000..9bad5790 --- /dev/null +++ b/utils/management/commands/__init__.py @@ -0,0 +1 @@ +# coding=utf-8 diff --git a/utils/management/commands/initadmin.py b/utils/management/commands/initadmin.py new file mode 100644 index 00000000..94d2c3ae --- /dev/null +++ b/utils/management/commands/initadmin.py @@ -0,0 +1,16 @@ +# coding=utf-8 +from django.core.management.base import BaseCommand, CommandError +from account.models import User, SUPER_ADMIN, UserProfile +from utils.shortcuts import rand_str + + +class Command(BaseCommand): + def handle(self, *args, **options): + user = User.objects.create(username="root", real_name="root", email="root@oj.com", admin_type=SUPER_ADMIN) + rand_password = rand_str(length=6) + user.set_password(rand_password) + user.save() + UserProfile.objects.create(user=user) + self.stdout.write("Successfully created super admin user.\nUsername: root\nPassword: %s\n" + "Remember to change password and turn on two factors auth " + "after installation." % rand_password)