add white label to create methods

Gary-Community-Ventures · Dec 10, 2024 · f71e4cc · f71e4cc
1 parent 5a3b2c8
commit f71e4cc
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/programs/models.py b/programs/models.py
@@ -779,11 +779,11 @@ def from_model_data(self, data: DataType):
         for county in data["counties"]:
             try:
                 county_instance = County.objects.get(name=county["name"])
+                county_instance.white_label = white_label
+                county_instance.save()
             except County.DoesNotExist:
-                county_instance = County.objects.create(name=county["name"])
+                county_instance = County.objects.create(name=county["name"], white_label=white_label)
 
-            county_instance.white_label = white_label
-            county_instance.save()
             counties.append(county_instance)
         navigator.counties.set(counties)
 
@@ -909,8 +909,10 @@ def from_model_data(self, data: DataType):
         for county in data["counties"]:
             try:
                 county_instance = County.objects.get(name=county["name"])
+                county_instance.white_label = white_label
+                county_instance.save()
             except County.DoesNotExist:
-                county_instance = County.objects.create(name=county["name"])
+                county_instance = County.objects.create(name=county["name"], white_label=white_label)
             counties.append(county_instance)
         warning.counties.set(counties)
 
@@ -1055,8 +1057,10 @@ def from_model_data(self, data: DataType):
         for county in data["counties"]:
             try:
                 county_instance = County.objects.get(name=county["name"])
+                county_instance.white_label = white_label
+                county_instance.save()
             except County.DoesNotExist:
-                county_instance = County.objects.create(name=county["name"])
+                county_instance = County.objects.create(name=county["name"], white_label=white_label)
             counties.append(county_instance)
         translation_override.counties.set(counties)