Update mailroom_db command to use Location model

nyaruka · Jan 10, 2025 · c53f19b · c53f19b
1 parent bfd6c9b
commit c53f19b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/temba/utils/management/commands/mailroom_db.py b/temba/utils/management/commands/mailroom_db.py
@@ -16,7 +16,7 @@
 from temba.contacts.models import Contact, ContactField, ContactGroup, ContactURN
 from temba.flows.models import Flow
 from temba.globals.models import Global
-from temba.locations.models import AdminBoundary
+from temba.locations.models import Location
 from temba.msgs.models import Label
 from temba.orgs.models import Org, OrgRole, User
 from temba.templates.models import Template, TemplateTranslation
@@ -152,7 +152,7 @@ def load_locations(self, path):
         # TODO figure out why this is needed
         time.sleep(1)
 
-        return AdminBoundary.objects.filter(level=0).get()
+        return Location.objects.filter(level=0).get()
 
     def reset_id_sequences(self, start: int):
         with connection.cursor() as cursor:
@@ -167,7 +167,7 @@ def create_org(self, spec, superuser, country):
             name=spec["name"],
             timezone=ZoneInfo("America/Los_Angeles"),
             flow_languages=spec["languages"],
-            country=country,
+            location=country,
             created_on=timezone.now(),
             created_by=superuser,
             modified_by=superuser,

diff --git a/test-data/nigeria.bin b/test-data/nigeria.bin