From 31f8757a1c44933e41f615b098e7fd28bb602cd0 Mon Sep 17 00:00:00 2001 From: ilyaTT Date: Tue, 16 Aug 2022 17:40:54 +0400 Subject: [PATCH] Give the ability to set any route class when adding a third-party router --- fastapi/routing.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/fastapi/routing.py b/fastapi/routing.py index 6f1a8e900..b86df20ca 100644 --- a/fastapi/routing.py +++ b/fastapi/routing.py @@ -685,6 +685,7 @@ class APIRouter(routing.Router): generate_unique_id_function: Callable[[APIRoute], str] = Default( generate_unique_id ), + route_class_override: Optional[Type[APIRoute]] = None, ) -> None: if prefix: assert prefix.startswith("/"), "A path prefix must start with '/'" @@ -756,7 +757,7 @@ class APIRouter(routing.Router): and include_in_schema, response_class=use_response_class, name=route.name, - route_class_override=type(route), + route_class_override=route_class_override or type(route), callbacks=current_callbacks, openapi_extra=route.openapi_extra, generate_unique_id_function=current_generate_unique_id,