refactor: revert to use geom_type assignment expression

hotosm · Jan 19, 2024 · 3c9d03c · 3c9d03c
1 parent bd9c4ab
commit 3c9d03c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/backend/app/projects/project_crud.py b/src/backend/app/projects/project_crud.py
@@ -615,13 +615,13 @@ async def get_data_extract_url(
         }
     }
 
-    if aoi.get("type") == "FeatureCollection":
+    if geom_type := aoi.get("type") == "FeatureCollection":
         # Convert each feature into a Shapely geometry
         geometries = [
             shape(feature.get("geometry")) for feature in aoi.get("features", [])
         ]
         merged_geom = unary_union(geometries)
-    elif aoi.get("type") == "Feature":
+    elif geom_type == "Feature":
         merged_geom = shape(aoi.get("geometry"))
     else:
         merged_geom = shape(aoi)