Fix habitat value in occurrence upload (#4224)

kartoza · Sep 5, 2024 · bec6c5a · bec6c5a
1 parent 7da84bf
commit bec6c5a
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/bims/scripts/occurrences_upload.py b/bims/scripts/occurrences_upload.py
@@ -695,13 +695,16 @@ def process_data(self, row):
                 row, PRESENT))
 
         # -- Optional data - Habitat
-        if HABITAT in row and DataCSVUpload.row_value(row, HABITAT):
+        habitat_value = DataCSVUpload.row_value(row, HABITAT)
+        if HABITAT in row and habitat_value:
             habitat_choices = {
                 v: k for k, v in
                 BiologicalCollectionRecord.HABITAT_CHOICES
             }
+            if habitat_value in habitat_choices:
+                habitat_value = habitat_choices[habitat_value]
             optional_data['collection_habitat'] = (
-                habitat_choices[DataCSVUpload.row_value(row, HABITAT)]
+                habitat_value
             )
 
         # -- Optional data - Sampling method