From e8a32c717babee132626d6574f7ca706338739dc Mon Sep 17 00:00:00 2001
From: Leon Sandøy <leon.sandoy@conmodo.com>
Date: Wed, 15 Jul 2020 15:32:23 +0200
Subject: Add a UniqueConstraint to prevent duplicates.

https://github.com/python-discord/site/issues/305
---
 pydis_site/apps/api/migrations/0057_allowlist.py   | 29 -----------------
 .../migrations/0057_create_new_allowlist_model.py  | 37 ++++++++++++++++++++++
 pydis_site/apps/api/models/bot/allowlist.py        | 13 ++++++++
 3 files changed, 50 insertions(+), 29 deletions(-)
 delete mode 100644 pydis_site/apps/api/migrations/0057_allowlist.py
 create mode 100644 pydis_site/apps/api/migrations/0057_create_new_allowlist_model.py

(limited to 'pydis_site')

diff --git a/pydis_site/apps/api/migrations/0057_allowlist.py b/pydis_site/apps/api/migrations/0057_allowlist.py
deleted file mode 100644
index 7d815e91..00000000
--- a/pydis_site/apps/api/migrations/0057_allowlist.py
+++ /dev/null
@@ -1,29 +0,0 @@
-# Generated by Django 3.0.8 on 2020-07-15 11:23
-
-from django.db import migrations, models
-import pydis_site.apps.api.models.mixins
-
-
-class Migration(migrations.Migration):
-
-    dependencies = [
-        ('api', '0056_allow_blank_user_roles'),
-    ]
-
-    operations = [
-        migrations.CreateModel(
-            name='AllowList',
-            fields=[
-                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
-                ('created_at', models.DateTimeField(auto_now_add=True)),
-                ('updated_at', models.DateTimeField(auto_now=True)),
-                ('type', models.CharField(choices=[('GUILD_INVITE_ID', 'Guild Invite Id'), ('FILE_FORMAT', 'File Format'), ('DOMAIN_NAME', 'Domain Name'), ('WORD_WATCHLIST', 'Word Watchlist')], help_text='The type of allowlist this is on.', max_length=50)),
-                ('allowed', models.BooleanField(help_text='Whether this item is on the allowlist or the denylist.')),
-                ('content', models.TextField(help_text='The data to add to the allowlist.')),
-            ],
-            options={
-                'abstract': False,
-            },
-            bases=(pydis_site.apps.api.models.mixins.ModelReprMixin, models.Model),
-        ),
-    ]
diff --git a/pydis_site/apps/api/migrations/0057_create_new_allowlist_model.py b/pydis_site/apps/api/migrations/0057_create_new_allowlist_model.py
new file mode 100644
index 00000000..45650d86
--- /dev/null
+++ b/pydis_site/apps/api/migrations/0057_create_new_allowlist_model.py
@@ -0,0 +1,37 @@
+# Generated by Django 3.0.8 on 2020-07-15 11:23
+
+from django.db import migrations, models
+import pydis_site.apps.api.models.mixins
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('api', '0056_allow_blank_user_roles'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='AllowList',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('created_at', models.DateTimeField(auto_now_add=True)),
+                ('updated_at', models.DateTimeField(auto_now=True)),
+                ('type', models.CharField(choices=[('GUILD_INVITE_ID', 'Guild Invite Id'), ('FILE_FORMAT', 'File Format'), ('DOMAIN_NAME', 'Domain Name'), ('WORD_WATCHLIST', 'Word Watchlist')], help_text='The type of allowlist this is on.', max_length=50)),
+                ('allowed', models.BooleanField(help_text='Whether this item is on the allowlist or the denylist.')),
+                ('content', models.TextField(help_text='The data to add to the allowlist.')),
+            ],
+            options={
+                'abstract': False,
+            },
+            bases=(pydis_site.apps.api.models.mixins.ModelReprMixin, models.Model),
+        ),
+        migrations.AlterModelTable(
+            name='allowlist',
+            table='allow_list',
+        ),
+        migrations.AddConstraint(
+            model_name='allowlist',
+            constraint=models.UniqueConstraint(fields=('content', 'type'), name='unique_allowlist'),
+        )
+    ]
diff --git a/pydis_site/apps/api/models/bot/allowlist.py b/pydis_site/apps/api/models/bot/allowlist.py
index b0aea066..fc57ef32 100644
--- a/pydis_site/apps/api/models/bot/allowlist.py
+++ b/pydis_site/apps/api/models/bot/allowlist.py
@@ -24,3 +24,16 @@ class AllowList(ModelTimestampMixin, ModelReprMixin, models.Model):
     content = models.TextField(
         help_text="The data to add to the allowlist."
     )
+
+    class Meta:
+        """Metaconfig for this model."""
+
+        db_table = 'allow_list'
+
+        # This constraint ensures only one allowlist with the same content
+        # can exist per type.This means that we cannot have both an allow
+        # and a deny for the same item, and we cannot have duplicates of the
+        # same item.
+        constraints = [
+            models.UniqueConstraint(fields=['content', 'type'], name='unique_allowlist'),
+        ]
-- 
cgit v1.2.3