add m2m field for QMRAReference in QMRATreatment

Author: adaurat

qmra/risk_assessment/admin.py

@@ -64,10 +64,14 @@ class QMRATreatmentAdmin(admin.ModelAdmin):
         ("name", "group"),
         ("bacteria_min", "bacteria_max"),
         "bacteria_reference",
+        "bacteria_references",
         ("viruses_min", "viruses_max"),
         "viruses_reference",
+        "viruses_references",
         ("protozoa_min", "protozoa_max"),
-        "protozoa_reference"
+        "protozoa_reference",
+        "protozoa_references"
     ]
+    filter_horizontal = ["bacteria_references", "viruses_references", "protozoa_references"]
 
     save_model = save_model

qmra/risk_assessment/migrations/0007_qmratreatment_bacteria_references_and_more.py

@@ -0,0 +1,28 @@
+# Generated by Django 5.0.6 on 2026-01-15 12:42
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('risk_assessment', '0006_qmrapathogen_qmrareference_qmrasource_qmraexposure_and_more'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='qmratreatment',
+            name='bacteria_references',
+            field=models.ManyToManyField(related_name='bacteria_lrvs', to='risk_assessment.qmrareference'),
+        ),
+        migrations.AddField(
+            model_name='qmratreatment',
+            name='protozoa_references',
+            field=models.ManyToManyField(related_name='protozoa_lrvs', to='risk_assessment.qmrareference'),
+        ),
+        migrations.AddField(
+            model_name='qmratreatment',
+            name='viruses_references',
+            field=models.ManyToManyField(related_name='viruses_lrvs', to='risk_assessment.qmrareference'),
+        ),
+    ]

qmra/risk_assessment/migrations/0008_auto_20260115_1342.py

@@ -0,0 +1,24 @@
+# Generated by Django 5.0.6 on 2026-01-15 12:42
+
+from django.db import migrations
+
+
+def migrate_references(apps, schema_editor):
+
+    QMRATreatment = apps.get_model("risk_assessment", "QMRATreatment")
+    for treatment in QMRATreatment.objects.all():
+
+        treatment.bacteria_references.add(treatment.bacteria_reference)
+        treatment.viruses_references.add(treatment.viruses_reference)
+        treatment.protozoa_references.add(treatment.protozoa_reference)
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('risk_assessment', '0007_qmratreatment_bacteria_references_and_more'),
+    ]
+
+    operations = [
+        migrations.RunPython(migrate_references),
+    ]

qmra/risk_assessment/qmra_models.py

@@ -270,14 +270,19 @@ class QMRATreatment(models.Model):
     bacteria_max: Optional[float] = models.FloatField(blank=True, null=True)
     bacteria_reference = models.ForeignKey(QMRAReference, blank=True, null=True, on_delete=models.CASCADE,
                                            related_name="bacteria_lrv")
+    bacteria_references = models.ManyToManyField(QMRAReference, related_name="bacteria_lrvs")
+
     viruses_min: Optional[float] = models.FloatField(blank=True, null=True)
     viruses_max: Optional[float] = models.FloatField(blank=True, null=True)
     viruses_reference = models.ForeignKey(QMRAReference, blank=True, null=True, on_delete=models.CASCADE,
                                           related_name="viruses_lrv")
+    viruses_references = models.ManyToManyField(QMRAReference, related_name="viruses_lrvs")
+
     protozoa_min: Optional[float] = models.FloatField(blank=True, null=True)
     protozoa_max: Optional[float] = models.FloatField(blank=True, null=True)
     protozoa_reference = models.ForeignKey(QMRAReference, blank=True, null=True, on_delete=models.CASCADE,
                                            related_name="protozoa_lrv")
+    protozoa_references = models.ManyToManyField(QMRAReference, related_name="protozoa_lrvs")
 
     @classmethod
     def from_dict(cls, data):