Fix migration 0023 subscriber model reference

radaram · Mar 29, 2017 · caaa59c · caaa59c
1 parent c19af62
commit caaa59c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/djstripe/migrations/0023_auto_20170307_0937.py b/djstripe/migrations/0023_auto_20170307_0937.py
@@ -6,6 +6,9 @@
 from django.db import migrations, models
 
 
+DJSTRIPE_SUBSCRIBER_MODEL = getattr(settings, "DJSTRIPE_SUBSCRIBER_MODEL", settings.AUTH_USER_MODEL)
+
+
 class Migration(migrations.Migration):
 
     dependencies = [
@@ -17,7 +20,7 @@ class Migration(migrations.Migration):
         migrations.AlterField(
             model_name='customer',
             name='subscriber',
-            field=models.ForeignKey(null=True, on_delete=models.SET_NULL, related_name='djstripe_customers', to=settings.AUTH_USER_MODEL),
+            field=models.ForeignKey(null=True, on_delete=models.SET_NULL, related_name='djstripe_customers', to=DJSTRIPE_SUBSCRIBER_MODEL),
         ),
         migrations.AlterUniqueTogether(
             name='customer',