Fix issue with location context (#4334)

kartoza · Oct 30, 2024 · dbe1a3e · dbe1a3e
1 parent 627a98e
commit dbe1a3e
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 10 deletions.
diff --git a/bims/models/location_site.py b/bims/models/location_site.py
@@ -360,14 +360,26 @@ def add_context_group(self, group_key):
 
                 for result in feature_data['result']:
                     if context_key in result['feature']:
-                        LocationContext.objects.update_or_create(
-                            site=self,
-                            group=context_group,
-                            defaults={
-                                'fetch_time': timezone.now(),
-                                'value': str(result['feature'][context_key])
-                            }
-                        )
+                        try:
+                            LocationContext.objects.update_or_create(
+                                site=self,
+                                group=context_group,
+                                defaults={
+                                    'fetch_time': timezone.now(),
+                                    'value': str(result['feature'][context_key])
+                                }
+                            )
+                        except LocationContext.MultipleObjectsReturned:
+                            LocationContext.objects.filter(
+                                site=self,
+                                group=context_group,
+                            ).delete()
+                            LocationContext.objects.create(
+                                site=self,
+                                group=context_group,
+                                fetch_time=timezone.now(),
+                                value=str(result['feature'][context_key])
+                            )
             return True, 'Added'
 
         try:

diff --git a/bims/views/context_layers.py b/bims/views/context_layers.py
@@ -36,8 +36,11 @@ class LocationContextGroupSerializer(serializers.ModelSerializer):
 
     def get_native_layer_name(self, obj: LocationContextGroup):
         if obj.key and is_uuid(obj.key):
-            layer = Layer.objects.get(unique_id=obj.key)
-            return layer.name
+            try:
+                layer = Layer.objects.get(unique_id=obj.key)
+                return layer.name
+            except Layer.DoesNotExist:
+                return ''
         return ''
 
     def get_is_native_layer(self, obj: LocationContextGroup):