Django将字段添加到createsuperuser命令

# myapp/models.py
from django.db import models
from django.contrib.auth.models import User

class UserProfile(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE)
    name = models.CharField(max_length=100)
    email = models.EmailField()

# myapp/admin.py
from django.contrib import admin
from .models import UserProfile

admin.site.register(UserProfile)

# myapp/management/commands/createsuperuser.py
from django.contrib.auth.management.commands.createsuperuser import Command as BaseCommand
from myapp.models import UserProfile

class Command(BaseCommand):
    help = 'Create a superuser with additional fields'

    def handle(self, *args, **options):
        username = options.get('username')
        email = options.get('email')
        name = options.get('name')

        user_data = {
            'username': username,
            'email': email,
            'password': options.get('password'),
        }

        user, created = User.objects.get_or_create(**user_data)

        if created:
            user_profile = UserProfile.objects.create(user=user, name=name)
            self.stdout.write(self.style.SUCCESS('Superuser created successfully.'))
        else:
            self.stdout.write(self.style.WARNING('Superuser already exists.'))