Fixing migration script as per swapper docs

https://github.com/wq/django-swappable-models
ninjabit · Jan 28, 2019 · 14fadaa · 14fadaa
1 parent a78b546
commit 14fadaa
Showing 1 changed file with 10 additions and 8 deletions.
diff --git a/flexible_plans/migrations/0001_initial.py b/flexible_plans/migrations/0001_initial.py
@@ -1,10 +1,9 @@
 # Generated by Django 2.1.5 on 2019-01-28 14:58
 
-from django.conf import settings
 from django.db import migrations, models
-import django.db.models.deletion
 import django.utils.timezone
 import model_utils.fields
+import swapper
 
 
 class Migration(migrations.Migration):
@@ -24,7 +23,8 @@ class Migration(migrations.Migration):
                 ('codename', models.CharField(db_index=True, max_length=50, unique=True, verbose_name='codename')),
             ],
             options={
-                'swappable': 'FLEXIBLE_PLANS_FEATURE_MODEL',
+                # 'swappable': 'FLEXIBLE_PLANS_FEATURE_MODEL',
+                'swappable': swapper.swappable_setting('flexible_plans', 'Feature'),
             },
         ),
         migrations.CreateModel(
@@ -41,13 +41,14 @@ class Migration(migrations.Migration):
                 ('provider', models.CharField(max_length=100)),
             ],
             options={
-                'swappable': 'FLEXIBLE_PLANS_PLAN_MODEL',
+                # 'swappable': 'FLEXIBLE_PLANS_PLAN_MODEL',
+                'swappable': swapper.swappable_setting('flexible_plans', 'Plan'),
             },
         ),
         migrations.CreateModel(
             name='CumulativeFeature',
             fields=[
-                ('feature_ptr', models.OneToOneField(auto_created=True, on_delete=django.db.models.deletion.CASCADE, parent_link=True, primary_key=True, serialize=False, to=settings.FLEXIBLE_PLANS_FEATURE_MODEL)),
+                ('feature_ptr', models.OneToOneField(auto_created=True, on_delete=django.db.models.deletion.CASCADE, parent_link=True, primary_key=True, serialize=False, to=swapper.get_model_name('flexible_plans', 'Feature'))),
                 ('usage', models.PositiveIntegerField(default=0)),
             ],
             options={
@@ -58,7 +59,7 @@ class Migration(migrations.Migration):
         migrations.CreateModel(
             name='MeteredFeature',
             fields=[
-                ('feature_ptr', models.OneToOneField(auto_created=True, on_delete=django.db.models.deletion.CASCADE, parent_link=True, primary_key=True, serialize=False, to=settings.FLEXIBLE_PLANS_FEATURE_MODEL)),
+                ('feature_ptr', models.OneToOneField(auto_created=True, on_delete=django.db.models.deletion.CASCADE, parent_link=True, primary_key=True, serialize=False, to=swapper.get_model_name('flexible_plans', 'Feature'))),
                 ('units', models.PositiveIntegerField(default=0)),
                 ('usage', models.PositiveIntegerField(default=0)),
             ],
@@ -70,7 +71,7 @@ class Migration(migrations.Migration):
         migrations.CreateModel(
             name='PermissionFeature',
             fields=[
-                ('feature_ptr', models.OneToOneField(auto_created=True, on_delete=django.db.models.deletion.CASCADE, parent_link=True, primary_key=True, serialize=False, to=settings.FLEXIBLE_PLANS_FEATURE_MODEL)),
+                ('feature_ptr', models.OneToOneField(auto_created=True, on_delete=django.db.models.deletion.CASCADE, parent_link=True, primary_key=True, serialize=False, to=swapper.get_model_name('flexible_plans', 'Feature'))),
                 ('permission', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='+', to='auth.Permission')),
             ],
             options={
@@ -81,6 +82,7 @@ class Migration(migrations.Migration):
         migrations.AddField(
             model_name='plan',
             name='features',
-            field=models.ManyToManyField(to=settings.FLEXIBLE_PLANS_FEATURE_MODEL),
+            # field=models.ManyToManyField(to=settings.FLEXIBLE_PLANS_FEATURE_MODEL),
+            field=models.ManyToManyField(to=swapper.get_model_name('flexible_plans', 'Feature')),
         ),
     ]