From: dmontagu <35119617+dmontagu@users.noreply.github.com>
Date: Sat, 31 Aug 2019 00:05:59 +0000 (-0700)
Subject: :sparkles: Add support for custom route class (#468)
X-Git-Tag: 0.37.0~2
X-Git-Url: http://git.ipfire.org/?a=commitdiff_plain;h=c70f3f119885d299c2987ee86a1e867beeb9ff3d;p=thirdparty%2Ffastapi%2Ffastapi.git

:sparkles: Add support for custom route class (#468)
---

diff --git a/fastapi/routing.py b/fastapi/routing.py
index 08f43bf524..0e8ff9739c 100644
--- a/fastapi/routing.py
+++ b/fastapi/routing.py
@@ -317,11 +317,13 @@ class APIRouter(routing.Router):
         redirect_slashes: bool = True,
         default: ASGIApp = None,
         dependency_overrides_provider: Any = None,
+        route_class: Type[APIRoute] = APIRoute,
     ) -> None:
         super().__init__(
             routes=routes, redirect_slashes=redirect_slashes, default=default
         )
         self.dependency_overrides_provider = dependency_overrides_provider
+        self.route_class = route_class
 
     def add_api_route(
         self,
@@ -347,7 +349,7 @@ class APIRouter(routing.Router):
         response_class: Type[Response] = JSONResponse,
         name: str = None,
     ) -> None:
-        route = APIRoute(
+        route = self.route_class(
             path,
             endpoint=endpoint,
             response_model=response_model,