add ABSTRACT_ONLY setting

Author: st4lk

affiliate/app_settings.py

@@ -14,6 +14,7 @@
 SAVE_IN_SESSION = getattr(settings, 'AFFILIATE_SAVE_IN_SESSION', True)
 SESSION_AGE = getattr(settings, 'AFFILIATE_SESSION_AGE', 5 * 24 * 60 * 60)
 DEFAULT_LINK = getattr(settings, 'AFFILIATE_DEFAULT_LINK', '/')
+ABSTRACT_ONLY = getattr(settings, 'AFFILIATE_ABSTRACT_ONLY', False)
 # deprecated
 BANNER_FOLDER = getattr(settings, 'AFFILIATE_BANNER_PATH', 'affiliate')
 START_AID = getattr(settings, 'AFFILIATE_START_AID', "1000")

affiliate/migrations/0001_initial.py

@@ -3,28 +3,42 @@
 
 from django.db import models, migrations
 from django.conf import settings
+from affiliate import app_settings
 
 
-class Migration(migrations.Migration):
-
-    dependencies = [
-        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
-    ]
-
-    operations = [
-        migrations.CreateModel(
-            name='Affiliate',
-            fields=[
-                ('id', models.AutoField(verbose_name='ID', serialize=False, auto_created=True, primary_key=True)),
-                ('reward_amount', models.DecimalField(default=10, verbose_name='reward amount', max_digits=5, decimal_places=2)),
-                ('reward_percentage', models.BooleanField(default=True, verbose_name='in percent')),
-                ('is_active', models.BooleanField(default=True, verbose_name='active')),
-                ('created_at', models.DateTimeField(auto_now_add=True, verbose_name='created at')),
-                ('user', models.OneToOneField(verbose_name='User', to=settings.AUTH_USER_MODEL)),
-            ],
-            options={
-                'swappable': 'AFFILIATE_AFFILIATE_MODEL',
-            },
-            bases=(models.Model,),
-        ),
-    ]
+if app_settings.ABSTRACT_ONLY:
+
+    class Migration(migrations.Migration):
+        """
+        Empty migration
+        """
+
+        dependencies = []
+
+        operations = []
+
+else:
+
+    class Migration(migrations.Migration):
+
+        dependencies = [
+            migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+        ]
+
+        operations = [
+            migrations.CreateModel(
+                name='Affiliate',
+                fields=[
+                    ('id', models.AutoField(verbose_name='ID', serialize=False, auto_created=True, primary_key=True)),
+                    ('reward_amount', models.DecimalField(default=10, verbose_name='reward amount', max_digits=5, decimal_places=2)),
+                    ('reward_percentage', models.BooleanField(default=True, verbose_name='in percent')),
+                    ('is_active', models.BooleanField(default=True, verbose_name='active')),
+                    ('created_at', models.DateTimeField(auto_now_add=True, verbose_name='created at')),
+                    ('user', models.OneToOneField(verbose_name='User', to=settings.AUTH_USER_MODEL)),
+                ],
+                options={
+                    'swappable': 'AFFILIATE_AFFILIATE_MODEL',
+                },
+                bases=(models.Model,),
+            ),
+        ]

affiliate/models.py

@@ -67,8 +67,9 @@ class NoAffiliate(object):
     def exists(self):
         return False
 
+if not app_settings.ABSTRACT_ONLY:
 
-class Affiliate(AbstractAffiliate):
+    class Affiliate(AbstractAffiliate):
 
-    class Meta:
-        swappable = 'AFFILIATE_AFFILIATE_MODEL'
+        class Meta:
+            swappable = 'AFFILIATE_AFFILIATE_MODEL'