Handle empty geoshapes and geotraces in data

Signed-off-by: Kipchirchir Sigei <[email protected]>
onaio · Oct 11, 2023 · 08d4e04 · 08d4e04
1 parent 0243b4f
commit 08d4e04
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/onadata/libs/serializers/geojson_serializer.py b/onadata/libs/serializers/geojson_serializer.py
@@ -141,9 +141,10 @@ def to_representation(self, instance):
                 points = instance.json.get(geo_field)
                 if geo_field in geotrace_xpaths or geo_field in polygon_xpaths:
                     value = get_values_matching_key(instance.json, geo_field)
+                    # handle empty geoms
                     try:
                         points = next(value)
-                    except TypeError:
+                    except StopIteration:
                         points = None
                 geometry = (
                     geometry_from_string(points, simple_style)

diff --git a/onadata/libs/utils/dict_tools.py b/onadata/libs/utils/dict_tools.py
@@ -22,8 +22,14 @@ def _get_values(doc, key):
                         yield item
                 elif isinstance(v, list):
                     for i in v:
-                        for j in _get_values(i, key):
-                            yield j
+                        if isinstance(i, (dict, list)):
+                            try:
+                                for j in _get_values(i, key):
+                                    yield j
+                            except StopIteration:
+                                continue
+                        elif i == key:
+                            yield i
 
     return _get_values(doc, key)