mirror of
https://github.com/yyhuni/xingrin.git
synced 2026-01-31 11:46:16 +08:00
65 lines
2.0 KiB
Python
65 lines
2.0 KiB
Python
"""
|
|
初始化 admin 用户
|
|
用法: python manage.py init_admin [--password <password>]
|
|
"""
|
|
import os
|
|
from django.core.management.base import BaseCommand
|
|
from django.contrib.auth import get_user_model
|
|
|
|
User = get_user_model()
|
|
|
|
DEFAULT_USERNAME = 'admin'
|
|
DEFAULT_PASSWORD = 'admin' # 默认密码，建议首次登录后修改
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = '初始化 admin 用户'
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument(
|
|
'--password',
|
|
type=str,
|
|
default=os.getenv('ADMIN_PASSWORD', DEFAULT_PASSWORD),
|
|
help='admin 用户密码 (默认: admin 或 ADMIN_PASSWORD 环境变量)'
|
|
)
|
|
parser.add_argument(
|
|
'--force',
|
|
action='store_true',
|
|
help='强制重置密码（如果用户已存在）'
|
|
)
|
|
|
|
def handle(self, *args, **options):
|
|
password = options['password']
|
|
force = options['force']
|
|
|
|
try:
|
|
user = User.objects.get(username=DEFAULT_USERNAME)
|
|
if force:
|
|
user.set_password(password)
|
|
user.save()
|
|
self.stdout.write(
|
|
self.style.SUCCESS(f'✓ admin 用户密码已重置')
|
|
)
|
|
else:
|
|
self.stdout.write(
|
|
self.style.WARNING(f'⚠ admin 用户已存在，跳过创建（使用 --force 重置密码）')
|
|
)
|
|
except User.DoesNotExist:
|
|
User.objects.create_superuser(
|
|
username=DEFAULT_USERNAME,
|
|
email='admin@localhost',
|
|
password=password
|
|
)
|
|
self.stdout.write(
|
|
self.style.SUCCESS(f'✓ admin 用户创建成功')
|
|
)
|
|
self.stdout.write(
|
|
self.style.WARNING(f' 用户名: {DEFAULT_USERNAME}')
|
|
)
|
|
self.stdout.write(
|
|
self.style.WARNING(f' 密码: {password}')
|
|
)
|
|
self.stdout.write(
|
|
self.style.WARNING(f' ⚠ 请首次登录后修改密码!')
|
|
)
|