fix: keyerror on address

Bunsly · Sep 19, 2023 · 043f091 · 043f091
1 parent eae8108
commit 043f091
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/homeharvest/core/scrapers/redfin/__init__.py b/homeharvest/core/scrapers/redfin/__init__.py
@@ -49,21 +49,21 @@ def get_value(key: str) -> Any | None:
             unit = parse_unit(get_value("streetLine"))
             address = Address(
                 street_address=street_address,
-                city=home["city"],
-                state=home["state"],
-                zip_code=home["zip"],
+                city=home.get("city"),
+                state=home.get("state"),
+                zip_code=home.get("zip"),
                 unit=unit,
                 country="USA",
             )
         else:
-            address_info = home["streetAddress"]
-            street_address, unit = parse_address_two(address_info["assembledAddress"])
+            address_info = home.get("streetAddress")
+            street_address, unit = parse_address_two(address_info.get("assembledAddress"))
 
             address = Address(
                 street_address=street_address,
-                city=home["city"],
-                state=home["state"],
-                zip_code=home["zip"],
+                city=home.get("city"),
+                state=home.get("state"),
+                zip_code=home.get("zip"),
                 unit=unit,
                 country="USA",
             )