From cacdde3f872ed008e5fd4eaeb02997fbf7f0e01a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Martin=20P=C3=A9pin?= Date: Sun, 14 May 2017 15:47:07 +0100 Subject: [PATCH] Remove duplicate locations --- gestion/migrations/0003_events.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/gestion/migrations/0003_events.py b/gestion/migrations/0003_events.py index 13d37097..40faca91 100644 --- a/gestion/migrations/0003_events.py +++ b/gestion/migrations/0003_events.py @@ -22,12 +22,12 @@ def import_events(apps, schema_editor): # in this table for each value of the old `location` field. OldEvent, NewEvent = models[0] Location = apps.get_model("gestion", "Location") - locations = [] + locations = set() # A set to prevent duplicate entries events = [] for event in OldEvent.objects.values(): - locations.append(Location(name=event["location"])) + locations.add(event["location"]) events.append(event) - Location.objects.bulk_create(locations) + Location.objects.bulk_create([Location(name=name) for name in locations]) map_loc = { loc.name: loc for loc in Location.objects.all()