feat: make spatial entry optional and authoritative
This commit is contained in:
@@ -9,7 +9,7 @@ from sqlalchemy.orm import Session
|
||||
from app.db.session import get_db
|
||||
from app.models import Area
|
||||
from app.schemas import Envelope
|
||||
from app.schemas.area import AreaCreate, AreaList, AreaRead, AreaUpdate
|
||||
from app.schemas.area import AreaCreate, AreaList, AreaRead, AreaUpdate, MunicipalitySearchList
|
||||
from app.services.area_service import AreaService
|
||||
from app.utils.response import envelope
|
||||
|
||||
@@ -33,6 +33,23 @@ def create_area(project_id: UUID, payload: AreaCreate, db: Session = Depends(get
|
||||
return envelope(AreaService.serialize_area(area))
|
||||
|
||||
|
||||
@router.get("/municipalities", response_model=Envelope[MunicipalitySearchList])
|
||||
def search_municipalities(
|
||||
project_id: UUID,
|
||||
query: str = Query(default="", max_length=120),
|
||||
limit: int = Query(default=20, ge=1, le=50),
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
items, total = AreaService.search_municipalities(db, project_id, query, limit)
|
||||
return envelope({"items": items, "total": total})
|
||||
|
||||
|
||||
@router.post("/municipalities/{niscode}/activate", response_model=Envelope[AreaRead])
|
||||
def activate_municipality(project_id: UUID, niscode: str, db: Session = Depends(get_db)):
|
||||
area = AreaService.activate_municipality(db, project_id, niscode)
|
||||
return envelope(AreaService.serialize_area(area))
|
||||
|
||||
|
||||
@router.get("/{area_id}", response_model=Envelope[AreaRead])
|
||||
def get_area(
|
||||
project_id: UUID,
|
||||
|
||||
@@ -39,3 +39,16 @@ class AreaList(BaseModel):
|
||||
total: int
|
||||
limit: int
|
||||
offset: int
|
||||
|
||||
|
||||
class MunicipalitySearchItem(BaseModel):
|
||||
niscode: str
|
||||
name: str
|
||||
name_nl: str | None = None
|
||||
name_fr: str | None = None
|
||||
name_de: str | None = None
|
||||
|
||||
|
||||
class MunicipalitySearchList(BaseModel):
|
||||
items: list[MunicipalitySearchItem]
|
||||
total: int
|
||||
|
||||
@@ -7,12 +7,90 @@ from geoalchemy2.shape import from_shape, to_shape
|
||||
from shapely.geometry import mapping
|
||||
|
||||
from app.core.errors import AppError
|
||||
from app.models import Area, Project
|
||||
from app.models import Area, Dataset, Project, VectorFeature
|
||||
from app.schemas.area import AreaCreate, AreaRead, AreaUpdate
|
||||
from app.utils.geometry import area_m2, geometry_bbox_polygon, normalize_to_multipolygon
|
||||
|
||||
|
||||
class AreaService:
|
||||
@staticmethod
|
||||
def _municipality_dataset(db: Session, project_id: uuid.UUID) -> Dataset | None:
|
||||
return (
|
||||
db.query(Dataset)
|
||||
.filter(
|
||||
Dataset.project_id == project_id,
|
||||
Dataset.reference_layer_name == "belgium_municipalities",
|
||||
Dataset.status == "ready",
|
||||
)
|
||||
.order_by(Dataset.created_at.desc())
|
||||
.first()
|
||||
)
|
||||
|
||||
@staticmethod
|
||||
def _filter_municipality_properties(properties_items: list[dict], query: str, limit: int) -> tuple[list[dict], int]:
|
||||
normalized = query.strip().casefold()
|
||||
matches: list[dict] = []
|
||||
for properties in properties_items:
|
||||
names = [str(properties.get(key) or "").strip() for key in ("namedut", "namefre", "nameger")]
|
||||
niscode = str(properties.get("niscode") or "").strip()
|
||||
if normalized and normalized not in " ".join([niscode, *names]).casefold():
|
||||
continue
|
||||
display_name = next((name for name in names if name), niscode)
|
||||
matches.append({
|
||||
"niscode": niscode,
|
||||
"name": display_name,
|
||||
"name_nl": names[0] or None,
|
||||
"name_fr": names[1] or None,
|
||||
"name_de": names[2] or None,
|
||||
})
|
||||
matches.sort(key=lambda item: (item["name"].casefold(), item["niscode"]))
|
||||
return matches[:limit], len(matches)
|
||||
|
||||
@staticmethod
|
||||
def search_municipalities(db: Session, project_id: uuid.UUID, query: str, limit: int = 20) -> tuple[list[dict], int]:
|
||||
dataset = AreaService._municipality_dataset(db, project_id)
|
||||
if dataset is None:
|
||||
return [], 0
|
||||
property_rows = (
|
||||
db.query(VectorFeature.properties_json)
|
||||
.filter(VectorFeature.dataset_id == dataset.id)
|
||||
.all()
|
||||
)
|
||||
properties_items = [row[0] for row in property_rows if isinstance(row[0], dict)]
|
||||
return AreaService._filter_municipality_properties(properties_items, query, limit)
|
||||
|
||||
@staticmethod
|
||||
def activate_municipality(db: Session, project_id: uuid.UUID, niscode: str) -> Area:
|
||||
normalized_code = niscode.strip()
|
||||
dataset = AreaService._municipality_dataset(db, project_id)
|
||||
if dataset is None:
|
||||
raise AppError(code="MUNICIPALITY_NOT_FOUND", message="Municipality is not available in the official NGI administrative layer", status_code=404)
|
||||
feature = (
|
||||
db.query(VectorFeature)
|
||||
.filter(
|
||||
VectorFeature.dataset_id == dataset.id,
|
||||
VectorFeature.properties_json["niscode"].as_string() == normalized_code,
|
||||
)
|
||||
.first()
|
||||
)
|
||||
if feature is not None:
|
||||
properties = feature.properties_json if isinstance(feature.properties_json, dict) else {}
|
||||
display_name = next(
|
||||
(str(properties.get(key) or "").strip() for key in ("namedut", "namefre", "nameger") if str(properties.get(key) or "").strip()),
|
||||
normalized_code,
|
||||
)
|
||||
area_name = f"Gemeente {display_name} - NIS {normalized_code}"
|
||||
existing = db.query(Area).filter(Area.project_id == project_id, Area.name == area_name).first()
|
||||
if existing is not None:
|
||||
return existing
|
||||
geometry = to_shape(feature.geometry)
|
||||
return AreaService.create_area(
|
||||
db,
|
||||
project_id,
|
||||
AreaCreate(name=area_name, geometry=mapping(geometry), crs="EPSG:4326"),
|
||||
)
|
||||
raise AppError(code="MUNICIPALITY_NOT_FOUND", message="Municipality is not available in the official NGI administrative layer", status_code=404)
|
||||
|
||||
@staticmethod
|
||||
def serialize_area(area: Area) -> dict:
|
||||
geometry = to_shape(area.geometry) if area.geometry else None
|
||||
|
||||
Reference in New Issue
Block a user