from plans.models import Plan, Quota
from decimal import Decimal

def run():
    # Create quotas first
    basic_quota = Quota.objects.create(
        codename='basic_quota',
        name='Basic Features',
        description='Basic plan features',
        is_boolean=True,
        url='pricing'
    )

    pro_quota = Quota.objects.create(
        codename='pro_quota',
        name='Pro Features',
        description='Pro plan features',
        is_boolean=True,
        url='pricing'
    )

    premium_quota = Quota.objects.create(
        codename='premium_quota',
        name='Premium Features',
        description='Premium plan features',
        is_boolean=True,
        url='pricing'
    )

    # Create the plans
    basic_plan = Plan.objects.create(
        name="Basic",
        description="Basic plan with limited features",
        price=Decimal("49.00"),  # 49 SAR
        period=30,  # 30 days
        default=True,
        available=True,
        visible=True,
        order=1
    )
    basic_plan.quotas.add(basic_quota)

    pro_plan = Plan.objects.create(
        name="Professional",
        description="Professional plan with advanced features",
        price=Decimal("149.00"),  # 149 SAR
        # period=30,
        available=True,
        visible=True,
        # order=2
    )
    pro_plan.quotas.add(basic_quota, pro_quota)

    premium_plan = Plan.objects.create(
        name="Premium",
        description="Premium plan with all features",
        price=Decimal("299.00"),  # 299 SAR
        period=30,
        available=True,
        visible=True,
        order=3
    )
    premium_plan.quotas.add(basic_quota, pro_quota, premium_quota)