From 8e140a0e45fe4de1fb6cc9a4be5e54aff21dea39 Mon Sep 17 00:00:00 2001 From: Cullen Watson Date: Mon, 18 Sep 2023 17:04:54 -0500 Subject: [PATCH] chore: format --- homeharvest/core/scrapers/zillow/__init__.py | 4 +++- homeharvest/utils.py | 7 ++++++- 2 files changed, 9 insertions(+), 2 deletions(-) diff --git a/homeharvest/core/scrapers/zillow/__init__.py b/homeharvest/core/scrapers/zillow/__init__.py index 96c5fb9..4e55e4b 100644 --- a/homeharvest/core/scrapers/zillow/__init__.py +++ b/homeharvest/core/scrapers/zillow/__init__.py @@ -130,7 +130,9 @@ class ZillowScraper(Scraper): home_info = result["hdpData"]["homeInfo"] address_data = { "street_address": home_info["streetAddress"], - "unit": parse_address_two(home_info['unit']) if 'unit' in home_info else None, + "unit": parse_address_two(home_info["unit"]) + if "unit" in home_info + else None, "city": home_info["city"], "state": home_info["state"], "zip_code": home_info["zipcode"], diff --git a/homeharvest/utils.py b/homeharvest/utils.py index db85762..bb88d9b 100644 --- a/homeharvest/utils.py +++ b/homeharvest/utils.py @@ -4,7 +4,11 @@ import re def parse_address_two(street_address: str): if not street_address: return None - apt_match = re.search(r"(APT\s*[\dA-Z]+|#[\dA-Z]+|UNIT\s*[\dA-Z]+|LOT\s*[\dA-Z]+)$", street_address, re.I) + apt_match = re.search( + r"(APT\s*[\dA-Z]+|#[\dA-Z]+|UNIT\s*[\dA-Z]+|LOT\s*[\dA-Z]+)$", + street_address, + re.I, + ) if apt_match: apt_str = apt_match.group().strip() @@ -13,6 +17,7 @@ def parse_address_two(street_address: str): else: return None + if __name__ == "__main__": print(parse_address_two("810 E Colter St APT 32")) print(parse_address_two("1234 Elm Street apt 2B"))