Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 7 additions & 2 deletions api/location.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
# ===============================================================================
from typing import Union

from fastapi import Depends
from fastapi import Depends, Query
from fastapi_pagination.ext.sqlalchemy import paginate
from sqlalchemy import select, func
from sqlalchemy.orm import Session
Expand All @@ -29,7 +29,7 @@
from schemas_v2.location import CreateLocation, LocationResponse, UpdateLocation
from schemas_v2.thing import LocationWellResponse
from services.geospatial_helper import make_within_wkt
from services.query_helper import make_query
from services.query_helper import make_query, order_sort_filter
from services.crud_helper import model_patcher

from fastapi import APIRouter
Expand Down Expand Up @@ -133,6 +133,9 @@ async def get_location(
within: str = None,
query: str = None,
expand: str = None,
sort: str = None,
order: str = None,
filter_: str = Query(alias="filter", default=None),
) -> CustomPage[Union[LocationResponse, LocationWellResponse]]:
"""
Retrieve all wells from the database.
Expand All @@ -159,6 +162,8 @@ def transformer(items):
return [LocationWellResponse.model_validate(item) for item in items]
return [LocationResponse.model_validate(item) for item in items]

sql = order_sort_filter(sql, Location, sort, order, filter_)

return paginate(query=sql, conn=session, transformer=transformer)


Expand Down
Loading