feat(tradein/domclick): JSONL ingest script for local detail-backfill runner
scripts/ingest_domclick_jsonl.py — idempotent ingest of domclick_local_runner output: filter ok+price, upsert via save_listings (fault-tolerant house-match hook in SAVEPOINT — non-matched listings still persist), deep-detail enrichment via save_detail_enrichment (owners/encumbrances/kitchen/views/wall/floor/ price-history). --limit / --dry-run. psycopg v3 ANY(:ids) list-bind. Per-record try/except in enrichment loop so one bad row never aborts the tail. Live-tested in tradein-backend on 30 records: updated=30, house_matched=30, detail_applied=30, price_history idempotent (75->75), enrich_failed=0.
This commit is contained in:
parent
75f1d36505
commit
f7426dc12b
1 changed files with 291 additions and 0 deletions
291
tradein-mvp/backend/scripts/ingest_domclick_jsonl.py
Normal file
291
tradein-mvp/backend/scripts/ingest_domclick_jsonl.py
Normal file
|
|
@ -0,0 +1,291 @@
|
|||
"""Ингест JSONL от локального DomClick-раннера (`scripts/domclick_local_runner.py`) в БД tradein.
|
||||
|
||||
Запускается ВНУТРИ tradein-контейнера (импортит `app`). Идемпотентно.
|
||||
|
||||
Каждая ok-запись (status=="ok" и валидный price_rub) →
|
||||
1. upsert в `listings` через save_listings() (house-match хук — fault-tolerant, в SAVEPOINT);
|
||||
2. deep-detail enrichment через save_detail_enrichment()
|
||||
(owners / encumbrances / kitchen / views / wall / floor / price-history).
|
||||
|
||||
Listing.id не возвращается из save_listings — достаём отдельным SELECT по (source, source_id).
|
||||
|
||||
Usage:
|
||||
python -m scripts.ingest_domclick_jsonl --jsonl /tmp/domclick.jsonl [--limit N] [--dry-run]
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import argparse
|
||||
import json
|
||||
import logging
|
||||
from typing import Any
|
||||
|
||||
from sqlalchemy import text
|
||||
|
||||
from app.core.db import SessionLocal
|
||||
from app.services.scrapers.base import ScrapedLot, save_listings
|
||||
from app.services.scrapers.domclick_detail import (
|
||||
DomClickDetailEnrichment,
|
||||
save_detail_enrichment,
|
||||
)
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
SOURCE = "domklik"
|
||||
|
||||
|
||||
def _to_int(v: Any) -> int | None:
|
||||
"""int() None-safe: возвращает None на None / нечисловом мусоре (не крэшит прогон)."""
|
||||
if v is None:
|
||||
return None
|
||||
try:
|
||||
return int(v)
|
||||
except (TypeError, ValueError):
|
||||
return None
|
||||
|
||||
|
||||
def _to_float(v: Any) -> float | None:
|
||||
"""float() None-safe: возвращает None на None / нечисловом мусоре."""
|
||||
if v is None:
|
||||
return None
|
||||
try:
|
||||
return float(v)
|
||||
except (TypeError, ValueError):
|
||||
return None
|
||||
|
||||
|
||||
def _build_lot(rec: dict[str, Any]) -> ScrapedLot:
|
||||
"""Построить ScrapedLot из JSONL-записи. price_rub уже провалидирован caller'ом."""
|
||||
raw_extra = rec.get("raw_extra") or {}
|
||||
return ScrapedLot(
|
||||
source=SOURCE,
|
||||
source_url=rec["source_url"],
|
||||
source_id=str(rec["id"]),
|
||||
address=rec.get("address"),
|
||||
lat=_to_float(rec.get("lat")),
|
||||
lon=_to_float(rec.get("lon")),
|
||||
rooms=_to_int(rec.get("rooms")),
|
||||
area_m2=_to_float(rec.get("area_m2")),
|
||||
floor=_to_int(rec.get("floor")),
|
||||
total_floors=_to_int(rec.get("total_floors")),
|
||||
year_built=_to_int(rec.get("year_built")),
|
||||
price_rub=int(rec["price_rub"]),
|
||||
listing_segment="vtorichka",
|
||||
repair_state=rec.get("repair_state"),
|
||||
raw_payload={
|
||||
"is_sber_collateral": rec.get("is_sber_collateral"),
|
||||
"square_price": raw_extra.get("square_price"),
|
||||
"avm": rec.get("avm") or {},
|
||||
},
|
||||
)
|
||||
|
||||
|
||||
def _build_enrichment(rec: dict[str, Any]) -> DomClickDetailEnrichment:
|
||||
"""Построить deep-detail enrichment из JSONL-записи."""
|
||||
raw_extra_in = rec.get("raw_extra") or {}
|
||||
raw_extra: dict[str, Any] = {
|
||||
"wall_type": rec.get("wall_type"),
|
||||
"floor_type": rec.get("floor_type"),
|
||||
"calls": rec.get("calls"),
|
||||
"is_sber_collateral": rec.get("is_sber_collateral"),
|
||||
"avm": rec.get("avm") or {},
|
||||
**raw_extra_in,
|
||||
}
|
||||
return DomClickDetailEnrichment(
|
||||
item_id=str(rec["id"]),
|
||||
source_url=rec["source_url"],
|
||||
repair_state=rec.get("repair_state"),
|
||||
repair_type=rec.get("repair_raw"),
|
||||
living_area_m2=_to_float(rec.get("living_area_m2")),
|
||||
kitchen_area_m2=_to_float(rec.get("kitchen_area_m2")),
|
||||
sale_type=rec.get("sale_type"),
|
||||
owners_count=_to_int(rec.get("owners_count")),
|
||||
encumbrances_clean=rec.get("encumbrances_clean"),
|
||||
year_built=_to_int(rec.get("year_built")),
|
||||
views_total=_to_int(rec.get("views")),
|
||||
price_changes=rec.get("price_changes") or [],
|
||||
raw_extra=raw_extra,
|
||||
)
|
||||
|
||||
|
||||
def run(jsonl_path: str, limit: int | None, dry_run: bool) -> dict[str, int]:
|
||||
"""Прочитать JSONL, отфильтровать ok+price, upsert listings + enrichment.
|
||||
|
||||
Returns: dict со счётчиками для итогового лога / тестов.
|
||||
"""
|
||||
lots: list[ScrapedLot] = []
|
||||
valid_records: list[dict[str, Any]] = [] # (parallel to lots) для enrichment-шага
|
||||
|
||||
skipped_status = 0
|
||||
skipped_no_price = 0
|
||||
parse_failed = 0
|
||||
|
||||
with open(jsonl_path, encoding="utf-8") as fh:
|
||||
for line in fh:
|
||||
line = line.strip()
|
||||
if not line:
|
||||
continue
|
||||
try:
|
||||
rec = json.loads(line)
|
||||
except json.JSONDecodeError as exc:
|
||||
logger.warning("битая JSON-строка пропущена: %s", exc)
|
||||
parse_failed += 1
|
||||
continue
|
||||
|
||||
if rec.get("status") != "ok":
|
||||
skipped_status += 1
|
||||
continue
|
||||
price = _to_int(rec.get("price_rub"))
|
||||
if price is None or price <= 0:
|
||||
skipped_no_price += 1
|
||||
continue
|
||||
|
||||
try:
|
||||
lot = _build_lot(rec)
|
||||
except Exception as exc:
|
||||
logger.warning("не удалось построить lot для id=%s: %s", rec.get("id"), exc)
|
||||
parse_failed += 1
|
||||
continue
|
||||
|
||||
lots.append(lot)
|
||||
valid_records.append(rec)
|
||||
|
||||
if limit is not None and len(lots) >= limit:
|
||||
break
|
||||
|
||||
total_valid = len(lots)
|
||||
logger.info(
|
||||
"прочитано: valid=%d skipped_status=%d skipped_no_price=%d parse_failed=%d",
|
||||
total_valid,
|
||||
skipped_status,
|
||||
skipped_no_price,
|
||||
parse_failed,
|
||||
)
|
||||
|
||||
inserted = 0
|
||||
updated = 0
|
||||
house_matched = 0
|
||||
match_failed = 0
|
||||
detail_applied = 0
|
||||
price_history_attempted = 0
|
||||
enrich_failed = 0
|
||||
|
||||
if dry_run:
|
||||
logger.info("[dry-run] %d записей распарсено, запись в БД пропущена", total_valid)
|
||||
return {
|
||||
"inserted": inserted,
|
||||
"updated": updated,
|
||||
"house_matched": house_matched,
|
||||
"match_failed": match_failed,
|
||||
"detail_applied": detail_applied,
|
||||
"price_history_attempted": price_history_attempted,
|
||||
"enrich_failed": enrich_failed,
|
||||
"skipped_status": skipped_status,
|
||||
"skipped_no_price": skipped_no_price,
|
||||
"parse_failed": parse_failed,
|
||||
}
|
||||
|
||||
db = SessionLocal()
|
||||
try:
|
||||
if lots:
|
||||
inserted, updated = save_listings(db, lots)
|
||||
logger.info("save_listings: inserted=%d updated=%d", inserted, updated)
|
||||
|
||||
# Достаём listing.id + house_id_fk одним SELECT по source_id.
|
||||
source_ids = [lot.source_id for lot in lots if lot.source_id is not None]
|
||||
id_map: dict[str, tuple[int, Any]] = {}
|
||||
if source_ids:
|
||||
rows = db.execute(
|
||||
text(
|
||||
"SELECT source_id, id, house_id_fk FROM listings "
|
||||
"WHERE source = :source AND source_id = ANY(:ids)"
|
||||
),
|
||||
{"source": SOURCE, "ids": source_ids},
|
||||
).fetchall()
|
||||
for sid, lid, house_id_fk in rows:
|
||||
id_map[str(sid)] = (int(lid), house_id_fk)
|
||||
|
||||
total_saved = len(id_map)
|
||||
house_matched = sum(1 for _, hid in id_map.values() if hid is not None)
|
||||
match_failed = total_saved - house_matched
|
||||
|
||||
# Deep-detail enrichment по каждой ok-записи.
|
||||
for rec in valid_records:
|
||||
sid = str(rec["id"])
|
||||
entry = id_map.get(sid)
|
||||
if entry is None:
|
||||
logger.warning("listing id не найден для source_id=%s — enrichment пропущен", sid)
|
||||
continue
|
||||
listing_id, _ = entry
|
||||
# Per-record изоляция: save_detail_enrichment коммитит сама, поэтому
|
||||
# падение одной записи иначе оборвало бы остаток valid_records трейсбэком.
|
||||
# Уже закоммиченные строки персистят → повторный прогон добьёт хвост.
|
||||
try:
|
||||
enrichment = _build_enrichment(rec)
|
||||
price_history_attempted += len(enrichment.price_changes)
|
||||
if save_detail_enrichment(db, listing_id, enrichment):
|
||||
detail_applied += 1
|
||||
except Exception as exc:
|
||||
db.rollback()
|
||||
enrich_failed += 1
|
||||
logger.warning("enrichment упал для source_id=%s: %s", sid, exc)
|
||||
continue
|
||||
finally:
|
||||
db.close()
|
||||
|
||||
logger.info(
|
||||
"ИТОГО: saved(inserted=%d/updated=%d) house_matched=%d match_failed=%d "
|
||||
"detail_applied=%d price_history_attempted=%d enrich_failed=%d "
|
||||
"skipped_status=%d skipped_no_price=%d parse_failed=%d",
|
||||
inserted,
|
||||
updated,
|
||||
house_matched,
|
||||
match_failed,
|
||||
detail_applied,
|
||||
price_history_attempted,
|
||||
enrich_failed,
|
||||
skipped_status,
|
||||
skipped_no_price,
|
||||
parse_failed,
|
||||
)
|
||||
|
||||
return {
|
||||
"inserted": inserted,
|
||||
"updated": updated,
|
||||
"house_matched": house_matched,
|
||||
"match_failed": match_failed,
|
||||
"detail_applied": detail_applied,
|
||||
"price_history_attempted": price_history_attempted,
|
||||
"enrich_failed": enrich_failed,
|
||||
"skipped_status": skipped_status,
|
||||
"skipped_no_price": skipped_no_price,
|
||||
"parse_failed": parse_failed,
|
||||
}
|
||||
|
||||
|
||||
def main() -> None:
|
||||
parser = argparse.ArgumentParser(
|
||||
description="Ингест DomClick JSONL (от domclick_local_runner) в БД tradein.",
|
||||
)
|
||||
parser.add_argument("--jsonl", required=True, help="путь к JSONL-файлу раннера")
|
||||
parser.add_argument(
|
||||
"--limit",
|
||||
type=int,
|
||||
default=None,
|
||||
help="обработать первые N валидных (ok+price) записей",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--dry-run",
|
||||
action="store_true",
|
||||
help="парсить + логировать, но НЕ писать в БД",
|
||||
)
|
||||
args = parser.parse_args()
|
||||
run(args.jsonl, args.limit, args.dry_run)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
logging.basicConfig(
|
||||
level=logging.INFO,
|
||||
format="%(asctime)s %(levelname)s %(name)s: %(message)s",
|
||||
)
|
||||
main()
|
||||
Loading…
Add table
Reference in a new issue