parent
3458a08383
commit
b23b55ca80
|
@ -52,6 +52,7 @@ class PropertyType(Enum):
|
||||||
|
|
||||||
@dataclass
|
@dataclass
|
||||||
class Address:
|
class Address:
|
||||||
|
full_line: str | None = None
|
||||||
street: str | None = None
|
street: str | None = None
|
||||||
unit: str | None = None
|
unit: str | None = None
|
||||||
city: str | None = None
|
city: str | None = None
|
||||||
|
|
|
@ -52,6 +52,7 @@ class RealtorScraper(Scraper):
|
||||||
listing_id
|
listing_id
|
||||||
}
|
}
|
||||||
address {
|
address {
|
||||||
|
line
|
||||||
street_direction
|
street_direction
|
||||||
street_number
|
street_number
|
||||||
street_name
|
street_name
|
||||||
|
@ -236,6 +237,7 @@ class RealtorScraper(Scraper):
|
||||||
stories
|
stories
|
||||||
}
|
}
|
||||||
address {
|
address {
|
||||||
|
line
|
||||||
street_direction
|
street_direction
|
||||||
street_number
|
street_number
|
||||||
street_name
|
street_name
|
||||||
|
@ -352,6 +354,7 @@ class RealtorScraper(Scraper):
|
||||||
street_number
|
street_number
|
||||||
street_name
|
street_name
|
||||||
street_suffix
|
street_suffix
|
||||||
|
line
|
||||||
unit
|
unit
|
||||||
city
|
city
|
||||||
state_code
|
state_code
|
||||||
|
@ -657,6 +660,8 @@ class RealtorScraper(Scraper):
|
||||||
if not self.extra_property_data:
|
if not self.extra_property_data:
|
||||||
return {}
|
return {}
|
||||||
|
|
||||||
|
#: TODO: migrate "advertisers" and "estimates" to general query
|
||||||
|
|
||||||
query = """query GetHome($property_id: ID!) {
|
query = """query GetHome($property_id: ID!) {
|
||||||
home(property_id: $property_id) {
|
home(property_id: $property_id) {
|
||||||
__typename
|
__typename
|
||||||
|
@ -765,6 +770,7 @@ class RealtorScraper(Scraper):
|
||||||
address = result["address"]
|
address = result["address"]
|
||||||
|
|
||||||
return Address(
|
return Address(
|
||||||
|
full_line=address.get("line"),
|
||||||
street=" ".join(
|
street=" ".join(
|
||||||
part for part in [
|
part for part in [
|
||||||
address.get("street_number"),
|
address.get("street_number"),
|
||||||
|
|
|
@ -10,6 +10,7 @@ ordered_properties = [
|
||||||
"status",
|
"status",
|
||||||
"text",
|
"text",
|
||||||
"style",
|
"style",
|
||||||
|
"full_street_line",
|
||||||
"street",
|
"street",
|
||||||
"unit",
|
"unit",
|
||||||
"city",
|
"city",
|
||||||
|
@ -55,6 +56,7 @@ def process_result(result: Property) -> pd.DataFrame:
|
||||||
|
|
||||||
if "address" in prop_data:
|
if "address" in prop_data:
|
||||||
address_data = prop_data["address"]
|
address_data = prop_data["address"]
|
||||||
|
prop_data["full_street_line"] = address_data.full_line
|
||||||
prop_data["street"] = address_data.street
|
prop_data["street"] = address_data.street
|
||||||
prop_data["unit"] = address_data.unit
|
prop_data["unit"] = address_data.unit
|
||||||
prop_data["city"] = address_data.city
|
prop_data["city"] = address_data.city
|
||||||
|
|
|
@ -1,6 +1,6 @@
|
||||||
[tool.poetry]
|
[tool.poetry]
|
||||||
name = "homeharvest"
|
name = "homeharvest"
|
||||||
version = "0.3.22"
|
version = "0.3.23"
|
||||||
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"
|
||||||
|
|
|
@ -156,3 +156,12 @@ def test_realtor_without_extra_details():
|
||||||
]
|
]
|
||||||
|
|
||||||
assert results[0] != results[1]
|
assert results[0] != results[1]
|
||||||
|
|
||||||
|
|
||||||
|
def test_pr_zip_code():
|
||||||
|
results = scrape_property(
|
||||||
|
location="00741",
|
||||||
|
listing_type="for_sale",
|
||||||
|
)
|
||||||
|
|
||||||
|
assert results is not None and len(results) > 0
|
||||||
|
|
Loading…
Reference in New Issue