Add class masking (draft)

ami/main/admin.py

@@ -396,7 +396,7 @@ class OccurrenceAdmin(admin.ModelAdmin[Occurrence]):
         "determination__rank",
         "created_at",
     )
-    search_fields = ("determination__name", "determination__search_names")
+    search_fields = ("determination__name", "determination__search_names", "pk")
 
     def get_queryset(self, request: HttpRequest) -> QuerySet[Any]:
         qs = super().get_queryset(request)
@@ -418,10 +418,33 @@ def get_queryset(self, request: HttpRequest) -> QuerySet[Any]:
     def detections_count(self, obj) -> int:
         return obj.detections_count
 
+    @admin.action(description="Apply class masking to occurrences")
+    def apply_class_mask(self, request: HttpRequest, queryset: QuerySet[Occurrence]) -> QuerySet[Occurrence]:
+        """
+        Apply class masking to the queryset.
+        This is a placeholder for the actual implementation.
+        """
+        from ami.main.models import TaxaList
+        from ami.ml.models import Algorithm
+        from ami.ml.post_processing import class_masking
+
+        taxa_list = TaxaList.objects.get(pk=5)
+        algorithm = Algorithm.objects.get(pk=11)
+
+        for occurrence in queryset:
+            class_masking.update_single_occurrence(
+                occurrence=occurrence,
+                algorithm=algorithm,
+                taxa_list=taxa_list,
+            )
+        return queryset
+
+    actions = [apply_class_mask]
+
     ordering = ("-created_at",)
 
     # Add classifications as inline
-    inlines = [DetectionInline]
+    # inlines = [DetectionInline]
 
 
 @admin.register(Classification)