backend as a parameter

pysnippet · Mar 6, 2024 · 1554f2a · 1554f2a
1 parent 800028a
commit 1554f2a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/fastapi_oauth2/middleware.py b/src/fastapi_oauth2/middleware.py
@@ -128,6 +128,7 @@ def __init__(
             app: ASGIApp,
             config: Union[OAuth2Config, dict],
             callback: Callable[[Auth, User], Union[Awaitable[None], None]] = None,
+            backend: Optional[OAuth2Backend]=None,
             **kwargs,  # AuthenticationMiddleware kwargs
     ) -> None:
         """Initiates the middleware with the given configuration.
@@ -141,7 +142,8 @@ def __init__(
         elif not isinstance(config, OAuth2Config):
             raise TypeError("config is not a valid type")
         self.default_application_middleware = app
-        self.auth_middleware = AuthenticationMiddleware(app, backend=OAuth2Backend(config, callback), **kwargs)
+        backend = backend or OAuth2Backend(config, callback)
+        self.auth_middleware = AuthenticationMiddleware(app, backend=backend, **kwargs)
 
     async def __call__(self, scope: Scope, receive: Receive, send: Send) -> None:
         if scope["type"] == "http":