turtlebasket
/
HomeHarvest
mirror of https://github.com/Bunsly/HomeHarvest.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

- style error bug fix

pull/82/head v0.3.30
Zachary Hampton 2024-06-06 15:24:12 -07:00
parent 93e6778a48
commit 011680f7d8
3 changed files with 25 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
homeharvest/utils.py
View File

@ -83,19 +83,20 @@ def process_result(result: Property) -> pd.DataFrame:
prop_data["nearby_schools"] = ", ".join(set(prop_data["nearby_schools"])) if prop_data["nearby_schools"] else None prop_data["nearby_schools"] = ", ".join(set(prop_data["nearby_schools"])) if prop_data["nearby_schools"] else None
description = result.description description = result.description
prop_data["primary_photo"] = description.primary_photo if description:
prop_data["alt_photos"] = ", ".join(description.alt_photos) if description.alt_photos else None prop_data["primary_photo"] = description.primary_photo
prop_data["style"] = description.style if type(description.style) == str else description.style.value prop_data["alt_photos"] = ", ".join(description.alt_photos) if description.alt_photos else None
prop_data["beds"] = description.beds prop_data["style"] = description.style if isinstance(description.style, str) else description.style.value if description.style else None
prop_data["full_baths"] = description.baths_full prop_data["beds"] = description.beds
prop_data["half_baths"] = description.baths_half prop_data["full_baths"] = description.baths_full
prop_data["sqft"] = description.sqft prop_data["half_baths"] = description.baths_half
prop_data["lot_sqft"] = description.lot_sqft prop_data["sqft"] = description.sqft
prop_data["sold_price"] = description.sold_price prop_data["lot_sqft"] = description.lot_sqft
prop_data["year_built"] = description.year_built prop_data["sold_price"] = description.sold_price
prop_data["parking_garage"] = description.garage prop_data["year_built"] = description.year_built
prop_data["stories"] = description.stories prop_data["parking_garage"] = description.garage
prop_data["text"] = description.text prop_data["stories"] = description.stories
prop_data["text"] = description.text
properties_df = pd.DataFrame([prop_data]) properties_df = pd.DataFrame([prop_data])
properties_df = properties_df.reindex(columns=ordered_properties) properties_df = properties_df.reindex(columns=ordered_properties)

2
pyproject.toml
View File

@ -1,6 +1,6 @@
[tool.poetry] [tool.poetry]
name = "homeharvest" name = "homeharvest"
version = "0.3.29" version = "0.3.30"
description = "Real estate scraping library" description = "Real estate scraping library"
authors = ["Zachary Hampton <zachary@bunsly.com>", "Cullen Watson <cullen@bunsly.com>"] authors = ["Zachary Hampton <zachary@bunsly.com>", "Cullen Watson <cullen@bunsly.com>"]
homepage = "https://github.com/Bunsly/HomeHarvest" homepage = "https://github.com/Bunsly/HomeHarvest"

10
tests/test_realtor.py
View File

@ -175,3 +175,13 @@ def test_exclude_pending():
) )
assert results is not None and len(results) > 0 assert results is not None and len(results) > 0
def test_style_value_error():
results = scrape_property(
location="Alaska, AK",
listing_type="sold",
extra_property_data=False,
)
assert results is not None and len(results) > 0