feat: restructure sql ingestion
This commit is contained in:
Stijnvandenbroek
389
data_platform/assets/ingestion/funda/funda.py
Normal file
389
data_platform/assets/ingestion/funda/funda.py
Normal file
@@ -0,0 +1,389 @@
|
||||
"""Funda real-estate ingestion assets."""
|
||||
|
||||
import json
|
||||
from pathlib import Path
|
||||
|
||||
from dagster import (
|
||||
AssetExecutionContext,
|
||||
Config,
|
||||
MaterializeResult,
|
||||
MetadataValue,
|
||||
asset,
|
||||
)
|
||||
from sqlalchemy import text
|
||||
|
||||
from data_platform.helpers import (
|
||||
format_area,
|
||||
format_euro,
|
||||
md_preview_table,
|
||||
render_sql,
|
||||
safe_int,
|
||||
)
|
||||
from data_platform.resources import FundaResource, PostgresResource
|
||||
|
||||
_SQL_DIR = Path(__file__).parent / "sql"
|
||||
_SCHEMA = "raw_funda"
|
||||
|
||||
|
||||
class FundaSearchConfig(Config):
|
||||
"""Search parameters for Funda."""
|
||||
|
||||
location: str = "woerden, utrecht, zeist, maarssen, nieuwegein, gouda"
|
||||
offering_type: str = "buy"
|
||||
price_min: int | None = 300000
|
||||
price_max: int | None = 500000
|
||||
area_min: int | None = None
|
||||
area_max: int | None = None
|
||||
plot_min: int | None = None
|
||||
plot_max: int | None = None
|
||||
object_type: str | None = None
|
||||
energy_label: str | None = None
|
||||
radius_km: int | None = None
|
||||
sort: str = "newest"
|
||||
max_pages: int = 3
|
||||
|
||||
|
||||
class FundaDetailsConfig(Config):
|
||||
"""Config for listing details fetch."""
|
||||
|
||||
fetch_all: bool = True
|
||||
|
||||
|
||||
class FundaPriceHistoryConfig(Config):
|
||||
"""Config for price history fetch."""
|
||||
|
||||
fetch_all: bool = True
|
||||
|
||||
|
||||
@asset(
|
||||
group_name="funda",
|
||||
kinds={"python", "postgres"},
|
||||
description="Search Funda listings and store results in Postgres.",
|
||||
)
|
||||
def funda_search_results(
|
||||
context: AssetExecutionContext,
|
||||
config: FundaSearchConfig,
|
||||
funda: FundaResource,
|
||||
postgres: PostgresResource,
|
||||
) -> MaterializeResult:
|
||||
client = funda.get_client()
|
||||
|
||||
kwargs: dict = {
|
||||
"location": [loc.strip() for loc in config.location.split(",")],
|
||||
"offering_type": config.offering_type,
|
||||
"sort": config.sort,
|
||||
}
|
||||
if config.price_min is not None:
|
||||
kwargs["price_min"] = config.price_min
|
||||
if config.price_max is not None:
|
||||
kwargs["price_max"] = config.price_max
|
||||
if config.area_min is not None:
|
||||
kwargs["area_min"] = config.area_min
|
||||
if config.area_max is not None:
|
||||
kwargs["area_max"] = config.area_max
|
||||
if config.plot_min is not None:
|
||||
kwargs["plot_min"] = config.plot_min
|
||||
if config.plot_max is not None:
|
||||
kwargs["plot_max"] = config.plot_max
|
||||
if config.object_type:
|
||||
kwargs["object_type"] = [t.strip() for t in config.object_type.split(",")]
|
||||
if config.energy_label:
|
||||
kwargs["energy_label"] = [lbl.strip() for lbl in config.energy_label.split(",")]
|
||||
if config.radius_km is not None:
|
||||
kwargs["radius_km"] = config.radius_km
|
||||
|
||||
all_listings = []
|
||||
for page in range(config.max_pages):
|
||||
context.log.info(f"Fetching search page {page + 1}/{config.max_pages} …")
|
||||
kwargs["page"] = page
|
||||
results = client.search_listing(**kwargs)
|
||||
if not results:
|
||||
context.log.info("No more results.")
|
||||
break
|
||||
all_listings.extend(results)
|
||||
context.log.info(f" got {len(results)} listings (total: {len(all_listings)})")
|
||||
|
||||
if not all_listings:
|
||||
context.log.warning("Search returned zero results.")
|
||||
return MaterializeResult(metadata={"count": 0})
|
||||
|
||||
engine = postgres.get_engine()
|
||||
with engine.begin() as conn:
|
||||
conn.execute(text(f"CREATE SCHEMA IF NOT EXISTS {_SCHEMA}"))
|
||||
conn.execute(
|
||||
text(render_sql(_SQL_DIR, "ddl/create_search_results.sql", schema=_SCHEMA))
|
||||
)
|
||||
conn.execute(
|
||||
text(
|
||||
render_sql(
|
||||
_SQL_DIR, "ddl/migrate_search_constraint.sql", schema=_SCHEMA
|
||||
)
|
||||
)
|
||||
)
|
||||
|
||||
rows = []
|
||||
for listing in all_listings:
|
||||
d = listing.to_dict()
|
||||
rows.append(
|
||||
{
|
||||
"global_id": d.get("global_id"),
|
||||
"title": d.get("title"),
|
||||
"city": d.get("city"),
|
||||
"postcode": d.get("postcode"),
|
||||
"province": d.get("province"),
|
||||
"neighbourhood": d.get("neighbourhood"),
|
||||
"price": safe_int(d.get("price")),
|
||||
"living_area": safe_int(d.get("living_area")),
|
||||
"plot_area": safe_int(d.get("plot_area")),
|
||||
"bedrooms": safe_int(d.get("bedrooms")),
|
||||
"rooms": safe_int(d.get("rooms")),
|
||||
"energy_label": d.get("energy_label"),
|
||||
"object_type": d.get("object_type"),
|
||||
"offering_type": d.get("offering_type"),
|
||||
"construction_type": d.get("construction_type"),
|
||||
"publish_date": d.get("publish_date"),
|
||||
"broker_id": str(d.get("broker_id", "")),
|
||||
"broker_name": d.get("broker_name"),
|
||||
"raw_json": json.dumps(d, default=str),
|
||||
}
|
||||
)
|
||||
|
||||
postgres.execute_many(
|
||||
render_sql(_SQL_DIR, "dml/insert_search_results.sql", schema=_SCHEMA), rows
|
||||
)
|
||||
|
||||
context.log.info(
|
||||
f"Inserted {len(rows)} search results into {_SCHEMA}.search_results"
|
||||
)
|
||||
|
||||
return MaterializeResult(
|
||||
metadata={
|
||||
"count": len(rows),
|
||||
"location": MetadataValue.text(config.location),
|
||||
"offering_type": MetadataValue.text(config.offering_type),
|
||||
"preview": MetadataValue.md(
|
||||
md_preview_table(
|
||||
rows[:10],
|
||||
columns=[
|
||||
("title", "Title"),
|
||||
("city", "City"),
|
||||
("price", "Price"),
|
||||
("living_area", "Area"),
|
||||
("bedrooms", "Bedrooms"),
|
||||
],
|
||||
formatters={"price": format_euro, "living_area": format_area},
|
||||
),
|
||||
),
|
||||
}
|
||||
)
|
||||
|
||||
|
||||
@asset(
|
||||
group_name="funda",
|
||||
kinds={"python", "postgres"},
|
||||
deps=[funda_search_results],
|
||||
description="Fetch full listing details for each search result and store in Postgres.",
|
||||
)
|
||||
def funda_listing_details(
|
||||
context: AssetExecutionContext,
|
||||
config: FundaDetailsConfig,
|
||||
funda: FundaResource,
|
||||
postgres: PostgresResource,
|
||||
) -> MaterializeResult:
|
||||
client = funda.get_client()
|
||||
engine = postgres.get_engine()
|
||||
|
||||
with engine.begin() as conn:
|
||||
conn.execute(text(f"CREATE SCHEMA IF NOT EXISTS {_SCHEMA}"))
|
||||
conn.execute(
|
||||
text(render_sql(_SQL_DIR, "ddl/create_listing_details.sql", schema=_SCHEMA))
|
||||
)
|
||||
conn.execute(
|
||||
text(
|
||||
render_sql(
|
||||
_SQL_DIR, "ddl/migrate_details_constraint.sql", schema=_SCHEMA
|
||||
)
|
||||
)
|
||||
)
|
||||
|
||||
with engine.connect() as conn:
|
||||
result = conn.execute(
|
||||
text(f"SELECT DISTINCT global_id FROM {_SCHEMA}.search_results")
|
||||
)
|
||||
ids = [row[0] for row in result if row[0]]
|
||||
|
||||
if not ids:
|
||||
context.log.warning("No search results found – run funda_search_results first.")
|
||||
return MaterializeResult(metadata={"count": 0})
|
||||
|
||||
context.log.info(f"Fetching details for {len(ids)} listings …")
|
||||
|
||||
rows = []
|
||||
errors = 0
|
||||
for i, gid in enumerate(ids):
|
||||
try:
|
||||
listing = client.get_listing(int(gid))
|
||||
d = listing.to_dict()
|
||||
rows.append(
|
||||
{
|
||||
"global_id": d.get("global_id"),
|
||||
"tiny_id": str(d.get("tiny_id", "")),
|
||||
"title": d.get("title"),
|
||||
"city": d.get("city"),
|
||||
"postcode": d.get("postcode"),
|
||||
"province": d.get("province"),
|
||||
"neighbourhood": d.get("neighbourhood"),
|
||||
"municipality": d.get("municipality"),
|
||||
"price": safe_int(d.get("price")),
|
||||
"price_formatted": d.get("price_formatted"),
|
||||
"status": d.get("status"),
|
||||
"offering_type": d.get("offering_type"),
|
||||
"object_type": d.get("object_type"),
|
||||
"house_type": d.get("house_type"),
|
||||
"construction_type": d.get("construction_type"),
|
||||
"construction_year": d.get("construction_year"),
|
||||
"energy_label": d.get("energy_label"),
|
||||
"living_area": safe_int(d.get("living_area")),
|
||||
"plot_area": safe_int(d.get("plot_area")),
|
||||
"bedrooms": safe_int(d.get("bedrooms")),
|
||||
"rooms": safe_int(d.get("rooms")),
|
||||
"description": d.get("description"),
|
||||
"publication_date": d.get("publication_date"),
|
||||
"latitude": d.get("latitude"),
|
||||
"longitude": d.get("longitude"),
|
||||
"has_garden": d.get("has_garden"),
|
||||
"has_balcony": d.get("has_balcony"),
|
||||
"has_solar_panels": d.get("has_solar_panels"),
|
||||
"has_heat_pump": d.get("has_heat_pump"),
|
||||
"has_roof_terrace": d.get("has_roof_terrace"),
|
||||
"is_energy_efficient": d.get("is_energy_efficient"),
|
||||
"is_monument": d.get("is_monument"),
|
||||
"url": d.get("url"),
|
||||
"photo_count": safe_int(d.get("photo_count")),
|
||||
"views": safe_int(d.get("views")),
|
||||
"saves": safe_int(d.get("saves")),
|
||||
"raw_json": json.dumps(d, default=str),
|
||||
}
|
||||
)
|
||||
except Exception as e:
|
||||
errors += 1
|
||||
context.log.warning(f"Failed to fetch listing {gid}: {e}")
|
||||
continue
|
||||
|
||||
if (i + 1) % 10 == 0:
|
||||
context.log.info(f" fetched {i + 1}/{len(ids)} …")
|
||||
|
||||
if rows:
|
||||
postgres.execute_many(
|
||||
render_sql(_SQL_DIR, "dml/insert_listing_details.sql", schema=_SCHEMA), rows
|
||||
)
|
||||
|
||||
context.log.info(
|
||||
f"Inserted {len(rows)} listing details ({errors} errors) into {_SCHEMA}.listing_details"
|
||||
)
|
||||
|
||||
return MaterializeResult(
|
||||
metadata={
|
||||
"count": len(rows),
|
||||
"errors": errors,
|
||||
"preview": MetadataValue.md(
|
||||
md_preview_table(
|
||||
rows[:10],
|
||||
columns=[
|
||||
("title", "Title"),
|
||||
("city", "City"),
|
||||
("price", "Price"),
|
||||
("status", "Status"),
|
||||
("energy_label", "Energy"),
|
||||
],
|
||||
formatters={"price": format_euro},
|
||||
),
|
||||
),
|
||||
}
|
||||
)
|
||||
|
||||
|
||||
@asset(
|
||||
group_name="funda",
|
||||
kinds={"python", "postgres"},
|
||||
deps=[funda_listing_details],
|
||||
description="Fetch price history for each detailed listing and store in Postgres.",
|
||||
)
|
||||
def funda_price_history(
|
||||
context: AssetExecutionContext,
|
||||
config: FundaPriceHistoryConfig,
|
||||
funda: FundaResource,
|
||||
postgres: PostgresResource,
|
||||
) -> MaterializeResult:
|
||||
client = funda.get_client()
|
||||
engine = postgres.get_engine()
|
||||
|
||||
with engine.begin() as conn:
|
||||
conn.execute(text(f"CREATE SCHEMA IF NOT EXISTS {_SCHEMA}"))
|
||||
conn.execute(
|
||||
text(render_sql(_SQL_DIR, "ddl/create_price_history.sql", schema=_SCHEMA))
|
||||
)
|
||||
conn.execute(
|
||||
text(
|
||||
render_sql(
|
||||
_SQL_DIR, "ddl/migrate_price_history_constraint.sql", schema=_SCHEMA
|
||||
)
|
||||
)
|
||||
)
|
||||
|
||||
with engine.connect() as conn:
|
||||
result = conn.execute(
|
||||
text(f"SELECT DISTINCT global_id FROM {_SCHEMA}.listing_details")
|
||||
)
|
||||
ids = [row[0] for row in result if row[0]]
|
||||
|
||||
if not ids:
|
||||
context.log.warning(
|
||||
"No listing details found – run funda_listing_details first."
|
||||
)
|
||||
return MaterializeResult(metadata={"count": 0})
|
||||
|
||||
context.log.info(f"Fetching price history for {len(ids)} listings …")
|
||||
|
||||
rows = []
|
||||
errors = 0
|
||||
for i, gid in enumerate(ids):
|
||||
try:
|
||||
listing = client.get_listing(int(gid))
|
||||
history = client.get_price_history(listing)
|
||||
for entry in history:
|
||||
rows.append(
|
||||
{
|
||||
"global_id": gid,
|
||||
"price": safe_int(entry.get("price")),
|
||||
"human_price": entry.get("human_price"),
|
||||
"date": entry.get("date"),
|
||||
"timestamp": entry.get("timestamp"),
|
||||
"source": entry.get("source"),
|
||||
"status": entry.get("status"),
|
||||
}
|
||||
)
|
||||
except Exception as e:
|
||||
errors += 1
|
||||
context.log.warning(f"Failed to fetch price history for {gid}: {e}")
|
||||
continue
|
||||
|
||||
if (i + 1) % 10 == 0:
|
||||
context.log.info(f" fetched {i + 1}/{len(ids)} …")
|
||||
|
||||
if rows:
|
||||
postgres.execute_many(
|
||||
render_sql(_SQL_DIR, "dml/insert_price_history.sql", schema=_SCHEMA), rows
|
||||
)
|
||||
|
||||
context.log.info(
|
||||
f"Inserted {len(rows)} price history records ({errors} errors) into {_SCHEMA}.price_history"
|
||||
)
|
||||
|
||||
return MaterializeResult(
|
||||
metadata={
|
||||
"count": len(rows),
|
||||
"errors": errors,
|
||||
"listings_processed": len(ids) - errors,
|
||||
}
|
||||
)
|
||||
Reference in New Issue
Block a user