From 1fa87b7bd517008a9cdd37824ed0ec0a6315b284 Mon Sep 17 00:00:00 2001 From: Thomas Maschler Date: Mon, 12 Feb 2024 12:22:41 -0500 Subject: [PATCH] test wrapper --- stac_fastapi/api/stac_fastapi/api/routes.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/stac_fastapi/api/stac_fastapi/api/routes.py b/stac_fastapi/api/stac_fastapi/api/routes.py index f4eb759af..1c2449ceb 100644 --- a/stac_fastapi/api/stac_fastapi/api/routes.py +++ b/stac_fastapi/api/stac_fastapi/api/routes.py @@ -1,4 +1,5 @@ """Route factories.""" + import functools import inspect from typing import Any, Callable, Dict, List, Optional, Type, TypedDict, Union @@ -6,6 +7,7 @@ from fastapi import Depends, params from fastapi.dependencies.utils import get_parameterless_sub_dependant from pydantic import BaseModel +from stac_pydantic.api import LandingPage from starlette.concurrency import run_in_threadpool from starlette.requests import Request from starlette.responses import JSONResponse, Response @@ -18,6 +20,8 @@ def _wrap_response(resp: Any, response_class: Type[Response]) -> Response: if isinstance(resp, Response): return resp + elif isinstance(resp, LandingPage): + return resp elif resp is not None: return response_class(resp) else: # None is returned as 204 No Content