BE: Add embargo to unique fields

- Add embargo field to APITimeSeries unique constraint
- Add embargo field to CoreTimeSeries unique constraint
- Add embargo field to CoreHeadline unique constraint

Fixes #2977

Author: tushortz

metrics/data/migrations/0041_remove_apitimeseries_the_apitimeseries_record_should_be_unique_if_force_write_is_false_and_more.py

@@ -0,0 +1,90 @@
+# Generated by Django 5.2.9 on 2025-12-18 10:30
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("data", "0040_remove_age_and_stratum_from_rbac_permission"),
+    ]
+
+    operations = [
+        migrations.RemoveConstraint(
+            model_name="apitimeseries",
+            name="The `APITimeSeries` record should be unique if `force_write` is False",
+        ),
+        migrations.RemoveConstraint(
+            model_name="coreheadline",
+            name="The `CoreHeadline` record should be unique if `force_write` is False",
+        ),
+        migrations.RemoveConstraint(
+            model_name="coretimeseries",
+            name="The `CoreTimeSeries` record should be unique if `force_write` is False",
+        ),
+        migrations.AddConstraint(
+            model_name="apitimeseries",
+            constraint=models.UniqueConstraint(
+                condition=models.Q(("force_write", False)),
+                fields=(
+                    "metric",
+                    "topic",
+                    "theme",
+                    "sub_theme",
+                    "geography",
+                    "geography_type",
+                    "geography_code",
+                    "stratum",
+                    "age",
+                    "sex",
+                    "year",
+                    "month",
+                    "epiweek",
+                    "date",
+                    "metric_value",
+                    "in_reporting_delay_period",
+                    "embargo",
+                ),
+                name="The `APITimeSeries` record should be unique if `force_write` is False",
+            ),
+        ),
+        migrations.AddConstraint(
+            model_name="coreheadline",
+            constraint=models.UniqueConstraint(
+                condition=models.Q(("force_write", False)),
+                fields=(
+                    "metric",
+                    "geography",
+                    "stratum",
+                    "age",
+                    "sex",
+                    "period_start",
+                    "period_end",
+                    "metric_value",
+                    "embargo",
+                ),
+                name="The `CoreHeadline` record should be unique if `force_write` is False",
+            ),
+        ),
+        migrations.AddConstraint(
+            model_name="coretimeseries",
+            constraint=models.UniqueConstraint(
+                condition=models.Q(("force_write", False)),
+                fields=(
+                    "metric",
+                    "geography",
+                    "stratum",
+                    "age",
+                    "sex",
+                    "year",
+                    "month",
+                    "epiweek",
+                    "date",
+                    "metric_value",
+                    "in_reporting_delay_period",
+                    "embargo",
+                ),
+                name="The `CoreTimeSeries` record should be unique if `force_write` is False",
+            ),
+        ),
+    ]

metrics/data/models/api_models.py

@@ -67,6 +67,7 @@ class Meta:
                     "date",
                     "metric_value",
                     "in_reporting_delay_period",
+                    "embargo",
                 ),
                 name="The `APITimeSeries` record should be unique if `force_write` is False",
                 condition=Q(force_write=False),

metrics/data/models/core_models/headline.py

@@ -67,6 +67,7 @@ class Meta:
                     "period_start",
                     "period_end",
                     "metric_value",
+                    "embargo",
                 ),
                 name="The `CoreHeadline` record should be unique if `force_write` is False",
                 condition=Q(force_write=False),

metrics/data/models/core_models/timeseries.py

@@ -81,6 +81,7 @@ class Meta:
                     "date",
                     "metric_value",
                     "in_reporting_delay_period",
+                    "embargo",
                 ),
                 name="The `CoreTimeSeries` record should be unique if `force_write` is False",
                 condition=Q(force_write=False),