gendesign/backend/app/services/site_finder/functional_zone_lookup.py
bot-backend 7cde3b8eaa
All checks were successful
CI Trade-In / changes (pull_request) Successful in 8s
CI / changes (pull_request) Successful in 10s
CI Trade-In / backend-tests (pull_request) Has been skipped
CI Trade-In / browser-tests (pull_request) Has been skipped
CI Trade-In / frontend-checks (pull_request) Has been skipped
CI / frontend-tests (pull_request) Has been skipped
CI / openapi-codegen-check (pull_request) Successful in 2m31s
CI / backend-tests (pull_request) Successful in 17m27s
fix(ptica): шесть lookup'ов /analyze глотали ошибку БД без SAVEPOINT (#2464)
Шесть функций в site_finder работают на ОБЩЕЙ с analyze_parcel сессии и
глотают (OperationalError, ProgrammingError) вокруг db.execute. Без SAVEPOINT
проглоченная ошибка оставляет транзакцию в aborted-состоянии: сам lookup
возвращает пустой результат «штатно», а падает следующий за ним запрос —
в другом месте отчёта и с другой причиной в логе.

Соседний ppt_tep_lookup — та же семья, та же таблица planning_projects —
SAVEPOINT имел. Правка сводит шесть отставших к нему:

  planning_lookup, granddoc_lookup, ird_overlay_lookup,
  functional_zone_lookup, reservation_lookup  — по одному db.execute
  zone_regulation                             — upsert был защищён, SELECT нет

Тест параметризован по всем шести именам, плюс два контроля: «без SAVEPOINT
сессия травится» и «здоровый путь по-прежнему читает строки». На origin/main
шесть падают с AbortedTransactionError, оба контроля зелёные.

Стабы _FakeDB/_DB в пяти существующих тестах получили no-op begin_nested —
без него 28 тестов падали на AttributeError. В шапке каждого стаба указано,
где SAVEPOINT проверяется по-настоящему, чтобы no-op не читался как покрытие.

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
2026-08-20 12:18:59 +05:00

74 lines
3.2 KiB
Python
Raw Permalink Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

"""Lookup функциональных зон генплана ЕКБ-2045 для analyze_parcel (#1058).
Читает ``ekb_genplan_functional_zone`` (м.139) и отдаёт зоны, пересекающие участок.
Graceful: таблица ещё не задеплоена / ошибка БД / пустой wkt → возвращает [].
Analyze-врезка (вшивание в build_ird_analyze_block / analyze_parcel) — отдельный
follow-up после merge #1131, чтобы не конфликтить с ird-analyze PR.
Зеркалит ird_overlay_lookup.py (стиль, graceful handling, CAST psycopg v3).
"""
from __future__ import annotations
import logging
from typing import Any
from sqlalchemy import text
from sqlalchemy.exc import OperationalError, ProgrammingError
from sqlalchemy.orm import Session
logger = logging.getLogger(__name__)
# Spatial overlap участка: ST_Intersects через GIST idx_ekb_genplan_fz_geom.
# Участок передаётся как WKT в 4326. CAST psycopg v3 (НИКОГДА :x::type).
_ZONE_OVERLAP_SQL = text(
"""
SELECT zone_type, status, area_ha, zone_class_code
FROM ekb_genplan_functional_zone
WHERE ST_Intersects(geom, ST_GeomFromText(CAST(:parcel_wkt AS text), 4326))
ORDER BY zone_type, status
"""
)
def parcel_functional_zones(db: Session, parcel_wkt: str | None) -> list[dict[str, Any]]:
"""Функциональные зоны генплана, пересекающие участок.
Args:
db: SQLAlchemy-сессия.
parcel_wkt: геометрия участка в WKT (EPSG:4326). None → пустой результат.
Returns:
Список dict: [{zone_type, status, area_ha, zone_class_code}].
Пусто при: нет wkt / таблица не задеплоена / нет пересечений / ошибка БД.
"""
if not parcel_wkt:
return []
try:
# #2464: SAVEPOINT перед проглатыванием ошибки. Сессия ОБЩАЯ с analyze_parcel
# (build_ird_analyze_block зовёт шесть таких lookup'ов подряд в одном словаре),
# и на Postgres упавший запрос оставляет транзакцию в aborted-состоянии —
# падают все следующие, включая запись прогона. Образец рядом:
# ppt_tep_lookup.py делает ровно так же.
with db.begin_nested():
rows = db.execute(_ZONE_OVERLAP_SQL, {"parcel_wkt": parcel_wkt}).mappings().all()
except (OperationalError, ProgrammingError) as exc:
# Таблица ещё не задеплоена или ошибка БД — graceful degrade (analyze не падает).
logger.warning(
"parcel_functional_zones: ekb_genplan_functional_zone недоступна, skip: %s", exc
)
return []
return [
{
"zone_type": r["zone_type"],
"status": r["status"],
"area_ha": float(r["area_ha"]) if r["area_ha"] is not None else None,
"zone_class_code": r["zone_class_code"],
}
for r in rows
]
__all__ = ["parcel_functional_zones"]