17 lines
704 B
Python
17 lines
704 B
Python
import os
|
|
from django.contrib.auth import get_user_model
|
|
from django.core.management.base import BaseCommand
|
|
class Command(BaseCommand):
|
|
def handle(self, *args, **kwargs):
|
|
User = get_user_model()
|
|
username = os.getenv("ADMIN_USERNAME", "admin")
|
|
user, _ = User.objects.get_or_create(username=username)
|
|
user.first_name = os.getenv("ADMIN_FIRST_NAME", "Admin")
|
|
user.last_name = os.getenv("ADMIN_LAST_NAME", "")
|
|
user.is_staff = True
|
|
user.is_superuser = True
|
|
user.is_active = True
|
|
user.set_password(os.getenv("ADMIN_PASSWORD", "admin123"))
|
|
user.save()
|
|
self.stdout.write(self.style.SUCCESS(f"Admin ready: {username}"))
|