haikal/scripts/set_plans.py
2025-04-29 14:06:04 +03:00

63 lines
1.6 KiB
Python

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)