# management/commands/create_plans.py
from django.core.management.base import BaseCommand
from plans.models import Plan, Quota, Pricing, PlanPricing
from decimal import Decimal
from django.db.models import Q

class Command(BaseCommand):
    help = 'Create basic subscription plans structure'

    def add_arguments(self, parser):
        parser.add_argument(
            '--reset',
            action='store_true',
            help='Delete existing plans and quotas before creating new ones'
        )

    def handle(self, *args, **options):
        if options['reset']:
            self.stdout.write(self.style.WARNING('Resetting existing plans data...'))
            Plan.objects.all().delete()
            Quota.objects.filter(
                Q(codename='basic') |
                Q(codename='pro') |
                Q(codename='premium')
            ).delete()

        # Ensure no existing plans are marked as default

        Plan.objects.all().delete()
        Quota.objects.all().delete()
        Pricing.objects.all().delete()

        # Create core quotas
        basic_quota, _ = Quota.objects.update_or_create(
            codename='basic',
            defaults={
                'name': 'Basic Features',
                'description': 'Essential platform access',
                'is_boolean': True
            }
        )

        pro_quota, _ = Quota.objects.update_or_create(
            codename='pro',
            defaults={
                'name': 'Pro Features',
                'description': 'Advanced functionality',
                'is_boolean': True
            }
        )

        premium_quota, _ = Quota.objects.update_or_create(
            codename='premium',
            defaults={
                'name': 'Premium Features',
                'description': 'Full platform access',
                'is_boolean': True
            }
        )

        # Create pricing period
        monthly_pricing, _ = Pricing.objects.update_or_create(
            name='Monthly',
            defaults={'period': 30}
        )

        # Define plan structure
        plans = [
            {
                'name': 'Basic',
                'description': 'Entry-level plan',
                'price': Decimal('0.00'),
                'period': None,
                'quotas': [basic_quota],
                'default': True
            },
            {
                'name': 'Pro',
                'description': 'Professional plan',
                'price': Decimal('29.00'),
                'period': 30,
                'quotas': [basic_quota, pro_quota],
                'default': False
            },
            {
                'name': 'Premium',
                'description': 'Full access plan',
                'price': Decimal('99.00'),
                'period': 30,
                'quotas': [basic_quota, pro_quota, premium_quota],
                'default': None
            }
        ]

        # Create plans and associations
        for plan_data in plans:
            plan, created = Plan.objects.update_or_create(
                name=plan_data['name'],
                defaults={
                    'description': plan_data['description'],
                    'default': plan_data.get('default', False),
                    'available': True,
                    'visible': True
                }
            )

            # Set quotas
            plan.quotas.set(plan_data['quotas'])

            # Create pricing if applicable
            if plan_data['price'] > 0:
                PlanPricing.objects.update_or_create(
                    plan=plan,
                    pricing=monthly_pricing,
                    defaults={
                        'price': plan_data['price'],
                        'visible': True
                    }
                )

            status = 'Created' if created else 'Updated'
            self.stdout.write(self.style.SUCCESS(f'{status} {plan.name} plan'))

        self.stdout.write(self.style.SUCCESS('Successfully created plans structure'))