Fix initial site creation, to check for SITE_ID

Fix suggested in #90 by @mrmachine
django-fluent · Oct 7, 2015 · 9f18106 · 9f18106
1 parent de2dbc6
commit 9f18106
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/fluent_pages/migrations/0001_initial.py b/fluent_pages/migrations/0001_initial.py
@@ -10,12 +10,12 @@
 
 def make_site(apps, schema_editor):
     Site = apps.get_model("sites", "Site")
-    if Site.objects.count() == 0:
-        site = Site()
-        site.pk = settings.SITE_ID
-        site.name = 'example'
-        site.domain = 'example.com'
-        site.save()
+    Site.objects.get_or_create(
+        pk=settings.SITE_ID,
+        defaults=dict(
+            name='example',
+            domain='example.com',
+        ))
 
 
 def remove_site(apps, schema_editor):