Help ensure occurrences always have detections

RolnickLab · Dec 7, 2023 · a9b9591 · a9b9591
1 parent 1e97b23
commit a9b9591
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/ami/main/api/views.py b/ami/main/api/views.py
@@ -429,7 +429,8 @@ class OccurrenceViewSet(DefaultViewSet):
     """
 
     queryset = (
-        Occurrence.objects.annotate(
+        Occurrence.objects.filter(detections=None)
+        .annotate(
             detections_count=models.Count("detections", distinct=True),
             duration=models.Max("detections__timestamp") - models.Min("detections__timestamp"),
             first_appearance_time=models.Min("detections__timestamp__time"),

diff --git a/ami/main/models.py b/ami/main/models.py
@@ -1567,6 +1567,12 @@ def save(self, update_determination=True, *args, **kwargs):
                 save=True,
             )
 
+        if self.determination and not self.determination_score:
+            # This may happen for legacy occurrences that were created
+            # before the determination_score field was added
+            self.determination_score = self.get_determination_score()
+            self.save(update_determination=False)
+
     class Meta:
         ordering = ["-determination_score"]