from django.core.management.base import BaseCommand
from apps.plans.models import Crop, Variety


class Command(BaseCommand):
    help = 'Initialize crops and varieties master data'

    def handle(self, *args, **options):
        crops_data = [
            {
                'name': '水稲',
                'varieties': ['コシヒカリ', 'ひとめぼれ', 'あきたこまち', 'つや姫', 'oniai']
            },
            {
                'name': '大豆',
                'varieties': ['タマホマレ', 'エンレイ', 'ミヤギром']
            },
            {
                'name': '小麦',
                'varieties': ['キタノカオリ', 'ホウライ']
            },
            {
                'name': 'そば',
                'varieties': ['信濃一号', 'はるか']
            },
            {
                'name': 'とうきび',
                'varieties': ['ゴールdent']
            },
        ]

        for crop_data in crops_data:
            crop, _ = Crop.objects.get_or_create(name=crop_data['name'])
            for variety_name in crop_data['varieties']:
                Variety.objects.get_or_create(crop=crop, name=variety_name)
            self.stdout.write(f'{crop.name}: {len(crop_data["varieties"])} varieties')

        self.stdout.write(self.style.SUCCESS('Successfully initialized crops and varieties'))