From e5de3da16ec107aebc072eabc070ed69be3c31cb Mon Sep 17 00:00:00 2001
From: Frankie Dunbar <franklyn.dunbar@umontana.edu>
Date: Mon, 13 Mar 2023 10:30:32 -0600
Subject: [PATCH 1/2] clean up geojson input

---
 driptorch/io.py | 45 ++++++++++++++++++++++++++++++++-------------
 1 file changed, 32 insertions(+), 13 deletions(-)

diff --git a/driptorch/io.py b/driptorch/io.py
index 6eb1fe3..5988c7d 100644
--- a/driptorch/io.py
+++ b/driptorch/io.py
@@ -144,6 +144,32 @@ def to_wgs84(cls, geometry: Union[BaseGeometry, dict], src_epsg: int) -> Union[B
         return projector.forward(geometry)
 
 
+def find_dicts_with_keys(d, key1: str = "type", key2: str = "coordinates"):
+    """Recursively traverse the dictionary and return a list of all children
+    dictionaries that contain the specific keys.
+
+    Wrote with intention to effectively parse geojson dicts
+
+    Args:
+        d (_type_): dictionary
+        key1 (str, optional): first key to search. Defaults to "type".
+        key2 (str, optional): second key to search. Defaults to "coordinates".
+
+    Returns:
+        _type_: list of found dicts containing key1 and key2
+    """
+    result = []
+    if isinstance(d, dict):
+        if key1 in d and key2 in d:
+            result.append(d)
+        for v in d.values():
+            result += find_dicts_with_keys(v, key1, key2)
+    elif isinstance(d, list):
+        for i in d:
+            result += find_dicts_with_keys(i, key1, key2)
+    return result
+
+
 def read_geojson_polygon(geojson: dict) -> Polygon:
     """Parse a GeoJSON to a shapely Polygon
 
@@ -163,24 +189,17 @@ def read_geojson_polygon(geojson: dict) -> Polygon:
         Shapely polygon geometry
     """
 
+    geojson = {
+        "geometry": find_dicts_with_keys(geojson)[0]
+    }
+
     # If the geojson is a feature collection then we loop over the features
     # and find the first instance of a polygon geometry type
-    if geojson['type'] == 'FeatureCollection':
-        for feature in geojson['features']:
-            if feature['geometry']['type'].lower() == 'polygon':
-                geometry = shape(feature['geometry'])
-                break
-
-    # Maybe it's just the geometry?
-    elif geojson['type'].lower() == 'Polygon':
-        geometry = shape(geojson)
-
-    # Fix your shit, we're not gonna to keep trying to guess
+    if geojson['geometry']['type'].lower() == "polygon":
+        return shape(geojson)
     else:
         raise GeojsonError(GeojsonError.read_error)
 
-    return geometry
-
 
 def write_geojson(geometries: list[BaseGeometry], src_epsg: int, dst_epsg: int = 4326, properties={},
                   style={}, elapsed_time=None) -> dict:

From a447dd9da165d600577ec47689df62b9c7f034e8 Mon Sep 17 00:00:00 2001
From: Frankie Dunbar <franklyn.dunbar@umontana.edu>
Date: Mon, 13 Mar 2023 10:34:45 -0600
Subject: [PATCH 2/2] docstring typo

---
 driptorch/io.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/driptorch/io.py b/driptorch/io.py
index 5988c7d..d407f2b 100644
--- a/driptorch/io.py
+++ b/driptorch/io.py
@@ -151,12 +151,12 @@ def find_dicts_with_keys(d, key1: str = "type", key2: str = "coordinates"):
     Wrote with intention to effectively parse geojson dicts
 
     Args:
-        d (_type_): dictionary
+        d (dict): input dictionary
         key1 (str, optional): first key to search. Defaults to "type".
         key2 (str, optional): second key to search. Defaults to "coordinates".
 
     Returns:
-        _type_: list of found dicts containing key1 and key2
+        list: list of found dicts containing key1 and key2
     """
     result = []
     if isinstance(d, dict):