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}"))