Update cities.py

Added workaround from here: coderholic#180 (comment)
sowinski · Oct 15, 2018 · 26f7833 · 26f7833
1 parent 1a60b31
commit 26f7833
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/cities/management/commands/cities.py b/cities/management/commands/cities.py
@@ -1006,6 +1006,19 @@ def import_postal_code(self):
                             region__country=pc.country)
                 except Subregion.DoesNotExist:
                     pc.subregion = None
+                except Subregion.MultipleObjectsReturned:
+                    self.logger.warn("Found multiple subregions for '{}' in '{}' - ignoring".format(
+                        pc.region_name,
+                        pc.subregion_name))
+                    self.logger.debug("item: {}\nsubregions: {}".format(
+                        item,
+                        Subregion.objects.filter(
+                            Q(region__name_std__iexact=pc.region_name) |
+                            Q(region__name__iexact=pc.region_name),
+                            Q(name_std__iexact=pc.subregion_name) |
+                            Q(name__iexact=pc.subregion_name),
+                            region__country=pc.country).values_list('id', flat=True)))
+                    pc.subregion = None
             else:
                 pc.subregion = None