From a79c4c6872298e3c21029bc70e5d45e59fa32dc7 Mon Sep 17 00:00:00 2001 From: Zachary Hampton <69336300+ZacharyHampton@users.noreply.github.com> Date: Fri, 15 Sep 2023 15:21:29 -0700 Subject: [PATCH] - housekeeping --- homeharvest/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/homeharvest/__init__.py b/homeharvest/__init__.py index 2b720a1..2c642eb 100644 --- a/homeharvest/__init__.py +++ b/homeharvest/__init__.py @@ -14,7 +14,7 @@ def scrape_property( listing_type: str = "for_sale", #: for_sale, for_rent, sold site_name: str = "redfin", ) -> list[Home]: #: eventually, return pandas dataframe - if site_name not in _scrapers: + if site_name.lower() not in _scrapers: raise InvalidSite(f"Provided site, '{site_name}', does not exist.") if listing_type.upper() not in ListingType.__members__: @@ -25,6 +25,6 @@ def scrape_property( listing_type=ListingType[listing_type.upper()], ) - site = _scrapers[site_name](scraper_input) + site = _scrapers[site_name.lower()](scraper_input) return site.search()