add number individuals

WildMeOrg · Oct 24, 2023 · c8c009e · c8c009e
1 parent 52ccbf4
commit c8c009e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/app/modules/sightings/models.py b/app/modules/sightings/models.py
@@ -949,6 +949,9 @@ def individuals(self):
                 indivs.add(enc.individual)
         return indivs
 
+    def get_number_individuals(self):
+        return len(self.individuals)
+
     def get_individual_names(self):
         names = set()
         for indiv in self.individuals:

diff --git a/app/modules/sightings/schemas.py b/app/modules/sightings/schemas.py
@@ -80,6 +80,7 @@ class ElasticsearchSightingSchema(BaseSightingSchema):
     encounters = base_fields.Function(lambda s: s.get_encounters_elasticsearch())
     numberImages = base_fields.Function(lambda s: s.get_number_assets())
     numberAnnotations = base_fields.Function(lambda s: s.get_number_annotations())
+    numberIndividuals = base_fields.Function(lambda s: s.get_number_individuals())
     # apparently sets do NOT work well, so we need to name-sets to lists here
     individualNames = base_fields.Function(lambda s: list(s.get_individual_names()))
     individualNamesWithContexts = base_fields.Function(
@@ -116,6 +117,7 @@ class Meta:
             'encounters',
             'numberImages',
             'numberAnnotations',
+            'numberIndividuals',
             'individualNames',
             'individualNamesWithContexts',
         )