feat(tradein): secondary_only — параметр расписания, выброшенное считается (#1781) #3008

Merged
bot-backend merged 1 commit from fix/1781-secondary-only-param into main 2026-08-20 20:30:01 +00:00
4 changed files with 158 additions and 2 deletions

View file

@ -1288,6 +1288,15 @@ class CianFullLoadRequest(BaseModel):
"cian_full_load прогона для resume. Без этого поля — full walk с нуля."
),
)
secondary_only: bool = Field(
default=True,
description=(
"True (дефолт, поведение без изменений) — новостройки отбрасываются "
"после разбора. False — сохраняются вместе со вторичкой. Дополнительных "
"HTTP-запросов не стоит: страницы с ними всё равно скачиваются, "
"totalOffers бакета считает обе категории (#1781)."
),
)
@router.post("/scrape/cian-full-load", response_model=CitySweepStartResponse)
@ -1326,6 +1335,7 @@ async def start_cian_full_load(
enrich_detail=payload.enrich_detail,
detail_top_n=payload.detail_top_n,
resume_run_id=payload.resume_run_id,
secondary_only=payload.secondary_only,
)
except Exception:
logger.exception("cian-full-load background task run_id=%d crashed", run_id)

View file

@ -0,0 +1,123 @@
"""`secondary_only` — параметр расписания, а не хардкод; выброшенное считается (#1781).
`run_cian_full_load` передавал `secondary_only=True` жёстко (`pipeline.py:3328`), поэтому
включить новостройки в полный обход можно было только деплоем. При этом новостройки
НЕ пропускаются при запросе они скачиваются, разбираются и выбрасываются последним
шагом (`cian/serp.py:671`), потому что SERP-параметр `object_type=1` у Cian ненадёжен
(~5 % выдачи) и фильтруют по authoritative `listing_segment` уже после парсинга.
Следствие, проверенное по коду: проба бакета берёт `totalOffers` из Redux-состояния
SERP и считает `pages_needed = ceil(totalOffers / offers_per_page)`; `totalOffers`
включает обе категории. То есть страницы с новостройками уже скачаны, лимит страниц и
антибан-бюджет за них уже заплачены включение стоит НОЛЬ дополнительных запросов.
Дефолт НЕ меняется: `True`, поведение прода прежнее. Меняется только то, что решение
становится правкой одной ячейки `scrape_schedules.default_params`, а не деплоем.
Замер прода 21.08.2026, ради которого это и делается: месячный охват свипа
cian/novostroyki 11.7 % против 100 % у cian/vtorichka и avito/novostroyki;
11 993 активные строки, медианный возраст 81 сутки.
"""
from __future__ import annotations
import os
os.environ.setdefault("DATABASE_URL", "postgresql+psycopg://test:test@localhost:5432/test")
import inspect
def test_full_load_no_longer_hardcodes_secondary_only() -> None:
"""Головной: значение берётся из параметра, а не из литерала.
На origin/main в теле стоит `secondary_only=True` включить новостройки
можно только правкой кода.
"""
from scraper_kit.orchestration import pipeline
src = inspect.getsource(pipeline.run_cian_full_load)
assert "secondary_only=secondary_only" in src, "значение не пробрасывается из параметра"
assert (
"secondary_only=True," not in src
), "в теле остался хардкод — расписание на него повлиять не сможет"
def test_default_is_unchanged() -> None:
"""Контроль: дефолт остаётся True — поведение прода не меняется этим PR.
Без этого контроля правка «сделать параметром» могла бы тихо включить сбор
новостроек на проде, а это отдельное решение с замером (#1781, #2994).
"""
from scraper_kit.orchestration.pipeline import run_cian_full_load
params = inspect.signature(run_cian_full_load).parameters
assert "secondary_only" in params, (
"параметра нет в сигнатуре — значение задано в теле, расписание на него "
f"повлиять не может; параметры: {list(params)}"
)
default = params["secondary_only"].default
assert default is True, f"дефолт изменён на {default!r} — это не входило в правку"
def test_admin_request_exposes_the_flag_with_same_default() -> None:
"""Параметр доезжает до расписания: у эндпоинта есть поле с тем же дефолтом."""
from app.api.v1.admin import CianFullLoadRequest
поля = CianFullLoadRequest.model_fields
assert "secondary_only" in поля, (
"поля нет в запросе — параметр не доедет из scrape_schedules.default_params; "
f"поля: {sorted(поля)}"
)
assert поля["secondary_only"].default is True
req = CianFullLoadRequest()
assert req.secondary_only is True
assert CianFullLoadRequest(secondary_only=False).secondary_only is False
def test_counters_persist_dropped_novostroyki() -> None:
"""Выброшенное обязано сохраняться, а не только логироваться.
Счётчик `dropped_nb` печатался в лог, но в `scrape_runs.counters` не попадал, и
ответить «сколько инвентаря выбрасывает полный обход» задним числом было нечем:
логи за нужную дату уже ротировались. Тот же довод, по которому рядом заведён
`partial_buckets`.
"""
from scraper_kit.orchestration.pipeline import CianFullLoadCounters
c = CianFullLoadCounters()
assert "dropped_novostroyki" in c.to_dict(), c.to_dict()
c.dropped_novostroyki = 7
assert c.to_dict()["dropped_novostroyki"] == 7
def test_scraper_resets_the_counter_per_run() -> None:
"""Контроль: счётчик сбрасывается на каждый прогон.
Инстанс скрапера переиспользуется; без сброса второй прогон унаследовал бы
число первого и записал бы в counters завышенное значение.
"""
from scraper_kit.providers.cian.serp import CianScraper
src = inspect.getsource(CianScraper.fetch_all_secondary)
assert "self.last_dropped_nb = 0" in src, "нет сброса в начале прогона"
assert "self.last_dropped_nb += " in inspect.getsource(
CianScraper._paginate_leaf_bucket
), "накопление не там, где считается dropped_nb"
assert (
getattr(CianScraper, "last_dropped_nb", None) == 0
), "нет класс-дефолта: атрибут не прочитается, если прогон упал до первого бакета"
def test_filter_still_drops_when_flag_is_on() -> None:
"""Контроль от переусердствования: при secondary_only=True фильтр остаётся.
Правка делает флаг управляемым, а не отменяет его.
"""
from scraper_kit.providers.cian.serp import CianScraper
# Фильтр лежит в _paginate_leaf_bucket — том методе, что собирает страницы
# бакета; fetch_all_secondary только раздаёт флаг вниз.
src = inspect.getsource(CianScraper._paginate_leaf_bucket)
assert 'lot.listing_segment != "novostroyki"' in src, "фильтр пропал"
assert "if secondary_only:" in src, "фильтр перестал зависеть от флага"

View file

@ -3147,6 +3147,14 @@ class CianFullLoadCounters:
# Без этого счётчика «сколько бакетов чекпоинт не покрывает» видно только грепом
# логов, которые теряются при редеплое.
partial_buckets: int = 0
# Новостройки, отброшенные фильтром secondary_only ПОСЛЕ скачивания и разбора
# (#1781). Ровно тот же довод, что у partial_buckets выше: счётчик логировался
# (`dropped_nb=` в cian/serp.py), но не сохранялся, и ответить «сколько инвентаря
# выбрасывает полный обход» задним числом было нечем — логи за нужную дату уже
# ротировались. Заодно это цена вопроса при обсуждении secondary_only=False:
# страницы с этими лотами УЖЕ скачаны, лимит страниц и антибан-бюджет за них
# уже заплачены, выбрасывается только результат разбора.
dropped_novostroyki: int = 0
def to_dict(self) -> dict[str, int]:
return {f.name: getattr(self, f.name) for f in fields(self)}
@ -3167,6 +3175,7 @@ async def run_cian_full_load(
concurrency: int = 5,
resume_run_id: int | None = None,
region_code: int = DEFAULT_REGION_CODE,
secondary_only: bool = True,
) -> CianFullLoadCounters:
"""Exhaustive региональный сбор Cian ЕКБ вторички (БЕЗ anchor'ов).
@ -3325,18 +3334,21 @@ async def run_cian_full_load(
await scraper.fetch_all_secondary(
price_cap_per_bucket=price_cap_per_bucket,
concurrency=concurrency,
secondary_only=True,
secondary_only=secondary_only,
on_bucket=_on_bucket,
on_progress=_on_progress,
skip_buckets=skip_set if skip_set else None,
)
counters.dropped_novostroyki = getattr(scraper, "last_dropped_nb", 0)
logger.info(
"cian-full-load run_id=%d: fetch done — unique=%d ins=%d upd=%d",
"cian-full-load run_id=%d: fetch done — unique=%d ins=%d upd=%d "
"dropped_novostroyki=%d",
run_id,
counters.unique_fetched,
counters.saved_inserted,
counters.saved_updated,
counters.dropped_novostroyki,
)
runs.update_heartbeat(db, run_id, counters.to_dict())

View file

@ -122,6 +122,10 @@ class CianScraper(BaseScraper):
base_url = "https://ekb.cian.ru"
# Класс-дефолт; реальное значение загружается из scraper_settings при создании экземпляра.
request_delay_sec = 5.0 # консервативно: Cian менее агрессивен чем Avito, но 5s безопасно
# Сколько новостроек отброшено фильтром secondary_only за последний прогон
# fetch_all_secondary (#1781). Класс-дефолт нужен, чтобы атрибут читался даже
# если прогон упал до первого бакета.
last_dropped_nb: int = 0
def __init__(
self,
@ -401,6 +405,9 @@ class CianScraper(BaseScraper):
"""
_buckets = rooms_buckets if rooms_buckets is not None else _DEFAULT_ROOMS_BUCKETS
seen: dict[str, ScrapedLot] = {}
# Сброс на КАЖДЫЙ прогон (#1781): инстанс скрапера переиспользуется, и без
# сброса счётчик копился бы между вызовами и врал бы в counters второго.
self.last_dropped_nb = 0
for rooms in _buckets:
room_label = f"room{'_'.join(str(r) for r in rooms)}"
@ -671,6 +678,10 @@ class CianScraper(BaseScraper):
filtered = [lot for lot in bucket_lots if lot.listing_segment != "novostroyki"]
dropped_nb = collected_this_bucket - len(filtered)
bucket_lots = filtered
# Копим по всему прогону в атрибут инстанса (#1781): вызывающий кладёт это
# в counters прогона. Атрибут, а не аргумент on_bucket — у колбэка уже есть
# внешние реализации, менять его сигнатуру ради счётчика нельзя.
self.last_dropped_nb += dropped_nb
# Дедуп в общий seen
for lot in bucket_lots: