add db constraint

shapiromatron · Mar 12, 2024 · 5534d61 · 5534d61
1 parent e445f72
commit 5534d61
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/hawc/apps/udf/migrations/0003_modeludfcontent.py b/hawc/apps/udf/migrations/0003_modeludfcontent.py
@@ -40,4 +40,8 @@ class Migration(migrations.Migration):
                 ),
             ],
         ),
+        migrations.AlterUniqueTogether(
+            name="modeludfcontent",
+            unique_together={("model_binding", "object_id")},
+        ),
     ]
diff --git a/hawc/apps/udf/models.py b/hawc/apps/udf/models.py
@@ -124,14 +124,14 @@ class ModelUDFContent(models.Model):
     )
     content_type = models.ForeignKey(ContentType, on_delete=models.CASCADE)
     object_id = models.PositiveIntegerField(null=True)
-    content_object = GenericForeignKey(
-        "content_type",
-        "object_id",
-    )
+    content_object = GenericForeignKey("content_type", "object_id")
     content = models.JSONField(blank=True, default=dict)
     created = models.DateTimeField(auto_now_add=True)
     last_updated = models.DateTimeField(auto_now=True)
 
+    class Meta:
+        unique_together = (("model_binding", "object_id"),)
+
     def get_content_as_list(self):
         schema = dynamic_forms.Schema.parse_obj(self.model_binding.form.schema)