307 lines
12 KiB
Python
307 lines
12 KiB
Python
"""Batch backfill для Cian historical data.
|
||
|
||
Запускается:
|
||
- Manual через POST /admin/scrape/cian-backfill-history
|
||
- Future: in-app scheduler (после bootstrap'а Celery beat)
|
||
|
||
Что делает:
|
||
1. SELECT listings WHERE source='cian' AND source_url IS NOT NULL
|
||
AND NOT EXISTS (SELECT 1 FROM offer_price_history WHERE listing_id=...)
|
||
LIMIT batch_size
|
||
→ fetch_detail + save_detail_enrichment per listing
|
||
2. Houses block: SELECT houses WHERE cian_zhk_url IS NOT NULL
|
||
AND NOT EXISTS (SELECT 1 FROM houses_price_dynamics WHERE house_id=...)
|
||
LIMIT batch_size
|
||
→ fetch_newbuilding(cian_zhk_url) + save_newbuilding_enrichment per house
|
||
Requires migration 071_houses_cian_zhk_url.sql (cian_zhk_url column).
|
||
|
||
Rate limit: scraper_settings.get_scraper_delay('cian') between requests.
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
import asyncio
|
||
import logging
|
||
import time
|
||
from dataclasses import dataclass, field
|
||
|
||
from sqlalchemy import text
|
||
from sqlalchemy.orm import Session
|
||
|
||
from app.services.scraper_settings import get_scraper_delay
|
||
from app.services.scrapers.cian_detail import fetch_detail, save_detail_enrichment
|
||
from app.services.scrapers.cian_valuation import estimate_via_cian_valuation
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
@dataclass
|
||
class CianBackfillResult:
|
||
listings_total: int = 0
|
||
listings_processed: int = 0
|
||
listings_succeeded: int = 0
|
||
listings_failed_fetch: int = 0
|
||
listings_failed_save: int = 0
|
||
price_changes_attempted: int = 0
|
||
houses_total: int = 0
|
||
houses_processed: int = 0
|
||
houses_succeeded: int = 0
|
||
houses_failed_fetch: int = 0
|
||
houses_failed_save: int = 0
|
||
valuations_total: int = 0
|
||
valuations_processed: int = 0
|
||
valuations_succeeded: int = 0
|
||
valuations_failed: int = 0
|
||
duration_sec: float = field(default=0.0)
|
||
|
||
|
||
async def backfill_cian_history(
|
||
db: Session,
|
||
*,
|
||
batch_size: int = 50,
|
||
do_listings: bool = True,
|
||
do_houses: bool = True,
|
||
do_valuations: bool = False,
|
||
dry_run: bool = False,
|
||
) -> CianBackfillResult:
|
||
"""Iterate Cian listings + houses with missing history, fetch+save.
|
||
|
||
Args:
|
||
db: SQLAlchemy session (caller-owned; each entity commits internally via
|
||
save_detail_enrichment / save_newbuilding_enrichment).
|
||
batch_size: max entities to process per call (listings + houses + valuations counted
|
||
separately).
|
||
do_listings: process listings batch (offer_price_history backfill).
|
||
do_houses: process houses batch (houses_price_dynamics backfill via
|
||
fetch_newbuilding). Requires migration 071_houses_cian_zhk_url.sql applied.
|
||
do_valuations: process Cian Valuation Calculator batch (external_valuations backfill).
|
||
Default False — opt-in because each call hits Cian auth-gated API.
|
||
dry_run: skip all fetch+save; only count and log pending rows.
|
||
|
||
Returns:
|
||
CianBackfillResult with per-domain counters + total wall-clock duration.
|
||
"""
|
||
result = CianBackfillResult()
|
||
t0 = time.time()
|
||
delay = get_scraper_delay("cian") # seconds; default 5.0
|
||
|
||
# ── 1. Listings: missing offer_price_history ──────────────────────────────
|
||
if do_listings:
|
||
rows = (
|
||
db.execute(
|
||
text("""
|
||
SELECT l.id, l.source_url
|
||
FROM listings l
|
||
LEFT JOIN offer_price_history oph ON oph.listing_id = l.id
|
||
WHERE l.source = 'cian'
|
||
AND l.source_url IS NOT NULL
|
||
AND oph.listing_id IS NULL
|
||
LIMIT :lim
|
||
"""),
|
||
{"lim": batch_size},
|
||
)
|
||
.mappings()
|
||
.all()
|
||
)
|
||
result.listings_total = len(rows)
|
||
|
||
if dry_run:
|
||
logger.info("dry_run: would process %d cian listings", result.listings_total)
|
||
else:
|
||
for row in rows:
|
||
listing_id: int = row["id"]
|
||
source_url: str = row["source_url"]
|
||
result.listings_processed += 1
|
||
|
||
enrichment = None
|
||
try:
|
||
enrichment = await fetch_detail(source_url)
|
||
except Exception as exc:
|
||
logger.warning(
|
||
"cian_detail fetch failed for listing_id=%s url=%s: %s",
|
||
listing_id,
|
||
source_url,
|
||
exc,
|
||
)
|
||
result.listings_failed_fetch += 1
|
||
await asyncio.sleep(delay)
|
||
continue
|
||
|
||
if enrichment is None:
|
||
logger.warning(
|
||
"cian_detail fetch returned None for listing_id=%s url=%s",
|
||
listing_id,
|
||
source_url,
|
||
)
|
||
result.listings_failed_fetch += 1
|
||
await asyncio.sleep(delay)
|
||
continue
|
||
|
||
try:
|
||
save_detail_enrichment(db, listing_id, enrichment)
|
||
result.listings_succeeded += 1
|
||
result.price_changes_attempted += len(enrichment.price_changes or [])
|
||
except Exception as exc:
|
||
logger.warning(
|
||
"cian_detail save failed for listing_id=%s: %s", listing_id, exc
|
||
)
|
||
result.listings_failed_save += 1
|
||
|
||
await asyncio.sleep(delay)
|
||
|
||
# ── 2. Houses: missing houses_price_dynamics ──────────────────────────────
|
||
if do_houses:
|
||
from app.services.scrapers.cian_newbuilding import (
|
||
fetch_newbuilding,
|
||
save_newbuilding_enrichment,
|
||
)
|
||
|
||
house_rows = (
|
||
db.execute(
|
||
text("""
|
||
SELECT h.id, h.cian_zhk_url
|
||
FROM houses h
|
||
WHERE h.cian_zhk_url IS NOT NULL
|
||
AND NOT EXISTS (
|
||
SELECT 1 FROM houses_price_dynamics hpd
|
||
WHERE hpd.house_id = h.id
|
||
)
|
||
LIMIT :lim
|
||
"""),
|
||
{"lim": batch_size},
|
||
)
|
||
.mappings()
|
||
.all()
|
||
)
|
||
result.houses_total = len(house_rows)
|
||
|
||
if dry_run:
|
||
logger.info("dry_run: would process %d cian houses", result.houses_total)
|
||
else:
|
||
for hrow in house_rows:
|
||
house_id: int = hrow["id"]
|
||
zhk_url: str = hrow["cian_zhk_url"]
|
||
result.houses_processed += 1
|
||
|
||
enrichment = None
|
||
try:
|
||
enrichment = await fetch_newbuilding(zhk_url)
|
||
except Exception as exc:
|
||
logger.warning(
|
||
"cian_newbuilding fetch failed for house_id=%s url=%s: %s",
|
||
house_id,
|
||
zhk_url,
|
||
exc,
|
||
)
|
||
result.houses_failed_fetch += 1
|
||
await asyncio.sleep(delay)
|
||
continue
|
||
|
||
if enrichment is None:
|
||
logger.warning(
|
||
"cian_newbuilding fetch returned None for house_id=%s url=%s",
|
||
house_id,
|
||
zhk_url,
|
||
)
|
||
result.houses_failed_fetch += 1
|
||
await asyncio.sleep(delay)
|
||
continue
|
||
|
||
try:
|
||
save_newbuilding_enrichment(db, house_id, enrichment)
|
||
result.houses_succeeded += 1
|
||
except Exception as exc:
|
||
logger.warning(
|
||
"cian_newbuilding save failed for house_id=%s: %s", house_id, exc
|
||
)
|
||
result.houses_failed_save += 1
|
||
# Roll back to clean session state so next house can proceed.
|
||
# save_newbuilding_enrichment commits on success; on failure the
|
||
# transaction is left open/dirty — rollback to avoid session poison.
|
||
try:
|
||
db.rollback()
|
||
except Exception as rb_exc:
|
||
logger.warning(
|
||
"cian_newbuilding rollback failed for house_id=%s: %s",
|
||
house_id,
|
||
rb_exc,
|
||
)
|
||
|
||
await asyncio.sleep(delay)
|
||
|
||
# ── 3. Cian listings без external_valuations (price prediction backfill) ──
|
||
if do_valuations:
|
||
rows = db.execute(
|
||
text("""
|
||
SELECT l.id, l.address, l.area_m2, l.rooms, l.floor, l.total_floors
|
||
FROM listings l
|
||
WHERE l.source = 'cian'
|
||
AND l.address IS NOT NULL
|
||
AND l.area_m2 IS NOT NULL
|
||
AND l.rooms IS NOT NULL
|
||
AND l.floor IS NOT NULL
|
||
AND l.total_floors IS NOT NULL
|
||
AND NOT EXISTS (
|
||
SELECT 1 FROM external_valuations ev
|
||
WHERE ev.source = 'cian_valuation'
|
||
AND ev.listing_id = l.id
|
||
AND ev.expires_at > NOW()
|
||
)
|
||
LIMIT :lim
|
||
"""),
|
||
{"lim": batch_size},
|
||
).mappings().all()
|
||
result.valuations_total = len(rows)
|
||
|
||
if dry_run:
|
||
logger.info(
|
||
"dry_run: would process %d cian listings for valuation", result.valuations_total
|
||
)
|
||
else:
|
||
for row in rows:
|
||
result.valuations_processed += 1
|
||
try:
|
||
cval = await estimate_via_cian_valuation(
|
||
db,
|
||
address=row["address"],
|
||
total_area=float(row["area_m2"]),
|
||
rooms_count=int(row["rooms"]),
|
||
floor=int(row["floor"]),
|
||
total_floors=int(row["total_floors"]),
|
||
repair_type="cosmetic",
|
||
deal_type="sale",
|
||
use_cache=False, # force fresh fetch — populate cache
|
||
listing_id=int(row["id"]), # canonical link via mig 044
|
||
)
|
||
if cval is not None and cval.sale_price_rub:
|
||
result.valuations_succeeded += 1
|
||
else:
|
||
result.valuations_failed += 1
|
||
except Exception as exc:
|
||
logger.warning(
|
||
"cian_valuation backfill failed for listing_id=%s: %s", row["id"], exc
|
||
)
|
||
result.valuations_failed += 1
|
||
await asyncio.sleep(delay)
|
||
|
||
result.duration_sec = time.time() - t0
|
||
logger.info(
|
||
"cian_backfill done: listings=%d/%d (ok=%d fetch_fail=%d save_fail=%d "
|
||
"price_rows=%d) houses=%d/%d (ok=%d fail=%d) "
|
||
"valuations=%d/%d (ok=%d fail=%d) %.1fs",
|
||
result.listings_processed,
|
||
result.listings_total,
|
||
result.listings_succeeded,
|
||
result.listings_failed_fetch,
|
||
result.listings_failed_save,
|
||
result.price_changes_attempted,
|
||
result.houses_processed,
|
||
result.houses_total,
|
||
result.houses_succeeded,
|
||
result.houses_failed_fetch + result.houses_failed_save,
|
||
result.valuations_processed,
|
||
result.valuations_total,
|
||
result.valuations_succeeded,
|
||
result.valuations_failed,
|
||
result.duration_sec,
|
||
)
|
||
return result
|