feature/55 first approach to campaigns

campaigns/admin.py

@@ -0,0 +1,16 @@
+from django.contrib import admin
+from campaigns.models import Campaign
+from drip.models import Drip
+
+
+class DripInline(admin.TabularInline):
+    model = Drip
+
+
+class CampaignAdmin(admin.ModelAdmin):
+    inlines = [
+        DripInline,
+    ]
+
+
+admin.site.register(Campaign, CampaignAdmin)

campaigns/apps.py

@@ -0,0 +1,5 @@
+from django.apps import AppConfig
+
+
+class CampaignsConfig(AppConfig):
+    name = 'campaigns'

campaigns/migrations/0001_initial.py

@@ -0,0 +1,22 @@
+# Generated by Django 3.0.7 on 2020-12-30 21:11
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='Campaign',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('name', models.CharField(default='Unnamed Campaign', max_length=256)),
+                ('delete_drips', models.BooleanField(default=True)),
+            ],
+        ),
+    ]

campaigns/models.py

@@ -0,0 +1,11 @@
+from django.db import models
+
+
+class Campaign(models.Model):
+    name = models.CharField(max_length=256, default='Unnamed Campaign')
+    delete_drips = models.BooleanField(default=True)
+
+    def delete(self, using=None, keep_parents=False):
+        if self.delete_drips:
+            self.drip_set.all().delete()
+        super().delete(using, keep_parents)

campaigns/tests.py

@@ -0,0 +1,65 @@
+from django.test import TestCase
+from drip.models import Drip
+from campaigns.models import Campaign
+
+DRIP_AMOUNT = 10
+
+
+def _drips_generator(amount, **drip_extra_args):
+    for i in range(amount):
+        yield Drip(
+            name='{}th drip'.format(i),
+            **drip_extra_args
+        )
+
+
+class DripsTestCase(TestCase):
+
+    def test_campaings_creation(self):
+        campaign = Campaign()
+        campaign.save()
+
+        Drip.objects.bulk_create(
+            _drips_generator(
+                DRIP_AMOUNT,
+                campaign=campaign
+            )
+        )
+
+        assert len(campaign.drip_set.all()) == DRIP_AMOUNT
+
+    def test_remove_campaings(self):
+        campaign_delete_drips = Campaign(name='remove me!', delete_drips=True)
+        campaign_delete_drips.save()
+
+        Drip.objects.bulk_create(
+            _drips_generator(
+                DRIP_AMOUNT,
+                campaign=campaign_delete_drips
+            )
+        )
+
+        drips_before_delete = Drip.objects.count()
+
+        campaign_delete_drips.delete()
+
+        assert drips_before_delete > Drip.objects.count()
+
+        campaign_do_not_delete_drips = Campaign(
+            name='remove me, but keep my drips alive!',
+            delete_drips=False,
+        )
+        campaign_do_not_delete_drips.save()
+
+        Drip.objects.bulk_create(
+            _drips_generator(
+                DRIP_AMOUNT,
+                campaign=campaign_do_not_delete_drips
+            )
+        )
+
+        drips_before_delete = Drip.objects.count()
+
+        campaign_do_not_delete_drips.delete()
+
+        assert drips_before_delete == Drip.objects.count()

drip/__init__.py

@@ -1 +1 @@
-__version__ = '1.9.3'
+__version__ = '1.10.3'

drip/migrations/0003_drip_campaign.py

@@ -0,0 +1,20 @@
+# Generated by Django 3.0.7 on 2020-12-30 22:10
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('campaigns', '0001_initial'),
+        ('drip', '0002_querysetrule_rule_type'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='drip',
+            name='campaign',
+            field=models.ForeignKey(blank=True, help_text='If set, this is the campaign to which this Drip belongs to.', null=True, on_delete=django.db.models.deletion.CASCADE, to='campaigns.Campaign'),
+        ),
+    ]

drip/migrations/0004_auto_20201230_2223.py

@@ -0,0 +1,20 @@
+# Generated by Django 3.0.7 on 2020-12-30 22:23
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('campaigns', '0001_initial'),
+        ('drip', '0003_drip_campaign'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='drip',
+            name='campaign',
+            field=models.ForeignKey(blank=True, default=None, help_text='If set, this is the campaign to which this Drip belongs to.', null=True, on_delete=django.db.models.deletion.SET_DEFAULT, to='campaigns.Campaign'),
+        ),
+    ]

drip/models.py

@@ -48,6 +48,15 @@ class AbstractDrip(models.Model):
         max_length=120, blank=True, default='default'
     )
 
+    campaign = models.ForeignKey(
+        'campaigns.Campaign',
+        null=True,
+        blank=True,
+        default=None,
+        on_delete=models.SET_DEFAULT,
+        help_text='If set, this is the campaign to which this Drip belongs to.'
+    )
+
     class Meta:
         abstract = True
 

testsettings.py

@@ -20,6 +20,7 @@
     'django.contrib.messages',
 
     'drip',
+    'campaigns',
 
     # testing only
     'credits',
@@ -68,3 +69,5 @@
 STATICFILES_DIRS = ()
 
 EMAIL_BACKEND = 'django.core.mail.backends.console.EmailBackend'
+
+DRIP_CLASS_NAME = 'drip.Drip'