mirror of
https://github.com/fastapi-users/fastapi-users.git
synced 2025-11-04 14:45:50 +08:00
c4de66b81c
* Implement Transport classes * Implement authentication strategy classes * Revamp authentication with Transport and Strategy * Revamp strategy and OAuth so that they can use a callable dependency * Update docstring * Make ErrorCode a proper Enum and cleanup unused OpenAPI utils * Remove useless check * Tweak typing in authenticator * Update docs * Improve logout/destroy token logic * Update docs * Update docs * Update docs and full examples * Apply formatting to examples * Update OAuth doc and examples * Add migration doc * Implement Redis session token * Add Redis Session documentation * RedisSession -> Redis * Fix links in docs
66 lines
1.7 KiB
Python
66 lines
1.7 KiB
Python
from typing import Any, Callable, Generic
|
|
|
|
from fastapi import Response
|
|
|
|
from fastapi_users import models
|
|
from fastapi_users.authentication.strategy import (
|
|
Strategy,
|
|
StrategyDestroyNotSupportedError,
|
|
)
|
|
from fastapi_users.authentication.transport import (
|
|
Transport,
|
|
TransportLogoutNotSupportedError,
|
|
)
|
|
|
|
|
|
class AuthenticationBackend(Generic[models.UC, models.UD]):
|
|
"""
|
|
Combination of an authentication transport and strategy.
|
|
|
|
Together, they provide a full authentication method logic.
|
|
|
|
:param name: Name of the backend.
|
|
:param transport: Authentication transport instance.
|
|
:param get_strategy: Dependency callable returning
|
|
an authentication strategy instance.
|
|
"""
|
|
|
|
name: str
|
|
transport: Transport
|
|
|
|
def __init__(
|
|
self,
|
|
name: str,
|
|
transport: Transport,
|
|
get_strategy: Callable[..., Strategy[models.UC, models.UD]],
|
|
):
|
|
self.name = name
|
|
self.transport = transport
|
|
self.get_strategy = get_strategy
|
|
|
|
async def login(
|
|
self,
|
|
strategy: Strategy[models.UC, models.UD],
|
|
user: models.UD,
|
|
response: Response,
|
|
) -> Any:
|
|
token = await strategy.write_token(user)
|
|
return await self.transport.get_login_response(token, response)
|
|
|
|
async def logout(
|
|
self,
|
|
strategy: Strategy[models.UC, models.UD],
|
|
user: models.UD,
|
|
token: str,
|
|
response: Response,
|
|
) -> Any:
|
|
try:
|
|
await strategy.destroy_token(token, user)
|
|
except StrategyDestroyNotSupportedError:
|
|
pass
|
|
|
|
try:
|
|
await self.transport.get_logout_response(response)
|
|
except TransportLogoutNotSupportedError:
|
|
return None
|