from django.core.management.base import BaseCommand
from api.management.commands.populate_django_models import main

class Command(BaseCommand):
    help = '填充Django模型数据'

    def handle(self, *args, **options):
        self.stdout.write('开始填充数据...')
        main()
        self.stdout.write(
            self.style.SUCCESS('数据填充完成!')
        )