Update mailroom_db command to use Location model

nyaruka · Jan 9, 2025 · 22194fb · 22194fb
1 parent 91ca4db
commit 22194fb
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 9 deletions.
diff --git a/temba/utils/management/commands/data/mailroom_db.json b/temba/utils/management/commands/data/mailroom_db.json
@@ -21,12 +21,6 @@
                     "first_name": "Ed",
                     "last_name": "McEditor"
                 },
-                {
-                    "email": "viewer1@textit.com",
-                    "role": "V",
-                    "first_name": "Veronica",
-                    "last_name": "Views"
-                },
                 {
                     "email": "agent1@textit.com",
                     "role": "T",

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