Add swappable user model to migration

orcasgit · Oct 2, 2018 · 0d4cc4d · 0d4cc4d
1 parent 268fc15
commit 0d4cc4d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/fitapp/migrations/0001_initial.py b/fitapp/migrations/0001_initial.py
@@ -6,6 +6,9 @@
 
 
 class Migration(migrations.Migration):
+
+    UserModel = getattr(settings, 'FITAPP_USER_MODEL', 'auth.User')
+
     dependencies = [
         migrations.swappable_dependency(settings.AUTH_USER_MODEL),
     ]