Clear org country field

temba/orgs/migrations/0166_clear_org_country_field.py

@@ -0,0 +1,20 @@
+# Generated by Django 5.1.4 on 2025-01-09 17:51
+
+from django.db import migrations
+
+
+def clear_org_country_field(apps, schema_editor):
+    Org = apps.get_model("orgs", "Org")
+
+    orgs = Org.objects.all().exclude(country=None)
+    if orgs:
+        orgs.update(country=None)
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("orgs", "0165_org_location"),
+    ]
+
+    operations = [migrations.RunPython(clear_org_country_field, migrations.RunPython.noop)]

temba/orgs/models.py

@@ -506,7 +506,6 @@ class Org(SmartModel):
         default=DATE_FORMAT_DAY_FIRST,
         help_text=_("Default formatting and parsing of dates in flows and messages."),
     )
-    country = models.ForeignKey("locations.AdminBoundary", null=True, on_delete=models.PROTECT)
     location = models.ForeignKey("locations.Location", null=True, on_delete=models.PROTECT)
     flow_languages = ArrayField(models.CharField(max_length=3), default=list, validators=[ArrayMinLengthValidator(1)])
     input_collation = models.CharField(max_length=32, choices=COLLATION_CHOICES, default=COLLATION_DEFAULT)
@@ -537,6 +536,9 @@ class Org(SmartModel):
     released_on = models.DateTimeField(null=True)
     deleted_on = models.DateTimeField(null=True)
 
+    # Deprecated
+    country = models.ForeignKey("locations.AdminBoundary", null=True, on_delete=models.PROTECT)
+
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)