pryscraper/routers/reconciliation.py
cryptorugmunch 9fee12796e
Some checks failed
CI / Security audit (bandit) (pull_request) Successful in 36s
CI / test (pull_request) Successful in 1m19s
CI / lint (pull_request) Failing after 49s
CI / typecheck (pull_request) Successful in 54s
CI / Secret scan (gitleaks) (pull_request) Successful in 34s
feat(routers): split remaining 174 api.py routes into per-tag routers
- AST-extract all remaining routes from api.py into routers/*.py
- Group routes by OpenAPI tag (AI, Advanced, Agency, ..., x402)
- Keep existing auth/health/scraping/templates routers; add *_api.py for remaining routes of those tags
- Move shared helpers/models/variables with the routes that need them
- Update tests/test_api.py to import vision helpers from routers.vision
- Regenerate openapi.json (186 paths)
- All 497 tests pass; ruff clean
2026-07-03 03:42:36 +02:00

71 lines
2.1 KiB
Python

"""Pry — Reconciliation router (remaining api.py routes).
Auto-extracted from api.py during the router-split refactor.
"""
# SPDX-License-Identifier: MIT
# Copyright (c) 2026 Rug Munch Media LLC
from __future__ import annotations
import logging
from typing import Any
from fastapi import APIRouter, Body
from errors import InvalidRequestError
logger = logging.getLogger(__name__)
router = APIRouter(tags=["Reconciliation"])
@router.post(
"/v1/reconcile",
tags=["Reconciliation"],
summary="Reconcile records from multiple sources into unified entities",
)
async def reconcile_endpoint(
records: list[dict[str, Any]] = Body(...),
vertical: str = Body("product"),
threshold: float = Body(0.7),
) -> dict[str, Any]:
"""Reconcile records from multiple sources into matched entities.
Matches records across sources using identity field similarity,
normalizes to a unified vertical schema, and returns entity groups
with confidence scores.
Verticals: product, job, real_estate, review
"""
from reconciliation import VERTICAL_SCHEMAS, reconcile
if vertical not in VERTICAL_SCHEMAS:
raise InvalidRequestError(
f"Unknown vertical: {vertical}. Supported: {list(VERTICAL_SCHEMAS.keys())}"
)
result = await reconcile(records, vertical, threshold)
return {"success": True, "data": result}
@router.get(
"/v1/reconcile/schemas",
tags=["Reconciliation"],
summary="List supported reconciliation schemas",
)
async def list_schemas() -> dict[str, Any]:
"""List all supported vertical schemas for entity reconciliation."""
from reconciliation import VERTICAL_SCHEMAS
schemas = {}
for key, schema in VERTICAL_SCHEMAS.items():
schemas[key] = {
"name": schema["name"],
"fields": {
k: {fk: fv for fk, fv in v.items() if fk != "type"}
for k, v in schema["fields"].items()
},
"identity_fields": schema["identity_fields"],
}
return {"success": True, "data": {"schemas": schemas, "total": len(schemas)}}