diff --git a/backend/app/api/v1/parcels.py b/backend/app/api/v1/parcels.py index 8b65121f..0e18adc4 100644 --- a/backend/app/api/v1/parcels.py +++ b/backend/app/api/v1/parcels.py @@ -1658,33 +1658,34 @@ def analyze_parcel( "lon": centroid_lon, } try: - zoning_row = ( - db.execute( - text(""" - SELECT zone_code, zone_name, description, rosreestr_id - FROM pzz_zones_ekb - WHERE ST_Within( - ST_Centroid(ST_GeomFromText(:wkt, 4326)), - geom - ) - LIMIT 1 - """), - {"wkt": geom_wkt}, - ) - .mappings() - .first() - ) - if zoning_row: - zoning.update( - { - "zone_code": zoning_row["zone_code"], - "zone_name": zoning_row["zone_name"], - "description": zoning_row["description"], - "rosreestr_id": zoning_row["rosreestr_id"], - "data_available": True, - "source": "rosreestr-pkk6-cached", - } + with db.begin_nested(): + zoning_row = ( + db.execute( + text(""" + SELECT zone_code, zone_name, description, rosreestr_id + FROM pzz_zones_ekb + WHERE ST_Within( + ST_Centroid(ST_GeomFromText(:wkt, 4326)), + geom + ) + LIMIT 1 + """), + {"wkt": geom_wkt}, + ) + .mappings() + .first() ) + if zoning_row: + zoning.update( + { + "zone_code": zoning_row["zone_code"], + "zone_name": zoning_row["zone_name"], + "description": zoning_row["description"], + "rosreestr_id": zoning_row["rosreestr_id"], + "data_available": True, + "source": "rosreestr-pkk6-cached", + } + ) except Exception as e: logger.warning("zoning query failed for %s: %s", cad_num, e) @@ -1692,9 +1693,10 @@ def analyze_parcel( success_recommendation: dict[str, Any] | None = None if district_row: try: - success_rows = ( - db.execute( - text(""" + with db.begin_nested(): + success_rows = ( + db.execute( + text(""" SELECT bucket, success_score, n_deals, avg_price_per_m2, avg_area_m2, velocity_z, price_z, area_z FROM v_bucket_success_score @@ -1702,35 +1704,35 @@ def analyze_parcel( ORDER BY success_score DESC LIMIT 5 """), - {"dn": district_row["district_name"]}, + {"dn": district_row["district_name"]}, + ) + .mappings() + .all() ) - .mappings() - .all() - ) - if success_rows: - success_recommendation = { - "district": district_row["district_name"], - "ranking": [ - { - "bucket": r["bucket"], - "success_score": round(float(r["success_score"]), 2), - "n_deals": int(r["n_deals"]), - "avg_price_per_m2": ( - int(r["avg_price_per_m2"]) if r["avg_price_per_m2"] else None - ), - "avg_area_m2": ( - round(float(r["avg_area_m2"]), 1) if r["avg_area_m2"] else None - ), - } - for r in success_rows - ], - "top_bucket": dict(success_rows[0]) if success_rows else None, - "note": ( - "Топ комнатность по 'успешности' = z-scores: velocity×0.5 + price×0.3 " - "- area×0.2. Min 30 сделок в группе за 24 мес. " - "Используй для квартирографии проекта." - ), - } + if success_rows: + success_recommendation = { + "district": district_row["district_name"], + "ranking": [ + { + "bucket": r["bucket"], + "success_score": round(float(r["success_score"]), 2), + "n_deals": int(r["n_deals"]), + "avg_price_per_m2": ( + int(r["avg_price_per_m2"]) if r["avg_price_per_m2"] else None + ), + "avg_area_m2": ( + round(float(r["avg_area_m2"]), 1) if r["avg_area_m2"] else None + ), + } + for r in success_rows + ], + "top_bucket": dict(success_rows[0]) if success_rows else None, + "note": ( + "Топ комнатность по 'успешности' = z-scores: velocity×0.5 + price×0.3 " + "- area×0.2. Min 30 сделок в группе за 24 мес. " + "Используй для квартирографии проекта." + ), + } except Exception as e: logger.warning("success_recommendation query failed for %s: %s", cad_num, e) success_recommendation = None diff --git a/backend/app/services/site_finder/velocity.py b/backend/app/services/site_finder/velocity.py index e7de6ec0..7ba7f9e2 100644 --- a/backend/app/services/site_finder/velocity.py +++ b/backend/app/services/site_finder/velocity.py @@ -74,11 +74,15 @@ def compute_velocity( # obj_class в domrf_kn_objects заполнен слабо (много NULL); фильтруем # только если явно передан. class_filter = "AND o.obj_class = :obj_class" if obj_class else "" + # SAVEPOINT per query: failure rollbacks ТОЛЬКО savepoint, не outer tx. + # db.rollback() здесь НЕЛЬЗЯ — он orphan'ит outer SessionTransaction + # (см. PR #155 bot review — SQLAlchemy 2.0 begin_nested context cleanup). try: - comp_rows = ( - db.execute( - text( - f""" + with db.begin_nested(): + comp_rows = ( + db.execute( + text( + f""" WITH latest_obj AS ( SELECT DISTINCT ON (obj_id) obj_id, @@ -114,18 +118,20 @@ def compute_velocity( ORDER BY distance_m ASC LIMIT 200 """ - ), - { - "parcel_wkt": parcel_geom_wkt, - "radius_m": radius_km * 1000.0, - "obj_class": obj_class, - }, + ), + { + "parcel_wkt": parcel_geom_wkt, + "radius_m": radius_km * 1000.0, + "obj_class": obj_class, + }, + ) + .mappings() + .all() ) - .mappings() - .all() - ) except Exception: logger.exception("velocity: competitor query failed for wkt=%s", parcel_geom_wkt[:80]) + # SAVEPOINT auto-rollbacks через __exit__ context manager. + # Outer tx остаётся clean — caller продолжает работать без cascade. return None if not comp_rows: @@ -147,10 +153,11 @@ def compute_velocity( # area_sq = м² за месяц (primary). Если NULL — realised * 45 м² heuristic. # type = 'apartments' — только жильё. try: - sales_rows = ( - db.execute( - text( - """ + with db.begin_nested(): + sales_rows = ( + db.execute( + text( + """ WITH latest_sg AS ( SELECT DISTINCT ON (obj_id, report_month) obj_id, @@ -175,17 +182,18 @@ def compute_velocity( WHERE area_sq > 0 OR realised > 0 GROUP BY obj_id """ - ), - { - "obj_ids": obj_ids, - "window_interval": f"{months_window} months", - }, + ), + { + "obj_ids": obj_ids, + "window_interval": f"{months_window} months", + }, + ) + .mappings() + .all() ) - .mappings() - .all() - ) except Exception: logger.exception("velocity: sale_graph query failed for obj_ids=%s", obj_ids[:5]) + # SAVEPOINT auto-rollback'нут — outer tx clean return None if not sales_rows: @@ -266,10 +274,11 @@ def _get_ekb_median(db: Session, months_window: int = 6) -> float | None: Fallback к _EKB_MEDIAN_FALLBACK_SQM_PER_MONTH если нет данных в БД. """ try: - row = ( - db.execute( - text( - """ + with db.begin_nested(): + row = ( + db.execute( + text( + """ WITH latest_sg AS ( SELECT DISTINCT ON (obj_id, report_month) obj_id, @@ -303,14 +312,15 @@ def _get_ekb_median(db: Session, months_window: int = 6) -> float | None: SELECT PERCENTILE_CONT(0.5) WITHIN GROUP (ORDER BY velocity) AS median FROM per_obj_velocity """ - ), - {"window_interval": f"{months_window} months"}, + ), + {"window_interval": f"{months_window} months"}, + ) + .mappings() + .first() ) - .mappings() - .first() - ) except Exception: logger.warning("velocity: ekb_median query failed, using fallback") + # SAVEPOINT auto-rollback'нут return None if row and row["median"] is not None: