27 lines
888 B
Python
27 lines
888 B
Python
|
from django.db import migrations
|
||
|
|
|
||
|
|
|
||
|
|
def backfill_farm_subscription_plans(apps, schema_editor):
|
||
|
|
SubscriptionPlan = apps.get_model("access_control", "SubscriptionPlan")
|
||
|
|
FarmHub = apps.get_model("farm_hub", "FarmHub")
|
||
|
|
|
||
|
|
default_plan = (
|
||
|
|
SubscriptionPlan.objects.filter(is_active=True, metadata__is_default=True).order_by("name").first()
|
||
|
|
or SubscriptionPlan.objects.filter(code="gold", is_active=True).first()
|
||
|
|
)
|
||
|
|
if default_plan is None:
|
||
|
|
return
|
||
|
|
|
||
|
|
FarmHub.objects.filter(subscription_plan__isnull=True).update(subscription_plan=default_plan)
|
||
|
|
|
||
|
|
|
||
|
|
class Migration(migrations.Migration):
|
||
|
|
dependencies = [
|
||
|
|
("access_control", "0004_enable_default_feature_access"),
|
||
|
|
("farm_hub", "0007_farmhub_subscription_plan"),
|
||
|
|
]
|
||
|
|
|
||
|
|
operations = [
|
||
|
|
migrations.RunPython(backfill_farm_subscription_plans, migrations.RunPython.noop),
|
||
|
|
]
|