async def _search_base(self, search_request: PgstacSearch,
**kwargs) -> Dict[str, Any]:
"""Cross catalog search (POST).
Called with `POST /search`.
Args:
search_request: search request parameters.
Returns:
ItemCollection containing items which match the search criteria.
"""
request = kwargs["request"]
pool = request.app.state.readpool
# pool = kwargs["request"].app.state.readpool
req = search_request.json(exclude_none=True)
async with pool.acquire() as conn:
q, p = render(
"""
SELECT * FROM search(:req::text::jsonb);
""",
req=req,
)
items = await conn.fetchval(q, *p)
next = items.pop("next", None)
prev = items.pop("prev", None)
collection = ItemCollection.construct(**items)
cleaned_features = []
if collection.features is None or len(collection.features) == 0:
raise NotFoundError("No features found")
for feature in collection.features:
feature = Item.construct(**feature)
if "links" not in search_request.fields.exclude:
links = await ItemLinks(
collection_id=feature.collection,
item_id=feature.id,
request=request,
).get_links()
feature.links = links
exclude = search_request.fields.exclude
if len(exclude) == 0:
exclude = None
include = search_request.fields.include
if len(include) == 0:
include = None
feature = feature.dict(exclude_none=True, )
cleaned_features.append(feature)
collection.features = cleaned_features
collection.links = await PagingLinks(
request=request,
next=next,
prev=prev,
).get_links()
return collection
async def _search_base(self, search_request: PgstacSearch,
**kwargs: Any) -> ItemCollection:
"""Cross catalog search (POST).
Called with `POST /search`.
Args:
search_request: search request parameters.
Returns:
ItemCollection containing items which match the search criteria.
"""
items: Dict[str, Any]
request: Request = kwargs["request"]
pool = request.app.state.readpool
# pool = kwargs["request"].app.state.readpool
req = search_request.json(exclude_none=True)
try:
async with pool.acquire() as conn:
q, p = render(
"""
SELECT * FROM search(:req::text::jsonb);
""",
req=req,
)
items = await conn.fetchval(q, *p)
except InvalidDatetimeFormatError:
raise InvalidQueryParameter(
f"Datetime parameter {search_request.datetime} is invalid.")
next: Optional[str] = items.pop("next", None)
prev: Optional[str] = items.pop("prev", None)
collection = ItemCollection(**items)
cleaned_features: List[Item] = []
for feature in collection.get("features") or []:
feature = Item(**feature)
if (search_request.fields.exclude is None
or "links" not in search_request.fields.exclude):
# TODO: feature.collection is not always included
# This code fails if it's left outside of the fields expression
# I've fields extension updated test cases to always include feature.collection
feature["links"] = await ItemLinks(
collection_id=feature["collection"],
item_id=feature["id"],
request=request,
).get_links(extra_links=feature.get("links"))
exclude = search_request.fields.exclude
if exclude and len(exclude) == 0:
exclude = None
include = search_request.fields.include
if include and len(include) == 0:
include = None
cleaned_features.append(feature)
collection["features"] = cleaned_features
collection["links"] = await PagingLinks(
request=request,
next=next,
prev=prev,
).get_links()
return collection